Update api/utils.py

api/utils.py

@@ -15,6 +15,7 @@ from api.config import (
     BASE_URL,
     AGENT_MODE,
     TRENDING_AGENT_MODE,
+    MODEL_PREFIXES,
     MODEL_REFERERS
 )
 from api.models import ChatRequest
@@ -87,6 +88,7 @@ async def process_streaming_response(request: ChatRequest):
 
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
+    model_prefix = MODEL_PREFIXES.get(request.model, "")
 
     headers_api_chat = get_headers_api_chat(referer_url)
 
@@ -155,6 +157,7 @@ async def process_non_streaming_response(request: ChatRequest):
 
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
+    model_prefix = MODEL_PREFIXES.get(request.model, "")
 
     headers_api_chat = get_headers_api_chat(referer_url)
     headers_chat = get_headers_chat(referer_url, next_action=str(uuid.uuid4()), next_router_state_tree=json.dumps([""]))