Update gen.py

gen.py

@@ -702,12 +702,12 @@ async def generate_text(
                     if r.status_code >= 400:
                         print("[STREAM FALLBACK] Primary provider failed → switching to Groq fallback")
                         async for chunk in stream_fallback(client, body):
-                            yield chunk, fallback_model
+                            yield chunk
                         return
     
                     async for line in r.aiter_lines():
                         if not line:
-                            yield "\n", chosen_model
+                            yield "\n"
                             continue
                         if line.startswith("event: error"):
                             fallback()
@@ -720,12 +720,12 @@ async def generate_text(
                                 except:
                                     pass
     
-                        yield line + "\n", chosen_model
+                        yield line + "\n"
     
             except Exception as e:
                 print(f"[STREAM ERROR] {e}")
                 async for chunk in stream_fallback(client, body):
-                    yield chunk, fallback_model
+                    yield chunk
     
     
         async def stream_fallback(client, body):
@@ -754,7 +754,7 @@ async def generate_text(
     
                 if r.status_code >= 400:
                     err = (await r.aread()).decode("utf-8", errors="replace")
-                    yield f'data: {{"error": "Fallback provider failed: {err[:500]}"}}\n\n', fallback_model
+                    yield f'data: {{"error": "Fallback provider failed: {err[:500]}"}}\n\n'
                     return
     
                 async for line in r.aiter_lines():
@@ -763,23 +763,21 @@ async def generate_text(
                         continue
     
                     if not line.startswith("data:"):
-                        yield f"data: {line}\n\n", fallback_model
+                        yield f"data: {line}\n\n"
                     else:
-                        yield (line + "\n"), fallback_model
+                        yield line + "\n"
     
     
         async def event_generator():
-            actual_model = None
             sent_metadata = False
         
             async with httpx.AsyncClient(timeout=None) as client:
-                async for chunk, model_used in stream_primary(client, url, body, headers):
+                async for chunk in stream_primary(client, url, body, headers):
         
                     if not sent_metadata:
-                        actual_model = MODEL_MAP[model_used]
                         meta = {
                             "router_metadata": {
-                                "model_name": MODEL_MAP.get(actual_model, actual_model)
+                                "model_name": MODEL_MAP.get(chosen_model, chosen_model)
                             }
                         }
                         yield f"data: {json.dumps(meta)}\n\n"