rehaan commited on
Commit
58fd291
·
1 Parent(s): 1fa7eee

Add type ignore comments for chatbot and RAG document retrieval

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. nivra_agent.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
89
  # -------------------------------
90
  # Chat UI
91
  # -------------------------------
92
- chatbot = gr.Chatbot(show_label=False, type="messages")
93
  txt = gr.Textbox(
94
  placeholder="Describe your symptoms (e.g. fever, headache, rash)..."
95
  )
 
89
  # -------------------------------
90
  # Chat UI
91
  # -------------------------------
92
+ chatbot = gr.Chatbot(show_label=False, type="messages") # type: ignore
93
  txt = gr.Textbox(
94
  placeholder="Describe your symptoms (e.g. fever, headache, rash)..."
95
  )
nivra_agent.py CHANGED
@@ -247,7 +247,7 @@ def nivra_chat(user_input, chat_history=None):
247
  tool_results.append(analyze_symptom_text.invoke(user_input))
248
  rag = get_rag()
249
  if rag:
250
- tool_results.append(rag.getRelevantDocs(user_input))
251
  else:
252
  tool_results.append("RAG UNAVAILABLE")
253
  except Exception:
 
247
  tool_results.append(analyze_symptom_text.invoke(user_input))
248
  rag = get_rag()
249
  if rag:
250
+ tool_results.append(rag.getRelevantDocs(user_input)) # type: ignore
251
  else:
252
  tool_results.append("RAG UNAVAILABLE")
253
  except Exception: