Langchain-Chatchat
11 строк · 434.0 Байт
1# Langchain 自带的 Wolfram Alpha API 封装
2from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
3from pydantic import BaseModel, Field
4wolfram_alpha_appid = "your key"
5def wolfram(query: str):
6wolfram = WolframAlphaAPIWrapper(wolfram_alpha_appid=wolfram_alpha_appid)
7ans = wolfram.run(query)
8return ans
9
10class WolframInput(BaseModel):
11location: str = Field(description="需要运算的具体问题")