Spaces:
Running
Running
Daniel Amendoeira
commited on
Update agent.py
Browse files
agent.py
CHANGED
@@ -15,23 +15,12 @@ from prompt import system_prompt
|
|
15 |
|
16 |
# gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
|
17 |
# for benchmarking use 04-mini (better reasoning) or gpt-4.1-mini
|
18 |
-
|
19 |
llm = ChatOpenAI(
|
20 |
model="gpt-4.1-nano",
|
21 |
api_key=os.getenv("OPENAI_API_KEY"),
|
22 |
temperature=0
|
23 |
)
|
24 |
|
25 |
-
|
26 |
-
#TEST
|
27 |
-
try:
|
28 |
-
response = llm.invoke([HumanMessage(content="Say hello")])
|
29 |
-
print("LLM Response:", response.content)
|
30 |
-
except Exception as e:
|
31 |
-
print("LLM Error:", e)
|
32 |
-
#TEST
|
33 |
-
|
34 |
-
|
35 |
python_tool = PythonAstREPLTool()
|
36 |
search_tool = BraveSearch.from_api_key(
|
37 |
api_key=os.getenv("BRAVE_SEARCH_API"),
|
|
|
15 |
|
16 |
# gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
|
17 |
# for benchmarking use 04-mini (better reasoning) or gpt-4.1-mini
|
|
|
18 |
llm = ChatOpenAI(
|
19 |
model="gpt-4.1-nano",
|
20 |
api_key=os.getenv("OPENAI_API_KEY"),
|
21 |
temperature=0
|
22 |
)
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
python_tool = PythonAstREPLTool()
|
25 |
search_tool = BraveSearch.from_api_key(
|
26 |
api_key=os.getenv("BRAVE_SEARCH_API"),
|