Spaces:
Sleeping
Sleeping
File size: 574 Bytes
8c42033 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from langchain.chat_models import ChatOpenAI
from langchain.vectorstores.pinecone import Pinecone
import openai
from langchain.chains import RetrievalQAWithSourcesChain
import os
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
def get_chain(vectorstore: Pinecone):
openai.api_key = OPENAI_API_KEY
llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo")
qa_chain = RetrievalQAWithSourcesChain.from_chain_type(llm=llm, chain_type="stuff",
retriever=vectorstore.as_retriever())
return qa_chain
|