mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-24 15:53:21 +08:00
17 lines
592 B
Python
17 lines
592 B
Python
# Langchain 自带的 Wolfram Alpha API 封装
|
|
|
|
from chatchat.server.pydantic_v1 import Field
|
|
from chatchat.server.utils import get_tool_config
|
|
from .tools_registry import regist_tool, BaseToolOutput
|
|
|
|
|
|
@regist_tool
|
|
def wolfram(query: str = Field(description="The formula to be calculated")):
|
|
'''Useful for when you need to calculate difficult formulas'''
|
|
|
|
from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
|
|
|
|
wolfram = WolframAlphaAPIWrapper(wolfram_alpha_appid=get_tool_config("wolfram").get("appid"))
|
|
ans = wolfram.run(query)
|
|
return BaseToolOutput(ans)
|