Niansuh commited on
Commit
3e0f1a7
·
verified ·
1 Parent(s): 5f9e187

Update api/utils.py

Browse files
Files changed (1) hide show
  1. 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
- "validated": validate.getHid(),
 
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
- "validated": validate.getHid(),
 
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: