diff --git a/docs/docs/tutorials/rag/langgraph_adaptive_rag.ipynb b/docs/docs/tutorials/rag/langgraph_adaptive_rag.ipynb index 00fe6e299..fa2bc124f 100644 --- a/docs/docs/tutorials/rag/langgraph_adaptive_rag.ipynb +++ b/docs/docs/tutorials/rag/langgraph_adaptive_rag.ipynb @@ -336,7 +336,8 @@ "rag_chain = prompt | llm | StrOutputParser()\n", "\n", "# Run\n", - "generation = rag_chain.invoke({\"context\": docs, \"question\": question})\n", + "docs_txt = format_docs(docs)\n", + "generation = rag_chain.invoke({\"context\": docs_txt, \"question\": question})\n", "print(generation)" ] }, @@ -625,7 +626,8 @@ " documents = state[\"documents\"]\n", "\n", " # RAG generation\n", - " generation = rag_chain.invoke({\"context\": documents, \"question\": question})\n", + " docs_txt = format_docs(documents)\n", + " generation = rag_chain.invoke({\"context\": docs_txt, \"question\": question})\n", " return {\"documents\": documents, \"question\": question, \"generation\": generation}\n", "\n", "\n",