dav74 commited on
Commit
77a97bc
·
verified ·
1 Parent(s): 8f01a0e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from dotenv import load_dotenv
3
  from langchain_core.output_parsers import StrOutputParser
4
- from langchain_groq import ChatGroq
5
  from langchain_core.prompts import ChatPromptTemplate
6
  from typing import List
7
  from langchain_huggingface import HuggingFaceEmbeddings
@@ -33,12 +33,10 @@ class Request(BaseModel):
33
  id : str
34
 
35
  load_dotenv()
36
- os.environ["GROQ_API_KEY"] = os.getenv('GROQ_API_KEY')
37
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
38
  persist_directory = 'db'
39
  embedding = HuggingFaceEmbeddings(model_name="OrdalieTech/Solon-embeddings-large-0.1")
40
- #embedding = HuggingFaceEmbeddings(model_name="BAAI/bge-multilingual-gemma2")
41
- llm = ChatGroq(model="llama-3.3-70b-versatile", temperature=0.5)
42
  memory = MemorySaver()
43
 
44
  if os.path.exists(persist_directory) :
 
1
  import os
2
  from dotenv import load_dotenv
3
  from langchain_core.output_parsers import StrOutputParser
4
+ from langchain_google_genai import ChatGoogleGenerativeAI
5
  from langchain_core.prompts import ChatPromptTemplate
6
  from typing import List
7
  from langchain_huggingface import HuggingFaceEmbeddings
 
33
  id : str
34
 
35
  load_dotenv()
 
36
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
37
  persist_directory = 'db'
38
  embedding = HuggingFaceEmbeddings(model_name="OrdalieTech/Solon-embeddings-large-0.1")
39
+ llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0.5)
 
40
  memory = MemorySaver()
41
 
42
  if os.path.exists(persist_directory) :