Niansuh commited on
Commit
648f616
·
verified ·
1 Parent(s): 3a19797

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +2 -3
api/utils.py CHANGED
@@ -11,7 +11,6 @@ from api import validate
11
  from api.config import (
12
  APP_SECRET,
13
  BASE_URL,
14
- MODEL_MAPPING,
15
  AGENT_MODE,
16
  TRENDING_AGENT_MODE,
17
  MODEL_PREFIXES,
@@ -94,7 +93,7 @@ async def process_streaming_response(request: ChatRequest):
94
  "clickedForceWebSearch": False,
95
  "visitFromDelta": False,
96
  "mobileClient": False,
97
- "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
98
  "validated": hid
99
  }
100
 
@@ -159,7 +158,7 @@ async def process_non_streaming_response(request: ChatRequest):
159
  "clickedForceWebSearch": False,
160
  "visitFromDelta": False,
161
  "mobileClient": False,
162
- "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
163
  "validated": hid
164
  }
165
  full_response = ""
 
11
  from api.config import (
12
  APP_SECRET,
13
  BASE_URL,
 
14
  AGENT_MODE,
15
  TRENDING_AGENT_MODE,
16
  MODEL_PREFIXES,
 
93
  "clickedForceWebSearch": False,
94
  "visitFromDelta": False,
95
  "mobileClient": False,
96
+ "userSelectedModel": request.model, # Use the requested model directly
97
  "validated": hid
98
  }
99
 
 
158
  "clickedForceWebSearch": False,
159
  "visitFromDelta": False,
160
  "mobileClient": False,
161
+ "userSelectedModel": request.model, # Use the requested model directly
162
  "validated": hid
163
  }
164
  full_response = ""