Niansuh commited on
Commit
aede964
·
verified ·
1 Parent(s): c8a657a

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +2 -3
api/utils.py CHANGED
@@ -9,7 +9,6 @@ from typing import Any, Dict, Optional
9
  import httpx
10
  from fastapi import HTTPException
11
  from api.config import (
12
- MODEL_MAPPING,
13
  get_headers_api_chat,
14
  get_headers_chat,
15
  BASE_URL,
@@ -99,7 +98,7 @@ async def process_streaming_response(request: ChatRequest):
99
  "previewToken": None,
100
  "trendingAgentMode": trending_agent_mode,
101
  "userId": None,
102
- "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
103
  "userSystemPrompt": None,
104
  "validated": "69783381-2ce4-4dbd-ac78-35e9063feabc",
105
  "visitFromDelta": False,
@@ -168,7 +167,7 @@ async def process_non_streaming_response(request: ChatRequest):
168
  "previewToken": None,
169
  "trendingAgentMode": trending_agent_mode,
170
  "userId": None,
171
- "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
172
  "userSystemPrompt": None,
173
  "validated": "69783381-2ce4-4dbd-ac78-35e9063feabc",
174
  "visitFromDelta": False,
 
9
  import httpx
10
  from fastapi import HTTPException
11
  from api.config import (
 
12
  get_headers_api_chat,
13
  get_headers_chat,
14
  BASE_URL,
 
98
  "previewToken": None,
99
  "trendingAgentMode": trending_agent_mode,
100
  "userId": None,
101
+ "userSelectedModel": request.model,
102
  "userSystemPrompt": None,
103
  "validated": "69783381-2ce4-4dbd-ac78-35e9063feabc",
104
  "visitFromDelta": False,
 
167
  "previewToken": None,
168
  "trendingAgentMode": trending_agent_mode,
169
  "userId": None,
170
+ "userSelectedModel": request.model,
171
  "userSystemPrompt": None,
172
  "validated": "69783381-2ce4-4dbd-ac78-35e9063feabc",
173
  "visitFromDelta": False,