Update main.py

main.py

@@ -213,7 +213,8 @@ async def chat(request: ChatRequest, background_tasks: BackgroundTasks, api_key:
             "conversation_id": conversation_id,
             "user_id": request.user_id
         }
-
+        
+        logger.info(f"Starting chat response generation for user: {request.user_id} Full request: {llm_request}")
         async def response_generator():
             full_response = ""
             async for chunk in stream_llm_request(api_key, llm_request):
@@ -224,7 +225,7 @@ async def chat(request: ChatRequest, background_tasks: BackgroundTasks, api_key:
             # For example:
             # background_tasks.add_task(save_conversation, request.user_id, conversation_id, request.query, full_response)
 
-        logger.info(f"Starting chat response generation for user: {request.user_id}")
+        logger.info(f"Finished chat response generation for user: {request.user_id} Full response{full_response}")
         return StreamingResponse(response_generator(), media_type="text/event-stream")
 
     except Exception as e: