From 51585ae2efa309d412cf6f558327032d304200cf Mon Sep 17 00:00:00 2001
From: FloRul <rumiel.florian@gmail.com>
Date: Wed, 20 Mar 2024 21:35:07 -0400
Subject: [PATCH] Add logging statements for debugging and improve code
 readability

---
 lambdas/inference/src/index.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/lambdas/inference/src/index.py b/lambdas/inference/src/index.py
index 117d003..07f7665 100644
--- a/lambdas/inference/src/index.py
+++ b/lambdas/inference/src/index.py
@@ -114,17 +114,21 @@ def lambda_handler(event, context):
     try:
         source = event.get("queryStringParameters", {}).get("source", "message")
         embedding_collection_name = event["queryStringParameters"]["collectionName"]
-
+        
+        logger.info(str(event))
+        
         sessionId = str(uuid.uuid1())
 
         if "sessionId" in event["queryStringParameters"]:
             sessionId = event["queryStringParameters"]["sessionId"]
 
+        logger.info(f"loading history for session {sessionId}")
         history = History(session_id=sessionId)
 
         query = event["queryStringParameters"]["query"]
         docs = []
 
+        logger.info(f"intializing retrieval for query {query}")
         # fetch documents
         retrieval = Retrieval(
             collection_name=embedding_collection_name,
@@ -132,9 +136,11 @@ def lambda_handler(event, context):
         )
         docs = retrieval.fetch_documents(query=query, top_k=ENV_VARS["top_k"])
 
+        logger.info("preparing system prompt...")
         # prepare the prompt
         system_prompt = prepare_system_prompt(docs, source)
 
+        logger.info("fetching chat history...")
         chat_history = history.get(limit=5)
         chat_history.append({"role": "user", "content": query})