dav74 commited on
Commit
2c0f32b
·
verified ·
1 Parent(s): d19e6fb

Update agent/graph.py

Browse files
Files changed (1) hide show
  1. agent/graph.py +1 -1
agent/graph.py CHANGED
@@ -23,7 +23,7 @@ class AgentState(TypedDict):
23
  # Ensure OPENAI_API_KEY is set in the environment
24
  #GOOGLE_API_KEY = ""
25
  try:
26
- llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite", temperature=0.7, api_key=GOOGLE_API_KEY)
27
  except Exception as e:
28
  print(f"Warning: Failed to initialize ChatOpenAI: {e}")
29
  llm = None
 
23
  # Ensure OPENAI_API_KEY is set in the environment
24
  #GOOGLE_API_KEY = ""
25
  try:
26
+ llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite", temperature=0.7)
27
  except Exception as e:
28
  print(f"Warning: Failed to initialize ChatOpenAI: {e}")
29
  llm = None