diff --git a/backend/rag_components/chain_links/rag_basic.py b/backend/rag_components/chain_links/rag_basic.py index 569a9fc..96de7d5 100644 --- 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