Spaces:
Sleeping
Sleeping
Update tools/getDeviceInfo.py
Browse files- tools/getDeviceInfo.py +7 -3
tools/getDeviceInfo.py
CHANGED
@@ -13,9 +13,11 @@ class getDeviceInfo(Tool):
|
|
13 |
name = "visit_webpage"
|
14 |
description = "Visits a webpage at the given URL, reads its content as a markdown string, and extracts device information."
|
15 |
inputs = {'url': {'type': 'string', 'description': 'The URL of the webpage to visit.'}}
|
16 |
-
output_type = "dict"
|
|
|
|
|
|
|
17 |
|
18 |
-
def forward(self, url: str) -> dict:
|
19 |
try:
|
20 |
import requests
|
21 |
from markdownify import markdownify
|
@@ -65,10 +67,12 @@ class getDeviceInfo(Tool):
|
|
65 |
device_info = page.evaluate(script)
|
66 |
browser.close()
|
67 |
|
68 |
-
|
|
|
69 |
"webpage_content": webpage_content,
|
70 |
"device_info": device_info
|
71 |
}
|
|
|
72 |
|
73 |
def __init__(self, *args, **kwargs):
|
74 |
self.is_initialized = False
|
|
|
13 |
name = "visit_webpage"
|
14 |
description = "Visits a webpage at the given URL, reads its content as a markdown string, and extracts device information."
|
15 |
inputs = {'url': {'type': 'string', 'description': 'The URL of the webpage to visit.'}}
|
16 |
+
output_type = "json" # ✅ Fix: Use "json" instead of "dict"
|
17 |
+
|
18 |
+
def forward(self, url: str) -> str: # ✅ Return a JSON string
|
19 |
+
import json
|
20 |
|
|
|
21 |
try:
|
22 |
import requests
|
23 |
from markdownify import markdownify
|
|
|
67 |
device_info = page.evaluate(script)
|
68 |
browser.close()
|
69 |
|
70 |
+
# ✅ Convert output to JSON string
|
71 |
+
result = {
|
72 |
"webpage_content": webpage_content,
|
73 |
"device_info": device_info
|
74 |
}
|
75 |
+
return json.dumps(result) # ✅ Ensure return type matches "json"
|
76 |
|
77 |
def __init__(self, *args, **kwargs):
|
78 |
self.is_initialized = False
|