Update main.py

main.py

@@ -66,10 +66,10 @@ def generate_search(query: str, stream: bool = True) -> str:
         if value.startswith("data: "):  
             try:
                 json_modified_value = json.loads(value[6:])  # Remove 'data: ' prefix
-                content = json_modified_value.get("choices", [{}])[0].get("delta", {}).get("content", "")
+                content = json_modified_value
                 # Include everything, even if it's just whitespace
                 if stream:
-                    yield f"data: {json.dumps({'response': content})}\n\n"
+                    yield f"data: {content}\n"
                 streaming_text += content
             except json.JSONDecodeError:
                 continue  # Skip lines that are not valid JSON