Merge pull request #156 from vinaybagade/main

NVIDIA · Jul 31, 2024 · 1a33175 · 1a33175
2 parents d498fcf + 02e7944
commit 1a33175
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb b/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb
@@ -571,11 +571,12 @@
     "    \"\"\"\n",
     "    print(\"---RETRIEVE---\")\n",
     "    sub_questions = state[\"sub_questions\"]\n",
+    "    question = state[\"question\"]\n",
     "\n",
     "    # Retrieval\n",
     "    documents = []\n",
-    "    for question in sub_questions:\n",
-    "        docs = hybrid_retriever.get_relevant_documents(question)\n",
+    "    for sub_question in sub_questions:\n",
+    "        docs = hybrid_retriever.get_relevant_documents(sub_question)\n",
     "        documents.extend(docs)\n",
     "    return {\"documents\": documents, \"question\": question}\n",
     "\n",