Quincy Hsieh commited on
Commit
731ee37
·
1 Parent(s): 6ee8606

Handle error

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +8 -2
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🤖
4
  colorFrom: blue
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 4.44.0
8
  python_version: '3.11'
9
  app_file: app.py
10
  pinned: false
 
4
  colorFrom: blue
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 4.44.1
8
  python_version: '3.11'
9
  app_file: app.py
10
  pinned: false
app.py CHANGED
@@ -188,8 +188,11 @@ def generate_embeddings(texts: list[str]) -> list[list[float]]:
188
  except http_requests.exceptions.HTTPError as e:
189
  logger.error(f"Embedding API call failed: {e} — {resp.text}")
190
  raise HTTPException(status_code=503, detail=f"Embedding service unavailable: {str(e)}")
 
 
 
191
  except (KeyError, IndexError) as e:
192
- logger.error(f"Unexpected embedding response format: {e}")
193
  raise HTTPException(status_code=502, detail="Unexpected response from embedding service")
194
 
195
 
@@ -279,8 +282,11 @@ def call_llm(prompt: str) -> str:
279
  except http_requests.exceptions.HTTPError as e:
280
  logger.error(f"LLM API call failed: {e} — {resp.text}")
281
  raise HTTPException(status_code=503, detail=f"LLM service unavailable: {str(e)}")
 
 
 
282
  except (KeyError, IndexError) as e:
283
- logger.error(f"Unexpected LLM response format: {e}")
284
  raise HTTPException(status_code=502, detail="Unexpected response from LLM service")
285
 
286
 
 
188
  except http_requests.exceptions.HTTPError as e:
189
  logger.error(f"Embedding API call failed: {e} — {resp.text}")
190
  raise HTTPException(status_code=503, detail=f"Embedding service unavailable: {str(e)}")
191
+ except (http_requests.exceptions.JSONDecodeError, ValueError) as e:
192
+ logger.error(f"Embedding API returned non-JSON response (status {resp.status_code}): {repr(resp.text)}")
193
+ raise HTTPException(status_code=502, detail="Embedding service returned an invalid response")
194
  except (KeyError, IndexError) as e:
195
+ logger.error(f"Unexpected embedding response format: {e} — body: {resp.text}")
196
  raise HTTPException(status_code=502, detail="Unexpected response from embedding service")
197
 
198
 
 
282
  except http_requests.exceptions.HTTPError as e:
283
  logger.error(f"LLM API call failed: {e} — {resp.text}")
284
  raise HTTPException(status_code=503, detail=f"LLM service unavailable: {str(e)}")
285
+ except (http_requests.exceptions.JSONDecodeError, ValueError) as e:
286
+ logger.error(f"LLM API returned non-JSON response (status {resp.status_code}): {repr(resp.text)}")
287
+ raise HTTPException(status_code=502, detail="LLM service returned an invalid response")
288
  except (KeyError, IndexError) as e:
289
+ logger.error(f"Unexpected LLM response format: {e} — body: {resp.text}")
290
  raise HTTPException(status_code=502, detail="Unexpected response from LLM service")
291
 
292
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  fastapi==0.115.0
2
  uvicorn==0.30.0
3
- gradio==4.44.0
4
  chromadb==0.5.0
5
  huggingface-hub==0.25.0
6
  langchain==0.3.0
 
1
  fastapi==0.115.0
2
  uvicorn==0.30.0
3
+ gradio==4.44.1
4
  chromadb==0.5.0
5
  huggingface-hub==0.25.0
6
  langchain==0.3.0