인프런 커뮤니티 질문&답변

달콤한 백로님의 프로필 이미지
달콤한 백로

작성한 질문수

프로젝트로 배우는 Python 챗봇 & RAG - LangChain, Gradio 활용

Chroma DB를 활용하여 PDF 파일에 대해 RAG 기반 질의응답 구현 (실습)

response = retrieval_chain.invoke 실행 에러 질문

작성

·

307

0

학Chroma DB를 활용하여 PDF 파일에 대해 RAG 기반 질의응답 구현 (실습)의 29:00 에 있는 코드를 똑같이 실행했는데, 다음과 같은 에러가 발생했습니다.

from langchain_openai import ChatOpenAI
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain.chains import create_retrieval_chain

model = ChatOpenAI(model ='gpt-3.5-turbo-0125', temperature=0, 
                   api_key=OPENAI_API_KEY)

document_chain = create_retrieval_chain(model, prompt)

retrieval_chain = create_retrieval_chain(retriever, document_chain)
response = retrieval_chain.invoke({"input": "what is the attention mechanism in transformers?"})

에러 메시지

Cell In[163], line 1 ----> 1 response = retrieval_chain.invoke({"input": "what is the attention mechanism in transformers?"}) 2 print(response)
ValueError: Invalid input type <class 'dict'>. Must be a PromptValue, str, or list of BaseMessages.
Output is truncated. View as a scrollable element or open in a text editor. Adjust cell output settings...

 

답변 1

0

판다스 스튜디오님의 프로필 이미지
판다스 스튜디오
지식공유자

안녕하세요.

document_chain을 구성하는 메소드를 강의 내용과 같이 create_stuff_documents_chain 함수를 적용해 주셔야 합니다. (보내주신 코드를 보면 create_retrieval_chain 함수가 적용되어 있습니다.)

document_chain = create_stuff_documents_chain(model, prompt)

감사합니다.

달콤한 백로님의 프로필 이미지
달콤한 백로

작성한 질문수

질문하기