Update app.py

app.py

@@ -47,7 +47,6 @@ SUPPORTED_MODELS = [
     "gpt-4o-mini",
     "claude-3-haiku-20240307",
     "meta-llama/Llama-3.3-70B-Instruct-Turbo",
-    "mistralai/Mistral-Small-24B-Instruct-2501",
 ]
 
 async def get_vqd():
@@ -90,13 +89,8 @@ async def duckduckgo_chat_stream(model, messages):
                     if line:
                         decoded_line = line.decode('utf-8')
                         if decoded_line.startswith("data: "):
-                            content = decoded_line[5:].strip()
-                            # Check if this is the final marker
-                            if content == "[DONE]":
-                                yield f"data: [DONE]\n\n"
-                                break
                             try:
-                                json_data = json.loads(content)
+                                json_data = json.loads(decoded_line[5:])
                                 message_content = json_data.get("message", "")
                                 if message_content:
                                     # Build OpenAI style streaming response
@@ -107,9 +101,11 @@ async def duckduckgo_chat_stream(model, messages):
                                         "model": model,
                                         "choices": [
                                             {
-                                                "delta": {"content": message_content},
+                                                "delta": {
+                                                    "content": message_content
+                                                },
                                                 "index": 0,
-                                                "finish_reason": None,
+                                                "finish_reason": None
                                             }
                                         ],
                                     }
@@ -259,4 +255,4 @@ async def http_exception_handler(request: Request, exc: HTTPException):
 
 @app.get("/")
 async def greet_json():
-    return {"Hello": "World!"}
+    return {"Hello": "World!"}