Merge pull request #15 from artefactory-skaff/fix/pydantic-error

Fix pydantic error in rag_basic
artefactory-skaff · May 16, 2024 · 88b0efe · 88b0efe
2 parents 6a11f13 + ed744d8
commit 88b0efe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/rag_components/chain_links/rag_basic.py b/backend/rag_components/chain_links/rag_basic.py
@@ -28,7 +28,7 @@ def rag_basic(llm, retriever: BaseRetriever) -> DocumentedRunnable:
     chain = (
         {
             "relevant_documents": fetch_docs_chain(retriever),
-            "question": RunnablePassthrough(Question),
+            "question": RunnablePassthrough(input_type=Question),
         }
         | ChatPromptTemplate.from_template(prompt)
         | llm