Update agent_service.py

agent_service.py

@@ -52,6 +52,7 @@ class AgentService:
         conversation_history: List[Dict],
         mode: str = "sales",  # "sales" or "feedback"
         user_id: Optional[str] = None,
+        access_token: Optional[str] = None,  # NEW: For authenticated API calls
         max_iterations: int = 3
     ) -> Dict[str, Any]:
         """
@@ -62,6 +63,7 @@ class AgentService:
             conversation_history: Previous messages [{"role": "user", "content": ...}, ...]
             mode: "sales" or "feedback"
             user_id: User ID (for feedback mode to check purchase history)
+            access_token: JWT token for authenticated API calls
             max_iterations: Maximum tool call iterations to prevent infinite loops
         
         Returns:
@@ -74,6 +76,9 @@ class AgentService:
         print(f"\n🤖 Agent Mode: {mode}")
         print(f"👤 User Message: {user_message}")
         
+        # Store access_token for tool calls
+        self.current_access_token = access_token
+        
         # Select system prompt
         system_prompt = self._get_system_prompt(mode)
         
@@ -103,7 +108,8 @@ class AgentService:
             print(f"🔧 Tool Called: {tool_call['tool_name']}")
             tool_result = await self.tools_service.execute_tool(
                 tool_call['tool_name'],
-                tool_call['arguments']
+                tool_call['arguments'],
+                access_token=self.current_access_token  # Pass access_token
             )
             
             # Record tool call
@@ -327,7 +333,8 @@ Example:
         # Format results
         formatted = []
         for i, result in enumerate(results, 1):
-            payload = result.payload or {}
+            # Result is a dict with keys: id, score, payload
+            payload = result.get("payload", {})
             texts = payload.get("texts", [])
             text = texts[0] if texts else ""
             event_id = payload.get("id_use", "")