Spaces:
Sleeping
Sleeping
Added explicit timeout
Browse files- llm/deepinfra_api.py +1 -1
- main.py +4 -4
llm/deepinfra_api.py
CHANGED
|
@@ -150,6 +150,6 @@ class DeepInfraApi(LlmApi):
|
|
| 150 |
"""
|
| 151 |
async with httpx.AsyncClient() as client:
|
| 152 |
request = await self.create_request(prompt, system_prompt)
|
| 153 |
-
response = await client.post(f"{self.params.url}/v1/openai/chat/completions", headers=super().create_headers(), json=request, timeout=httpx.Timeout(connect=5.0, read=60.0))
|
| 154 |
if response.status_code == 200:
|
| 155 |
return response.json()["choices"][0]["message"]["content"]
|
|
|
|
| 150 |
"""
|
| 151 |
async with httpx.AsyncClient() as client:
|
| 152 |
request = await self.create_request(prompt, system_prompt)
|
| 153 |
+
response = await client.post(f"{self.params.url}/v1/openai/chat/completions", headers=super().create_headers(), json=request, timeout=httpx.Timeout(connect=5.0, read=60.0, write=180, pool=10))
|
| 154 |
if response.status_code == 200:
|
| 155 |
return response.json()["choices"][0]["message"]["content"]
|
main.py
CHANGED
|
@@ -49,15 +49,15 @@ async def extracttable_route(request: TextRequest):
|
|
| 49 |
|
| 50 |
result = {"response": None, "error": None, "raw": response}
|
| 51 |
|
| 52 |
-
if "JSON
|
| 53 |
-
result["error"] = "Строка не содержит 'JSON
|
| 54 |
return result
|
| 55 |
|
| 56 |
-
prefix, json_str = response.split("JSON
|
| 57 |
json_str = json_str.strip()
|
| 58 |
|
| 59 |
if not json_str:
|
| 60 |
-
result["error"] = "После 'JSON
|
| 61 |
return result
|
| 62 |
|
| 63 |
try:
|
|
|
|
| 49 |
|
| 50 |
result = {"response": None, "error": None, "raw": response}
|
| 51 |
|
| 52 |
+
if "JSON" not in response:
|
| 53 |
+
result["error"] = "Строка не содержит 'JSON'"
|
| 54 |
return result
|
| 55 |
|
| 56 |
+
prefix, json_str = response.split("JSON", 1)
|
| 57 |
json_str = json_str.strip()
|
| 58 |
|
| 59 |
if not json_str:
|
| 60 |
+
result["error"] = "После 'JSON' отсутствует JSON"
|
| 61 |
return result
|
| 62 |
|
| 63 |
try:
|