1
from langchain_core.tools import tool
2
import math
3
from datetime import datetime, timedelta
4

5
@tool
6
def get_weather(city: str) -> str:
7
    """获取指定城市的实时天气信息
8

9
    支持中国主要城市的天气查询
10

11
    Args:
12
        city: 城市名称，如"北京"、"上海"、"深圳"等
13

14
    Returns:
15
        包含温度、天气状况、空气质量的详细信息
16

17
    Examples:
18
        get_weather("北京") 返回 "多云，15-22℃，空气质量良"
19
    """
20
    weather_db = {
21
        "北京": "多云，15-22℃，空气质量良，湿度 45%",
22
        "上海": "晴天，18-25℃，空气质量优，湿度 60%",
23
        "深圳": "小雨，22-28℃，空气质量优，湿度 75%",
24
    }
25
    result = weather_db.get(city)
26
    if result:
27
        return f"{city}：{result}"
28
    return f"抱歉，暂不支持查询{city}的天气信息。当前支持：北京、上海、深圳"

1
@tool
2
def calculator(expression: str) -> str:
3
    """执行数学计算
4

5
    支持基本运算符（+、-、*、/、**）和常用数学函数
6

7
    Args:
8
        expression: 数学表达式，可以包含：
9
            - 基本运算：2 + 3, 10 * 5, 100 / 4
10
            - 幂运算：2 ** 10
11
            - 函数：sqrt(16), abs(-5), pow(2, 3)
12

13
    Returns:
14
        计算结果或错误信息
15

16
    Examples:
17
        calculator("2 + 3 * 4") 返回 "14"
18
        calculator("sqrt(16)") 返回 "4.0"
19
    """
20
    try:
21
        safe_functions = {
22
            "sqrt": math.sqrt, "pow": pow, "abs": abs, "round": round,
23
            "sin": math.sin, "cos": math.cos, "tan": math.tan, "log": math.log,
24
            "pi": math.pi, "e": math.e,
25
        }
26
        result = eval(expression, {"__builtins__": {}}, safe_functions)
27
        return f"{expression} = {result}"
28
    except Exception as e:
29
        return f"计算出错：{str(e)}\n提示：请检查表达式格式，支持的函数有 sqrt, abs, pow, sin, cos, tan, log"

1
@tool
2
def get_time_info(query_type: str = "current") -> str:
3
    """获取时间相关信息
4

5
    Args:
6
        query_type: 查询类型
7
            - "current": 当前时间
8
            - "date": 今天日期
9
            - "tomorrow": 明天日期
10
            - "yesterday": 昨天日期
11
            - "weekday": 星期几
12

13
    Returns:
14
        时间信息字符串
15

16
    Examples:
17
        get_time_info("current") 返回 "2025年1月25日 14:30:25"
18
        get_time_info("weekday") 返回 "星期六"
19
    """
20
    now = datetime.now()
21
    if query_type == "current":
22
        return now.strftime("当前时间：%Y年%m月%d日 %H:%M:%S")
23
    elif query_type == "date":
24
        return now.strftime("今天是：%Y年%m月%d日")
25
    elif query_type == "tomorrow":
26
        return (now + timedelta(days=1)).strftime("明天是：%Y年%m月%d日")
27
    elif query_type == "yesterday":
28
        return (now - timedelta(days=1)).strftime("昨天是：%Y年%m月%d日")
29
    elif query_type == "weekday":
30
        weekdays = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期日"]
31
        return f"今天是{weekdays[now.weekday()]}"
32
    return f"不支持的查询类型：{query_type}。支持：current, date, tomorrow, yesterday, weekday"

1
@tool
2
def convert_currency(amount: float, from_curr: str, to_curr: str) -> str:
3
    """货币转换工具
4
    支持主要货币之间的实时汇率转换
5

6
    Args:
7
        amount: 金额数值
8
        from_curr: 源货币代码（CNY/USD/EUR/GBP/JPY/HKD）
9
        to_curr: 目标货币代码（CNY/USD/EUR/GBP/JPY/HKD）
10

11
    Returns:
12
        转换结果
13

14
    Examples:
15
        convert_currency(100, "CNY", "USD") 返回 "100 CNY = 14.00 USD"
16
    """
17
    exchange_rates = {"CNY": 1.0, "USD": 0.14, "EUR": 0.13, "GBP": 0.11, "JPY": 20.8, "HKD": 1.09}
18
    currency_names = {"CNY": "人民币", "USD": "美元", "EUR": "欧元",
19
                      "GBP": "英镑", "JPY": "日元", "HKD": "港币"}
20

21
    from_curr, to_curr = from_curr.upper(), to_curr.upper()
22
    if from_curr not in exchange_rates:
23
        return f"不支持的源货币：{from_curr}。支持的货币：CNY, USD, EUR, GBP, JPY, HKD"
24
    if to_curr not in exchange_rates:
25
        return f"不支持的目标货币：{to_curr}。支持的货币：CNY, USD, EUR, GBP, JPY, HKD"
26

