pdrMottaS commited on
Commit
4c46c3b
·
1 Parent(s): f789d71

add llm endpoint

Browse files
Files changed (2) hide show
  1. main.py +2 -2
  2. models/prompt.py +1 -1
main.py CHANGED
@@ -46,7 +46,7 @@ async def query_database(query_data: SQL):
46
  @app.post("/llm")
47
  async def llm(prompt_data: Prompt):
48
  global query_engine
49
- response = query_engine.query(prompt_data.promt)
50
- return JSONResponse({"promt":prompt_data.promt,"response":response})
51
 
52
  uvicorn.run(app,host='0.0.0.0',port=7860)
 
46
  @app.post("/llm")
47
  async def llm(prompt_data: Prompt):
48
  global query_engine
49
+ response = query_engine.query(prompt_data.prompt)
50
+ return JSONResponse({"promt":prompt_data.prompt,"response":response})
51
 
52
  uvicorn.run(app,host='0.0.0.0',port=7860)
models/prompt.py CHANGED
@@ -2,4 +2,4 @@ from pydantic import BaseModel
2
 
3
 
4
  class Prompt(BaseModel):
5
- promt:str
 
2
 
3
 
4
  class Prompt(BaseModel):
5
+ prompt:str