Update api/utils.py
Browse files- api/utils.py +14 -2
api/utils.py
CHANGED
@@ -90,6 +90,9 @@ async def process_streaming_response(request: ChatRequest):
|
|
90 |
|
91 |
headers_api_chat = headers.copy()
|
92 |
headers_api_chat['Referer'] = referer_url
|
|
|
|
|
|
|
93 |
|
94 |
if request.model == 'o1-preview':
|
95 |
delay_seconds = random.randint(1, 60)
|
@@ -116,10 +119,13 @@ async def process_streaming_response(request: ChatRequest):
|
|
116 |
"userId": None,
|
117 |
"userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
|
118 |
"userSystemPrompt": None,
|
119 |
-
|
|
|
120 |
"visitFromDelta": False,
|
121 |
}
|
122 |
|
|
|
|
|
123 |
async with httpx.AsyncClient() as client:
|
124 |
try:
|
125 |
async with client.stream(
|
@@ -168,6 +174,9 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
168 |
|
169 |
headers_api_chat = headers.copy()
|
170 |
headers_api_chat['Referer'] = referer_url
|
|
|
|
|
|
|
171 |
|
172 |
if request.model == 'o1-preview':
|
173 |
delay_seconds = random.randint(20, 60)
|
@@ -194,10 +203,13 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
194 |
"userId": None,
|
195 |
"userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
|
196 |
"userSystemPrompt": None,
|
197 |
-
|
|
|
198 |
"visitFromDelta": False,
|
199 |
}
|
200 |
|
|
|
|
|
201 |
full_response = ""
|
202 |
async with httpx.AsyncClient() as client:
|
203 |
try:
|
|
|
90 |
|
91 |
headers_api_chat = headers.copy()
|
92 |
headers_api_chat['Referer'] = referer_url
|
93 |
+
headers_api_chat['Cookie'] = f'hid={validate.getHid()}'
|
94 |
+
|
95 |
+
logger.debug(f"Headers being sent: {headers_api_chat}")
|
96 |
|
97 |
if request.model == 'o1-preview':
|
98 |
delay_seconds = random.randint(1, 60)
|
|
|
119 |
"userId": None,
|
120 |
"userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
|
121 |
"userSystemPrompt": None,
|
122 |
+
# Remove 'validated' if not required
|
123 |
+
# "validated": validate.getHid(),
|
124 |
"visitFromDelta": False,
|
125 |
}
|
126 |
|
127 |
+
logger.debug(f"JSON payload being sent: {json.dumps(json_data)}")
|
128 |
+
|
129 |
async with httpx.AsyncClient() as client:
|
130 |
try:
|
131 |
async with client.stream(
|
|
|
174 |
|
175 |
headers_api_chat = headers.copy()
|
176 |
headers_api_chat['Referer'] = referer_url
|
177 |
+
headers_api_chat['Cookie'] = f'hid={validate.getHid()}'
|
178 |
+
|
179 |
+
logger.debug(f"Headers being sent: {headers_api_chat}")
|
180 |
|
181 |
if request.model == 'o1-preview':
|
182 |
delay_seconds = random.randint(20, 60)
|
|
|
203 |
"userId": None,
|
204 |
"userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
|
205 |
"userSystemPrompt": None,
|
206 |
+
# Remove 'validated' if not required
|
207 |
+
# "validated": validate.getHid(),
|
208 |
"visitFromDelta": False,
|
209 |
}
|
210 |
|
211 |
+
logger.debug(f"JSON payload being sent: {json.dumps(json_data)}")
|
212 |
+
|
213 |
full_response = ""
|
214 |
async with httpx.AsyncClient() as client:
|
215 |
try:
|