Spaces:
Sleeping
Sleeping
pdrMottaS
commited on
Commit
·
4c46c3b
1
Parent(s):
f789d71
add llm endpoint
Browse files- main.py +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.
|
50 |
-
return JSONResponse({"promt":prompt_data.
|
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 |
-
|
|
|
2 |
|
3 |
|
4 |
class Prompt(BaseModel):
|
5 |
+
prompt:str
|