add logging

agent_api.py

@@ -340,6 +340,20 @@ class CyberLegalAPI:
                 error=result.get("error")
             )
             
+            # Log detailed response being sent to client
+            logger.info("=" * 80)
+            logger.info("📤 SENDING CHAT RESPONSE TO CLIENT")
+            logger.info("=" * 80)
+            logger.info(f"👤 User ID: {request.userId}")
+            logger.info(f"📅 Timestamp: {response.timestamp}")
+            logger.info(f"⏱️ Processing time: {response.processing_time:.2f}s")
+            logger.info(f"📝 Response length: {len(response.response)} characters")
+            logger.info(f"💬 Response preview (first 200 chars): {response.response}")
+            if response.error:
+                logger.warning(f"⚠️ Error in response: {response.error}")
+            logger.info(f"🔢 Full response object: {response.model_dump()}")
+            logger.info("=" * 80)
+            
             logger.info(f"📤 Returning response to user")
             return response
             
@@ -416,6 +430,28 @@ class CyberLegalAPI:
                     error=result.get("error")
                 )
                 
+                # Log detailed response being sent to client
+                logger.info("=" * 80)
+                logger.info("📤 SENDING ANALYZE_PDF RESPONSE TO CLIENT")
+                logger.info("=" * 80)
+                logger.info(f"👤 User ID: {request.userId}")
+                logger.info(f"📄 Filename: {request.filename}")
+                logger.info(f"📅 Timestamp: {response.timestamp}")
+                logger.info(f"⏱️ Processing time: {response.processing_time:.2f}s")
+                logger.info(f"🎭 Actors: {response.actors}")
+                logger.info(f"🔑 Key details: {response.key_details}")
+                logger.info(f"📝 Summary: {response.summary}")
+                if response.extracted_text:
+                    logger.info(f"📄 Extracted text length: {len(response.extracted_text)} characters")
+                    logger.info(f"📝 Extracted text preview (first 200 chars): {response.extracted_text[:200]}...")
+                    logger.info(f"📝 Extracted text preview (last 200 chars): ...{response.extracted_text[-200:]}")
+                else:
+                    logger.info("📄 Extracted text: None")
+                if response.error:
+                    logger.warning(f"⚠️ Error in response: {response.error}")
+                logger.info(f"🔢 Full response object: {response.model_dump()}")
+                logger.info("=" * 80)
+                
                 logger.info(f"✅ PDF analysis completed in {processing_time:.2f}s")
                 return response
                 
@@ -539,6 +575,23 @@ class CyberLegalAPI:
                 error=None if result['success'] else result.get('message')
             )
             
+            # Log detailed response being sent to client
+            logger.info("=" * 80)
+            logger.info("📤 SENDING DOC_CREATOR RESPONSE TO CLIENT")
+            logger.info("=" * 80)
+            logger.info(f"👤 User ID: {request.userId}")
+            logger.info(f"📅 Timestamp: {response.timestamp}")
+            logger.info(f"⏱️ Processing time: {response.processing_time:.2f}s")
+            logger.info(f"💬 Response message: {response.response}")
+            if response.modifiedDocument:
+                logger.info(f"📄 Modified document size: {len(response.modifiedDocument)} bytes")
+                logger.info(f"📝 Document preview (first 200 chars): {response.modifiedDocument[:200]}...")
+                logger.info(f"📝 Document preview (last 200 chars): ...{response.modifiedDocument[-200:]}")
+            else:
+                logger.info("📄 Modified document: None")
+            if response.error:
+                logger.warning(f"⚠️ Error in response: {response.error}")
+            logger.info(f"🔢 Full response object: {response.model_dump()}")
             logger.info("=" * 80)
             logger.info("✅ DOCUMENT EDITING COMPLETED SUCCESSFULLY")
             logger.info("=" * 80)

agents/chat_agent.py

@@ -24,7 +24,8 @@ class CyberLegalAgent:
         self.tools_facade = tools_facade
         self.llm = llm
         self.performance_monitor = PerformanceMonitor()
-        self.llm_with_tools = self.llm.bind_tools(self.tools_facade)
+        # Constrain to one tool call at a time
+        self.llm_with_tools = self.llm.bind_tools(self.tools_facade, parallel_tool_calls=False)
         self.workflow = self._build_workflow()
     
     def _build_workflow(self) -> StateGraph: