Update gen.py

gen.py

@@ -7,7 +7,7 @@ from fastapi import APIRouter, Request, HTTPException, Header
 from fastapi.responses import Response, JSONResponse, StreamingResponse
 import re
 from typing import Optional
-
+import json
 from helper.assets import (
     save_base64_image,
     cleanup_image,
@@ -771,7 +771,7 @@ async def generate_text(
                 async for chunk, model_used in stream_primary(client, url, body, headers):
         
                     if not sent_metadata:
-                        actual_model = model_used
+                        actual_model = MODEL_MAP[model_used]
                         meta = {
                             "router_metadata": {
                                 "model_name": MODEL_MAP.get(actual_model, actual_model)