27
    cny_amount = amount / exchange_rates[from_curr]        # 先换成 CNY 再换目标货币
28
    result_amount = cny_amount * exchange_rates[to_curr]
29
    return f"{amount} {currency_names[from_curr]}（{from_curr}）= {result_amount:.2f} {currency_names[to_curr]}（{to_curr}）"

1
@tool
2
def search_info(keyword: str, category: str = "all") -> str:
3
    """搜索各类信息
4

5
    Args:
6
        keyword: 搜索关键词
7
        category: 搜索分类
8
            - "product": 搜索产品
9
            - "news": 搜索新闻
10
            - "all": 搜索所有
11

12
    Returns:
13
        搜索结果
14
    """
15
    products = {
16
        "手机": "iPhone 15 (¥5999), 小米14 (¥3999), 华为Mate60 (¥6999)",
17
        "笔记本": "MacBook Pro (¥12999), ThinkPad X1 (¥9999), 华为MateBook (¥7999)",
18
        "耳机": "AirPods Pro (¥1999), Sony WH-1000XM5 (¥2499)",
19
    }
20
    news = {
21
        "AI": "1. GPT-5 即将发布  2. AI 芯片市场增长 30%  3. 新AI法规出台",
22
        "科技": "1. 量子计算新突破  2. 6G 技术测试  3. 新能源汽车销量创新高",
23
    }
24

25
    results = []
26
    if category in ["product", "all"]:
27
        for key, value in products.items():
28
            if keyword in key:
29
                results.append(f"【产品】{key}：{value}")
30
    if category in ["news", "all"]:
31
        for key, value in news.items():
32
            if keyword in key or keyword in value:
33
                results.append(f"【新闻】{key} 相关：{value}")
34

35
    return "\n".join(results) if results else f"未找到关于 '{keyword}' 的{category}信息"

1
from langchain.agents import create_agent
2

3
class SmartAssistant:
4
    """多功能智能助手"""
5

6
    def __init__(self):
7
        self.model = model
8

9
        self.tools = [get_weather, calculator, get_time_info, convert_currency, search_info]
10

11
        system_prompt = """你是一个多功能智能助手，可以帮助用户：
12

13
          🌤️ 查询天气：使用 get_weather 工具
14
          🔢 数学计算：使用 calculator 工具
15
          ⏰ 时间查询：使用 get_time_info 工具
16
          💱 货币转换：使用 convert_currency 工具
17
          🔍 信息搜索：使用 search_info 工具
18

19
          重要提示：
20
          1. 仔细阅读用户问题，确定需要使用哪个工具
21
          2. 如果需要多个工具，按顺序调用
22
          3. 总是用友好、专业的语气回答
23
          4. 如果工具返回了数据，要用通俗易懂的语言解释给用户
24
          5. 如果无法完成任务，诚实地告诉用户原因
25

26
          请始终使用中文回答。"""
27

28
        self.agent = create_agent(
29
            model=self.model,
30
            tools=self.tools,
31
            system_prompt=system_prompt,
32
        )
33

34
        self.messages = []          # 对话历史
35

36
    def chat(self, user_input: str) -> str:
37
        """对话接口"""
38
        self.messages.append({"role": "user", "content": user_input})     # 1. 追加用户消息
39

40
        result = self.agent.invoke({"messages": self.messages})           # 2. 调用 agent（带上历史）
41

42
        self.messages = result["messages"]                                # 3. 用返回的完整消息列表更新历史
43

44
        for msg in reversed(self.messages):                               # 4. 从后往前找最后一条有内容的 AI 消息
45
            if msg.type == "ai" and msg.content:
46
                return msg.content
47

48
        return "抱歉，我无法处理这个请求。"
49

50
    def reset(self):
51
        """重置对话历史"""
52
        self.messages = []

1
def main():
2
    assistant = SmartAssistant()
3

4
    print("=" * 40)
5
    print("🤖 多功能智能助手（LangChain 1.2）")
6
    print("=" * 40)
7
    print("\n输入 'quit' 退出，输入 'reset' 重置对话\n")
8

9
    # 先自动跑几条演示，确认每个工具都能被正确调用
10
    demos = [
11
        "北京今天天气怎么样？",
12
        "帮我算一下 (25 + 17) * 3",
13
        "现在几点了？",
14
        "100 美元等于多少人民币？",
15
    ]
16

17
    for demo in demos:
18
        print(f"👤 {demo}")
19
        response = assistant.chat(demo)
20
        print(f"🤖 {response}\n")
21

22
    assistant.reset()
23

24
    # 交互模式
25
    while True:
26
        user_input = input("\n👤 你: ")
27
        if user_input.lower() == "quit":
28
            print("再见！👋")
29
            break
30
        if user_input.lower() == "reset":
31
            assistant.reset()
32
            print("✅ 对话已重置")
33
            continue
34
        if not user_input.strip():
35
            continue
36
        print(f"🤖 助手: {assistant.chat(user_input)}")
37

38
if __name__ == "__main__":
39
    main()