fcastrovilli commited on
Commit
29b2c47
·
1 Parent(s): 787d79c

feat: add an heathcheck endpoint

Browse files
Files changed (1) hide show
  1. main.py +8 -0
main.py CHANGED
@@ -189,5 +189,13 @@ async def compute_similarities(embeddings: Dict[str, List[List[float]]]):
189
  similarities = embedding_manager.compute_similarities(embeddings)
190
  return SimilarityResponse(similarities=similarities)
191
 
 
 
 
 
 
 
 
 
192
  if __name__ == "__main__":
193
  uvicorn.run(app, host="0.0.0.0", port=8000)
 
189
  similarities = embedding_manager.compute_similarities(embeddings)
190
  return SimilarityResponse(similarities=similarities)
191
 
192
+ @app.get("/health")
193
+ async def health_check():
194
+ """Basic healthcheck endpoint that returns the status of the service."""
195
+ return {
196
+ "status": "healthy",
197
+ "model_device": embedding_manager.device
198
+ }
199
+
200
  if __name__ == "__main__":
201
  uvicorn.run(app, host="0.0.0.0", port=8000)