Skip to content

Commit 1a33175

Browse files
authored
Merge pull request #156 from vinaybagade/main
2 parents d498fcf + 02e7944 commit 1a33175

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

notebooks/agentic_rag_with_nemo_retriever_nims.ipynb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -571,11 +571,12 @@
571571
" \"\"\"\n",
572572
" print(\"---RETRIEVE---\")\n",
573573
" sub_questions = state[\"sub_questions\"]\n",
574+
" question = state[\"question\"]\n",
574575
"\n",
575576
" # Retrieval\n",
576577
" documents = []\n",
577-
" for question in sub_questions:\n",
578-
" docs = hybrid_retriever.get_relevant_documents(question)\n",
578+
" for sub_question in sub_questions:\n",
579+
" docs = hybrid_retriever.get_relevant_documents(sub_question)\n",
579580
" documents.extend(docs)\n",
580581
" return {\"documents\": documents, \"question\": question}\n",
581582
"\n",

0 commit comments

Comments
 (0)