Langchain-Chatchat

1
# LangChain 的 ArxivQueryRun 工具
2
from pydantic import BaseModel, Field
3
from langchain.tools.arxiv.tool import ArxivQueryRun
4
def arxiv(query: str):
5
    tool = ArxivQueryRun()
6
    return tool.run(tool_input=query)
7

8
class ArxivInput(BaseModel):
9
    query: str = Field(description="The search query title")