""" title: WolframAlpha author: sweetbrulee descriptions: This tool sends a request to the WolframAlpha API with a relevant query to provide an augmented answer, along with the links. version: 0.2.0 licence: MIT """ # Base code from # https://openwebui.com/t/s94rz/wolframalpha_llm_api/ # - Thanks mate! # 1. Get a free WolframAlpha API key at https://developer.wolframalpha.com/ # 2. Set it using the valve (in workspace==>tools==>settings of WolframAlpha LLM API) or the env variable WOLFRAMALPHA_APP_ID # Enjoy :) import os import logging import json import urllib.parse from typing import Callable, Awaitable import httpx from pydantic import BaseModel, Field logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) class WolframalphaException(Exception): """Base exception for WolframAlpha API related errors.""" class WolframalphaHTTPException(WolframalphaException): """Exception for HTTP related errors when interacting with the WolframAlpha API.""" class WolframalphaJSONDecodeException(WolframalphaException): """Exception for JSON decode errors when parsing responses from the WolframAlpha API.""" async def query_impl( input_url_encoded: str, appid: str, __event_emitter__: Callable[[dict], Awaitable[None]], ) -> str: url = f"https://www.wolframalpha.com/api/v1/llm-api?input={input_url_encoded}&appid={appid}" try: async with httpx.AsyncClient(timeout=60) as client: response = await client.get(url) response.raise_for_status() response_text = response.text if not response_text.strip(): await __event_emitter__( { "data": { "description": "No result found for the query", "status": "complete", "done": True, }, "type": "status", } ) return ( f"WolframAlpha Query: {input_url_encoded}\n" f"WolframAlpha Response: No result found for the query '{input_url_encoded}'" ) await __event_emitter__( { "data": { "description": "WolframAlpha query completed", "status": "complete", "done": True, }, "type": "status", } ) return response_text except httpx.HTTPStatusError as errh: logger.error("HTTP Error: %s", errh) raise WolframalphaHTTPException( f"HTTP Error: {errh}, Hint: {errh.response.text}" ) from errh except json.JSONDecodeError as e: logger.error("JSON Decode Error: %s", e) raise WolframalphaJSONDecodeException(f"JSON Decode Error: {e}") from e except Exception as e: logger.error("WolframAlpha Error: %s", e) raise WolframalphaException(f"WolframAlpha Error: {e}") from e class Tools: class Valves(BaseModel): wolframalpha_APP_ID: str = Field( default=None, description="The App ID (api key) to authorize WolframAlpha", ) def __init__(self): self.valves = self.Valves() # query_wolframalpha_llm_api async def action( self, input_in_english: str, __event_emitter__: Callable[[dict], Awaitable[None]], ) -> str: """ - Convert inputs to simplified keyword queries whenever possible (e.g. convert "how many people live in France" to "France population"). - ALWAYS use this exponent notation: `6*10^14`, NEVER `6e14`. - ALWAYS use proper Markdown formatting for all math, scientific, and chemical formulas, symbols, etc.: '$$\n[expression]\n$$' for standalone cases and '\( [expression] \)' when inline. - Use ONLY single-letter variable names, with or without integer subscript (e.g., n, n1, n_1). - Use named physical constants (e.g., 'speed of light') without numerical substitution. - Include a space between compound units (e.g., "Ω m" for "ohm*meter"). - To solve for a variable in an equation with units, consider solving a corresponding equation without units; exclude counting units (e.g., books), include genuine units (e.g., kg). - If data for multiple properties is needed, make separate calls for each property. -- ONLY simplify or rephrase the initial query if a more relevant 'Assumption' or other input suggestions are not provided. :param input_in_english: text specifying the input string in English """ # Here we are assuming the LLM already knows how to correctly query and prompt the WolframAlpha API appid = self.valves.wolframalpha_APP_ID or os.getenv("WOLFRAMALPHA_APP_ID") logger.info("App ID = %s", appid) if not appid: await __event_emitter__( { "data": { "description": "Error: WolframAlpha APP_ID is not set", "status": "complete", "done": True, }, "type": "status", } ) return ( "You are required to report the following error message to the user:" "App ID is not set in the Valves or the environment variable 'WolframAlpha_APP_ID'." ) await __event_emitter__( { "data": { "description": f"WolframAlpha query started", "status": "in_progress", "done": False, }, "type": "status", } ) try: # URL encode the input string input_url_encoded = urllib.parse.quote(input_in_english) api_answer = await query_impl(input_url_encoded, appid, __event_emitter__) await __event_emitter__( { "data": { "description": input_in_english, "status": "complete", "done": True, }, "type": "status", } ) return ( f"API Response:\n{api_answer}\n\n" f"Instructions:\n" f"- Write a digestible answer based on the API response and the previous message(s).\n" f"- If there are several result(s) formats in the API response, present them all directly.\n" f"- Use KaTeX syntax for all mathematical expressions.\n" f"- Display all the image URLs found in the API response with the Markdown syntax ''.\n" # f"- Provide the links for both the WolframAlpha result webpage (found in the API response) and the API response webpage (which is https://www.wolframalpha.com/api/v1/llm-api?appid={app_id}&input={query_string_english}, but in a URL-encoded format)." ) except WolframalphaException as e: await __event_emitter__( { "data": { "description": f"{str(e)}", "status": "complete", "done": True, }, "type": "status", } ) return ( f"WolframAlpha Query: {input_in_english}\n" f"There was an error fetching the response: {str(e)}" )