pdrMottaS commited on
Commit
f2b24dc
·
1 Parent(s): 6b17147

add llm endpoint

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -48,6 +48,6 @@ async def llm(prompt_data: Prompt):
48
  global query_engine
49
  response = query_engine.query(prompt_data.prompt)
50
  print(response)
51
- return PromptResponse(prompt=prompt_data.prompt,response= response)
52
 
53
  uvicorn.run(app,host='0.0.0.0',port=7860)
 
48
  global query_engine
49
  response = query_engine.query(prompt_data.prompt)
50
  print(response)
51
+ return PromptResponse(prompt=prompt_data.prompt,response= response.response)
52
 
53
  uvicorn.run(app,host='0.0.0.0',port=7860)