shamimjony1000 commited on
Commit
72b325b
·
verified ·
1 Parent(s): 139e4ac

Update llm_part.py

Browse files
Files changed (1) hide show
  1. llm_part.py +3 -0
llm_part.py CHANGED
@@ -6,7 +6,10 @@ from langchain_google_genai import ChatGoogleGenerativeAI
6
  from langchain_groq import ChatGroq
7
  from datetime import datetime
8
 
 
9
  # Initialize LLM models
 
 
10
  llm_1 = ChatGoogleGenerativeAI(model="gemini-pro", api_key=gemeni_key)
11
  llm_2 = ChatGroq(model="llama3-groq-70b-8192-tool-use-preview", groq_api_key=llama_key)
12
 
 
6
  from langchain_groq import ChatGroq
7
  from datetime import datetime
8
 
9
+
10
  # Initialize LLM models
11
+ gemeni_key=os.getenv("gemeni_key")
12
+ llama_key=os.getenv("llama_key")
13
  llm_1 = ChatGoogleGenerativeAI(model="gemini-pro", api_key=gemeni_key)
14
  llm_2 = ChatGroq(model="llama3-groq-70b-8192-tool-use-preview", groq_api_key=llama_key)
15