apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-06 22:32:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add history of messages

Browse source
This commit is contained in:
cotran 2024-10-10 15:37:08 -07:00
parent 1dba11bfd9
commit 60cadf3e8a
3 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model_server/app/main.py
View file

@ -179,7 +179,7 @@ async def hallucination(req: HallucinationRequest, res: Response):
"""
Take input as text and return the prediction of hallucination for each parameter
"""
logger.info(f"hallucination request: {req}")
if req.model != zero_shot_model["model_name"]:
raise HTTPException(status_code=400, detail="unknown model: " + req.model)