Spaces:
Sleeping
Sleeping
Commit
·
29b2c47
1
Parent(s):
787d79c
feat: add an heathcheck endpoint
Browse files
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)
|