Update api.py

api.py

@@ -52,7 +52,7 @@ class AgentProcessor:
         # Update API base to use environment variable with fallback
         self.api_base = os.getenv(
             "API_BASE_URL", 
-            "http://severian.a.pinggy.link/v1"
+            "https://ai-engine.yamamotoqa.com/v1"
         )
         self.formatter = ResponseFormatter()
         self.client = httpx.AsyncClient(timeout=60.0)
@@ -120,6 +120,7 @@ class AgentProcessor:
             citations = []
             metadata = {}
             tool_outputs = []
+            buffer = ""
             
             try:
                 async with self.client.stream(
@@ -134,12 +135,12 @@ class AgentProcessor:
                         f"Headers: {dict(response.headers)}"
                     )
                     
-                    buffer = ""
                     async for line in response.aiter_lines():
                         if not line.strip():
                             continue
                         
                         self.logger.debug(f"Raw SSE line: {line}")
+                        buffer += line + "\n"
                         
                         if "data:" in line:
                             try:
@@ -205,30 +206,29 @@ class AgentProcessor:
                                     e, 
                                     {"line": line, "event": "parse_data"}
                                 )
-                    
-                    buffer += line + "\n"
-                    
-                    if line.startswith("data:") or buffer.strip().endswith("}"):
-                        try:
-                            processed_response = parser.parse_sse_event(buffer)
-                            if processed_response and isinstance(processed_response, dict):
-                                cleaned_response = self.clean_response(processed_response)
-                                if cleaned_response:
-                                    xml_content = cleaned_response.get("content", "")
-                                    yield f"data: {xml_content}\n\n"
-                        except Exception as parse_error:
-                            await self.log_error(
-                                parse_error,
-                                {"buffer": buffer, "event": "process_buffer"}
-                            )
-                            error_xml = (
-                                f"<agent_response>"
-                                f"<error>{str(parse_error)}</error>"
-                                f"</agent_response>"
-                            )
-                            yield f"data: {error_xml}\n\n"
-                        finally:
-                            buffer = ""
+                        
+                        # Process buffer when we have a complete message
+                        if buffer.strip().endswith("}"):
+                            try:
+                                processed_response = parser.parse_sse_event(buffer)
+                                if processed_response and isinstance(processed_response, dict):
+                                    cleaned_response = self.clean_response(processed_response)
+                                    if cleaned_response:
+                                        xml_content = cleaned_response.get("content", "")
+                                        yield f"data: {xml_content}\n\n"
+                            except Exception as parse_error:
+                                await self.log_error(
+                                    parse_error,
+                                    {"buffer": buffer, "event": "process_buffer"}
+                                )
+                                error_xml = (
+                                    f"<agent_response>"
+                                    f"<error>{str(parse_error)}</error>"
+                                    f"</agent_response>"
+                                )
+                                yield f"data: {error_xml}\n\n"
+                            finally:
+                                buffer = ""
                             
             except Exception as e:
                 self.logger.error(f"Stream processing error: {str(e)}")