kevinconka commited on
Commit
8b7033c
·
1 Parent(s): 84f5caf

Changed to ChatOpenAI

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import urllib.request
3
  from langchain.chains import RetrievalQA
4
  from langchain_community.document_loaders import UnstructuredHTMLLoader
5
  from langchain_openai import OpenAIEmbeddings
6
- from langchain_openai.llms import OpenAI
7
  from langchain.text_splitter import CharacterTextSplitter
8
  from langchain_community.vectorstores import Chroma
9
 
@@ -30,7 +30,7 @@ db = Chroma.from_documents(texts, embeddings)
30
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
31
  # create a chain to answer questions
32
  qa = RetrievalQA.from_chain_type(
33
- llm=OpenAI(model="gpt-3.5-turbo-0125"),
34
  chain_type="stuff",
35
  retriever=retriever,
36
  return_source_documents=True,
 
3
  from langchain.chains import RetrievalQA
4
  from langchain_community.document_loaders import UnstructuredHTMLLoader
5
  from langchain_openai import OpenAIEmbeddings
6
+ from langchain_openai import ChatOpenAI
7
  from langchain.text_splitter import CharacterTextSplitter
8
  from langchain_community.vectorstores import Chroma
9
 
 
30
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
31
  # create a chain to answer questions
32
  qa = RetrievalQA.from_chain_type(
33
+ llm=ChatOpenAI(),
34
  chain_type="stuff",
35
  retriever=retriever,
36
  return_source_documents=True,