API: model name updates and str response for health check

This commit is contained in:
Adam Wilson
2025-05-20 10:13:14 -06:00
parent 8cb01137ff
commit 1e7890b007
+4 -5
View File
@@ -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'