Mohamed-Maher commited on
Commit
2619341
·
verified ·
1 Parent(s): 83e0ace

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,9 +10,9 @@ class HadithChatApp:
10
  def __init__(self):
11
  self.QDRANT_URL = os.getenv('QDRANT_URL')
12
  self.QDRANT_API_KEY = os.getenv('QDRANT_API_KEY')
13
- self.OPEN_AI_TOKEN = os.getenv('OPEN_AI_TOKEN')
14
 
15
- self.collection_name = "Rag-with-Langchain-qdrant-Hadith"
16
 
17
  self.client = qdrant_client.QdrantClient(
18
  url=self.QDRANT_URL,
@@ -34,7 +34,7 @@ class HadithChatApp:
34
  embeddings=self.embeddings
35
  )
36
 
37
- self.chat = ChatOpenAI(openai_api_key=self.OPEN_AI_TOKEN, model='gpt-3.5-turbo', temperature=0.5)
38
 
39
  def clean_text(self, text):
40
  text = re.sub(r'<[^>]*>', '', text)
 
10
  def __init__(self):
11
  self.QDRANT_URL = os.getenv('QDRANT_URL')
12
  self.QDRANT_API_KEY = os.getenv('QDRANT_API_KEY')
13
+ self.GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
14
 
15
+ self.collection_name = "Cluster0"
16
 
17
  self.client = qdrant_client.QdrantClient(
18
  url=self.QDRANT_URL,
 
34
  embeddings=self.embeddings
35
  )
36
 
37
+ self.chat = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest",google_api_key=self.GOOGLE_API_KEY)
38
 
39
  def clean_text(self, text):
40
  text = re.sub(r'<[^>]*>', '', text)