Spaces:
Runtime error
Runtime error
File size: 2,765 Bytes
e67043b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import requests
from ..tool import Tool
from typing import Any
import os
import xmltodict
def build_tool(config) -> Tool:
tool = Tool(
"Wolfram",
"Wolfram",
name_for_model="Wolfram",
name_for_human="Wolfram",
description_for_model=""""Dynamic computation and curated data from WolframAlpha and Wolfram Cloud.\nOnly use the getWolframAlphaResults endpoints; all other Wolfram endpoints are deprecated.\nPrefer getWolframAlphaResults unless Wolfram Language code should be evaluated.\nTry to include images returned by getWolframAlphaResults. Queries to getWolframAlphaResults must ALWAYS have this structure: {\"input\": query}.\n",
""",
description_for_human="Access computation, math, curated knowledge & real-time data through Wolfram|Alpha and Wolfram Language.",
logo_url="https://www.wolframcdn.com/images/icons/Wolfram.png",
contact_email="[email protected]",
legal_info_url="[email protected]",
)
@tool.get("/getWolframAlphaResults")
def getWolframAlphaResults(input: str):
"""Get Wolfram|Alpha results using natural query. Queries to getWolframAlphaResults must ALWAYS have this structure: {\"input\": query}. And please directly read the output json."""
URL = "https://api.wolframalpha.com/v2/query"
APPID = config["subscription_key"]
params = {"appid": APPID, "input": input}
response = requests.get(URL, params=params)
json_data = xmltodict.parse(response.text)
if "pod" not in json_data["queryresult"]:
return "WolframAlpha API cannot parse the input query."
rets = json_data["queryresult"]["pod"]
cleaned_rets = []
blacklist = [
"@scanner",
"@id",
"@position",
"@error",
"@numsubpods",
"@width",
"@height",
"@type",
"@themes",
"@colorinvertable",
"expressiontypes",
]
def filter_dict(d, blacklist):
if isinstance(d, dict):
return {
k: filter_dict(v, blacklist)
for k, v in d.items()
if k not in blacklist
}
elif isinstance(d, list):
return [filter_dict(i, blacklist) for i in d]
else:
return d
for ret in rets:
ret = filter_dict(ret, blacklist=blacklist)
# Do further cleaning to retain only the input and result pods
if "@title" in ret:
if ret["@title"] == "Input" or ret["@title"] == "Result":
cleaned_rets.append(ret)
return cleaned_rets
return tool
|