Merge pull request #79 from pattern-tech/refactor/ai-v2

Adjust timeout values and update media type for streaming messages

Author: yasinfakhar

src/agent/services/agent_service.py

@@ -146,9 +146,9 @@ def __init__(self, tools, memory=None, streaming: bool = True):
         self.streaming_handler = None
 
         # Default timeout values that can be adjusted if needed
-        self.token_timeout = 0.5  # Increased from 0.01
-        self.buffer_timeout = 0.1  # Increased from 0.005
-        self.poll_interval = 0.1  # Increased from 0.01
+        self.token_timeout = 0.2
+        self.buffer_timeout = 0.05
+        self.poll_interval = 0.1
 
         # Set up the streaming callback if streaming is enabled.
         if streaming:

src/agentflow/providers/moralis_tools.py

@@ -252,3 +252,4 @@ def get_token_approvals(wallet_address: str, chain: str, output_include: list[st
 
     return {"cursor": api_result.get("cursor", None),
             "results": final_results}
+

src/conversation/routers/playground_conversation_router.py

@@ -426,7 +426,13 @@ async def send_message(
                                                   project_id,
                                                   input.message_type,
                                                   input.stream),
-                media_type="text/plain"
+                media_type="text/event-stream",
+                headers={
+                    "Content-Type": "text/event-stream",
+                    "Cache-Control": "no-cache",
+                    "Connection": "keep-alive",
+                    "X-Accel-Buffering": "no",
+                },
             )
         else:
             response = None