Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
@@ -18,7 +18,7 @@ from tools import (
|
|
18 |
)
|
19 |
|
20 |
# βββββββββββββββββββββββββββ Configuration βββββββββββββββββββββββββββββββ
|
21 |
-
|
22 |
MAX_TOOL_CALLS = 5
|
23 |
|
24 |
# βββββββββββββββββββββββββββ Helper utilities ββββββββββββββββββββββββββββ
|
@@ -35,7 +35,8 @@ MAX_TOOL_CALLS = 5
|
|
35 |
def build_graph():
|
36 |
graph = StateGraph(AgentState)
|
37 |
|
38 |
-
|
|
|
39 |
llm_tools = [
|
40 |
wikipedia_search_tool,
|
41 |
arxiv_search_tool,
|
@@ -43,7 +44,7 @@ def build_graph():
|
|
43 |
excel_tool,
|
44 |
analyze_code_tool,
|
45 |
]
|
46 |
-
llm = llm.bind_tools(llm_tools)
|
47 |
agent = create_react_agent(llm, llm_tools)
|
48 |
|
49 |
return agent
|
|
|
18 |
)
|
19 |
|
20 |
# βββββββββββββββββββββββββββ Configuration βββββββββββββββββββββββββββββββ
|
21 |
+
|
22 |
MAX_TOOL_CALLS = 5
|
23 |
|
24 |
# βββββββββββββββββββββββββββ Helper utilities ββββββββββββββββββββββββββββ
|
|
|
35 |
def build_graph():
|
36 |
graph = StateGraph(AgentState)
|
37 |
|
38 |
+
llm = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.3)
|
39 |
+
|
40 |
llm_tools = [
|
41 |
wikipedia_search_tool,
|
42 |
arxiv_search_tool,
|
|
|
44 |
excel_tool,
|
45 |
analyze_code_tool,
|
46 |
]
|
47 |
+
# llm = llm.bind_tools(llm_tools)
|
48 |
agent = create_react_agent(llm, llm_tools)
|
49 |
|
50 |
return agent
|