LangChain核心组件-其它

工具tools

  • 扩展LLM能力边界的外部功能接口
from langchain.agents import tool
from datetime import datetime

# 自定义工具
@tool
def get_current_time(timezone: str = "Asia/Shanghai") -> str:
    """获取指定时区的当前时间"""
    now = datetime.now()
    return f"当前时间：{now.strftime('%Y-%m-%d %H:%M:%S')}"

@tool 
def query_database(query: str) -> str:
    """执行数据库查询"""
    # 实际的数据查询逻辑
    return f"查询结果：{query}"

# 工具注册
tools = [get_current_time, query_database]

# 在Agent中使用
agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION)