2023-10-23 13:22:10 +08:00
|
|
|
# Langchain 自带的 Wolfram Alpha API 封装
|
|
|
|
|
from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
|
2023-10-27 22:53:43 +08:00
|
|
|
from pydantic import BaseModel, Field
|
2023-10-23 13:22:10 +08:00
|
|
|
wolfram_alpha_appid = "your key"
|
|
|
|
|
def wolfram(query: str):
|
|
|
|
|
wolfram = WolframAlphaAPIWrapper(wolfram_alpha_appid=wolfram_alpha_appid)
|
|
|
|
|
ans = wolfram.run(query)
|
2023-10-27 22:53:43 +08:00
|
|
|
return ans
|
|
|
|
|
|
|
|
|
|
class WolframInput(BaseModel):
|
|
|
|
|
location: str = Field(description="需要运算的具体问题")
|
