kshitijthakkar commited on
Commit
c976bf6
·
verified ·
1 Parent(s): 4c8870a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -45,16 +45,6 @@ class GaiaAgent:
45
  self.model = OpenAIServerModel(
46
  model_id="gpt-4", # or "gpt-3.5-turbo" if you prefer
47
  api_key=self.openai_key,
48
- )
49
- # 2) Define the tools
50
- self.search_tool = DuckDuckGoSearchTool()
51
- self.python_tool = PythonREPLTool(timeout=10) # Initialize PythonREPLTool
52
- # 3) Create the CodeAgent
53
- self.agent = CodeAgent(
54
- model=self.model,
55
- tools=[self.search_tool, self.python_tool],
56
- # Encourage the agent to think step-by-step in code
57
- max_steps=20,
58
  system_prompt=(
59
  "You are a meticulous AI agent. "
60
  "Always think in Python code using the available tools. "
@@ -75,6 +65,16 @@ class GaiaAgent:
75
  "# ```python\n"
76
  "# print('hello'[::-1])\n"
77
  "# ```\n"
 
 
 
 
 
 
 
 
 
 
78
  ),
79
  )
80
 
 
45
  self.model = OpenAIServerModel(
46
  model_id="gpt-4", # or "gpt-3.5-turbo" if you prefer
47
  api_key=self.openai_key,
 
 
 
 
 
 
 
 
 
 
48
  system_prompt=(
49
  "You are a meticulous AI agent. "
50
  "Always think in Python code using the available tools. "
 
65
  "# ```python\n"
66
  "# print('hello'[::-1])\n"
67
  "# ```\n"
68
+ )
69
+ # 2) Define the tools
70
+ self.search_tool = DuckDuckGoSearchTool()
71
+ self.python_tool = PythonREPLTool(timeout=10) # Initialize PythonREPLTool
72
+ # 3) Create the CodeAgent
73
+ self.agent = CodeAgent(
74
+ model=self.model,
75
+ tools=[self.search_tool, self.python_tool],
76
+ # Encourage the agent to think step-by-step in code
77
+ max_steps=20
78
  ),
79
  )
80