From 1e7890b0070391b3960d02d2dfceb34236747606 Mon Sep 17 00:00:00 2001
From: Adam Wilson <lightbrok3r@gmail.com>
Date: Tue, 20 May 2025 10:13:14 -0600
Subject: [PATCH] API: model name updates and str response for health check

---
 src/api/http_api.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/api/http_api.py b/src/api/http_api.py
index a10c2251a..6e0235d9e 100644
--- a/src/api/http_api.py
+++ b/src/api/http_api.py
@@ -9,14 +9,13 @@ app = FastAPI(
     description='HTTP API for interacting with Phi-3 Mini 4K language model'
 )
 
-class Prompt(BaseModel):
+class LanguageModelPrompt(BaseModel):
     prompt: str
 
-class Response(BaseModel):
+class LanguageModelResponse(BaseModel):
     response: str
 
 
-@app.get('/', response_model=Response)
+@app.get('/', response_model=str)
 async def health_check():
-    return ({ 'response': 'success' })
-
+    return 'success'