From ee1484ddb0c28ecaed2477a2c78f786bb4badb15 Mon Sep 17 00:00:00 2001 From: Alexander Neumann Date: Wed, 20 Dec 2023 16:56:38 +0100 Subject: [PATCH] fix mypy issues, ignore langchain missing imports --- mypy.ini | 3 +++ team_red/frontends/qa_frontend.py | 17 ++++++++++------- 2 files changed, 13 insertions(+), 7 deletions(-) diff --git a/mypy.ini b/mypy.ini index a1c36cd..24bfd81 100644 --- a/mypy.ini +++ b/mypy.ini @@ -17,6 +17,9 @@ ignore_missing_imports = True [mypy-gradio.*] ignore_missing_imports = True +[mypy-gradio.langchain.*] +ignore_missing_imports = True + ;; gradio is not PEP 561 compliant (no py.typed) yet [mypy-team_red.frontends.*] disallow_any_unimported = False \ No newline at end of file diff --git a/team_red/frontends/qa_frontend.py b/team_red/frontends/qa_frontend.py index 176e42e..b3bcfda 100644 --- a/team_red/frontends/qa_frontend.py +++ b/team_red/frontends/qa_frontend.py @@ -18,17 +18,20 @@ def query(question: str, search_type: str, k_source: int, search_strategy: str) question=question, search_strategy=search_strategy, max_sources=k_source ) if search_type == "LLM": - res = TRANSPORTER.qa_query(q) - if res.status != 200: - msg = f"Query was unsuccessful: {res.error_msg} (Error Code {res.status})" + qa_res = TRANSPORTER.qa_query(q) + if qa_res.status != 200: + msg = ( + f"Query was unsuccessful: {qa_res.error_msg}" + f" (Error Code {qa_res.status})" + ) raise gr.Error(msg) - return res.answer - res = TRANSPORTER.db_query(q) - if not res: + return qa_res.answer + db_res = TRANSPORTER.db_query(q) + if not db_res: msg = f"Database query returned empty!" raise gr.Error(msg) output = "" - for doc in res: + for doc in db_res: output += f"{doc.content}\n" output += f"({doc.name} / {doc.page})\n----------\n\n" return output