shuv25 commited on
Commit
fb2ea68
·
verified ·
1 Parent(s): 0ac9ae2

Update deep_agent.py

Browse files
Files changed (1) hide show
  1. deep_agent.py +11 -11
deep_agent.py CHANGED
@@ -12,20 +12,20 @@ import os
12
 
13
  load_dotenv()
14
 
15
- gemini_api_key = os.getenv("gemini_api_key")
16
- llm = ChatGoogleGenerativeAI(
17
- model="gemini-2.0-flash",
18
- google_api_key=gemini_api_key,
19
- temperature=0.3
20
- )
21
-
22
- # groq_api_key = os.getenv("GROQ_API_KEY")
23
- # llm = ChatGroq(
24
- # model="llama-3.3-70b-versatile", # or "llama-3.1-70b-versatile"
25
- # api_key=groq_api_key,
26
  # temperature=0.3
27
  # )
28
 
 
 
 
 
 
 
 
29
  store = InMemoryStore()
30
  memory = InMemorySaver()
31
 
 
12
 
13
  load_dotenv()
14
 
15
+ # gemini_api_key = os.getenv("gemini_api_key")
16
+ # llm = ChatGoogleGenerativeAI(
17
+ # model="gemini-2.0-flash",
18
+ # google_api_key=gemini_api_key,
 
 
 
 
 
 
 
19
  # temperature=0.3
20
  # )
21
 
22
+ groq_api_key = os.getenv("GROQ_API_KEY")
23
+ llm = ChatGroq(
24
+ model="llama-3.3-70b-versatile", # or "llama-3.1-70b-versatile"
25
+ api_key=groq_api_key,
26
+ temperature=0.3
27
+ )
28
+
29
  store = InMemoryStore()
30
  memory = InMemorySaver()
31