Update agent.py
Browse files
agent.py
CHANGED
|
@@ -28,7 +28,7 @@ class AgentState(TypedDict):
|
|
| 28 |
def build_gemini_llm():
|
| 29 |
if not os.environ.get("GOOGLE_API_KEY"):
|
| 30 |
raise ValueError("GOOGLE_API_KEY environment variable is not set.")
|
| 31 |
-
return ChatGoogleGenerativeAI(model = "gemini-3.7-flash",
|
| 32 |
|
| 33 |
@tool
|
| 34 |
def add_numbers(a: int, b: int) -> int:
|
|
|
|
| 28 |
def build_gemini_llm():
|
| 29 |
if not os.environ.get("GOOGLE_API_KEY"):
|
| 30 |
raise ValueError("GOOGLE_API_KEY environment variable is not set.")
|
| 31 |
+
return ChatGoogleGenerativeAI(model = "gemini-3.7-flash", temperature = 0, max_output_tokens = 1024)
|
| 32 |
|
| 33 |
@tool
|
| 34 |
def add_numbers(a: int, b: int) -> int:
|