Ephraimmm commited on
Commit
b8243e2
·
verified ·
1 Parent(s): 3ab0db4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -13,11 +13,11 @@ from typing import List, Dict, Optional
13
 
14
  # LangChain imports
15
  from langchain.text_splitter import RecursiveCharacterTextSplitter
16
- from langchain.vectorstores import FAISS
17
- from langchain.embeddings import OpenAIEmbeddings
 
18
  from langchain.docstore.document import Document
19
  from langchain.chains import RetrievalQA
20
- from langchain.llms import OpenAI
21
  from langchain.prompts import PromptTemplate
22
  from langchain.memory import ConversationBufferMemory
23
  from langchain.chains import ConversationalRetrievalChain
@@ -67,8 +67,8 @@ class EnhancedGPTDriveIntegration:
67
 
68
  # Initialize OpenAI and LangChain components
69
  openai.api_key = os.getenv('OPENAI_API_KEY')
70
- self.embeddings = OpenAIEmbeddings(openai_api_key=os.getenv('OPENAI_API_KEY'))
71
- self.llm = OpenAI(temperature=0.7, openai_api_key=os.getenv('OPENAI_API_KEY'))
72
 
73
  # Text splitter for better chunking
74
  self.text_splitter = RecursiveCharacterTextSplitter(
 
13
 
14
  # LangChain imports
15
  from langchain.text_splitter import RecursiveCharacterTextSplitter
16
+ from langchain_community.vectorstores import FAISS
17
+ from langchain_openai import OpenAIEmbeddings, ChatOpenAI
18
+ from langchain_community.vectorstores import FAISS
19
  from langchain.docstore.document import Document
20
  from langchain.chains import RetrievalQA
 
21
  from langchain.prompts import PromptTemplate
22
  from langchain.memory import ConversationBufferMemory
23
  from langchain.chains import ConversationalRetrievalChain
 
67
 
68
  # Initialize OpenAI and LangChain components
69
  openai.api_key = os.getenv('OPENAI_API_KEY')
70
+ self.embeddings = OpenAIEmbeddings()
71
+ self.llm = ChatOpenAI(temperature=0.7, model="gpt-3.5-turbo")
72
 
73
  # Text splitter for better chunking
74
  self.text_splitter = RecursiveCharacterTextSplitter(