standinothegr8 commited on
Commit
17ee6fb
·
verified ·
1 Parent(s): d09b89a

Upload server.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. server.py +1 -1
server.py CHANGED
@@ -70,7 +70,7 @@ def ask(body: Ask, request: Request):
70
  from rag import answer
71
  t = time.time()
72
  res = answer(body.question.strip(), STATE["ret"], STATE["gen"], k=body.k, per_speech=2, lang=None if body.lang == "auto" else body.lang)
73
- return {"question": res["question"], "answer": res["answer"], "model": res["model"], "retriever": res["retriever"], "lang": res["lang"],
74
  "passages": [{"n": p["n"], "citation": p["citation"], "text": p["text"], "speech_id": p["speech_id"], "score": p["score"]} for p in res["passages"]],
75
  "seconds": round(time.time() - t, 2)}
76
 
 
70
  from rag import answer
71
  t = time.time()
72
  res = answer(body.question.strip(), STATE["ret"], STATE["gen"], k=body.k, per_speech=2, lang=None if body.lang == "auto" else body.lang)
73
+ return {"question": res["question"], "answer": res["answer"], "model": res["model"], "retriever": res["retriever"], "lang": res["lang"], "query_my": res.get("query_my"),
74
  "passages": [{"n": p["n"], "citation": p["citation"], "text": p["text"], "speech_id": p["speech_id"], "score": p["score"]} for p in res["passages"]],
75
  "seconds": round(time.time() - t, 2)}
76