PYAE1994 commited on
Commit
2df004c
·
verified ·
1 Parent(s): 62eebb7

Update app/agent/tools_runtime.py

Browse files
Files changed (1) hide show
  1. app/agent/tools_runtime.py +47 -1
app/agent/tools_runtime.py CHANGED
@@ -1,6 +1,26 @@
1
  import os
2
  import subprocess
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  # =========================
5
  # BASIC SYSTEM TOOLS
6
  # =========================
@@ -92,7 +112,7 @@ def delete_file(path: str):
92
 
93
 
94
  # =========================
95
- # AGENT TOOL ROUTER (STEP 3.3)
96
  # =========================
97
 
98
  def execute_tool_call(tool_call: dict):
@@ -103,6 +123,9 @@ def execute_tool_call(tool_call: dict):
103
  tool = tool_call.get("tool")
104
  args = tool_call.get("args", {})
105
 
 
 
 
106
  if tool == "run_shell":
107
  return run_shell(args.get("cmd", ""))
108
 
@@ -121,6 +144,29 @@ def execute_tool_call(tool_call: dict):
121
  elif tool == "delete_file":
122
  return delete_file(args.get("path", ""))
123
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
124
  return {
125
  "tool": tool,
126
  "error": "unknown_tool"
 
1
  import os
2
  import subprocess
3
 
4
+ # =========================
5
+ # BROWSER LAYER
6
+ # =========================
7
+ from browser_tools import BrowserTool
8
+
9
+ browser = BrowserTool()
10
+
11
+ def browser_open(url: str):
12
+ return browser.open_url(url)
13
+
14
+ def browser_click(selector: str):
15
+ return browser.click(selector)
16
+
17
+ def browser_type(selector: str, text: str):
18
+ return browser.type(selector, text)
19
+
20
+ def browser_screenshot(path: str = "screen.png"):
21
+ return browser.screenshot(path)
22
+
23
+
24
  # =========================
25
  # BASIC SYSTEM TOOLS
26
  # =========================
 
112
 
113
 
114
  # =========================
115
+ # TOOL ROUTER (AI → ACTION)
116
  # =========================
117
 
118
  def execute_tool_call(tool_call: dict):
 
123
  tool = tool_call.get("tool")
124
  args = tool_call.get("args", {})
125
 
126
+ # -------------------------
127
+ # SYSTEM TOOLS
128
+ # -------------------------
129
  if tool == "run_shell":
130
  return run_shell(args.get("cmd", ""))
131
 
 
144
  elif tool == "delete_file":
145
  return delete_file(args.get("path", ""))
146
 
147
+ # -------------------------
148
+ # BROWSER TOOLS (STEP 4)
149
+ # -------------------------
150
+ elif tool == "browser_open":
151
+ return browser_open(args.get("url", ""))
152
+
153
+ elif tool == "browser_click":
154
+ return browser_click(args.get("selector", ""))
155
+
156
+ elif tool == "browser_type":
157
+ return browser_type(
158
+ args.get("selector", ""),
159
+ args.get("text", "")
160
+ )
161
+
162
+ elif tool == "browser_screenshot":
163
+ return browser_screenshot(
164
+ args.get("path", "screen.png")
165
+ )
166
+
167
+ # -------------------------
168
+ # UNKNOWN TOOL
169
+ # -------------------------
170
  return {
171
  "tool": tool,
172
  "error": "unknown_tool"