feat: 1.demo更新 2. 提示词更新

llmagent/__init__.py

@@ -34,12 +34,13 @@ PROMPT_TEMPLATE = {
         4. 确保每个操作都正确地基于前面的操作。
         通过等待并仔细考虑用户在每次使用工具后的回应，你可以相应地做出反应并做出关于如何继续任务的明智决定。这个迭代过程有助于确保你工作的整体成功和准确性。
         用户的指令是：{user_input}
+        今天的日期是：{today}
         """
     },
     'TOOL_CALLER': {
         'description': '工具调用器',
         'template': """
-        你是工具调用器，用户将会告诉你一条指令 {user_input}, 请根据指令推理出需要调用哪个工具。
+        你是工具调用器，用户将会告诉你一条指令 {user_input}, 请根据指令推理出需要调用哪个工具。今天的日期是:{today}。
         # 要求
         1.根据指令和提供的工具描述选择最合适的工具,并仔细阅读工具参数说明。
         2.你需要根据工具参数说明进行参数校验，评估用户输入的参数是否满足条件，如果不满足需要返回参数校验错误信息。

llmagent/llm_agent.py

@@ -19,6 +19,7 @@ from langchain_core.runnables.history import RunnableWithMessageHistory
 from langchain_core.messages import HumanMessage
 from log_conf import log
 from llmagent.llm_config import base_conf
+from datetime import datetime
 
 # debug模式,有更多输出
 set_debug(base_conf.debug)
@@ -55,6 +56,14 @@ def get_session_history(session_id: str) -> BaseChatMessageHistory:
     return his_store[session_id]
 
 
+def get_today():
+    """
+    获取今天日期
+    :return:
+    """
+    return datetime.now().strftime('%Y-%m-%d')
+
+
 class BaseChatAgent(ABC):
     """
     抽象Agent类
@@ -135,11 +144,12 @@ class BaseChatAgent(ABC):
                 """
         config = {"configurable": {"session_id": session_id}}
         # 总体任务描述及提示词
-        user_msg = PROMPT_TEMPLATE.get('VOICE_ASSISTANT')['template'].format(user_input=user_input)
+        user_msg = PROMPT_TEMPLATE.get('VOICE_ASSISTANT')['template'].format(user_input=user_input,
+                                                                             today=get_today())
         messages = [HumanMessage(user_msg)]
         llm_with_tools = llm.bind_tools(STRUCT_TOOLS, tool_choice='auto')
         # 判断使用哪个工具，需要加提示词让模型判断参数是否符合规则
-        user_input = PROMPT_TEMPLATE.get('TOOL_CALLER')['template'].format(user_input=user_input)
+        user_input = PROMPT_TEMPLATE.get('TOOL_CALLER')['template'].format(user_input=user_input, today=get_today())
         # 工具chain加入历史对话
         too_chain = self.multi_round_prompt | llm_with_tools
         with_message_history = RunnableWithMessageHistory(too_chain, get_session_history, input_messages_key="messages")

llmtools/tool_impl.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 # @Time    : 2025/3/16 16:38
 # @Author  : old-tom
-# @File    : basic_tool
+# @File    : tool_impl
 # @Project : llmFunctionCallDemo
 # @Desc    : 工具实现, @tool 装饰器是langchain提供的,包含了invoke方法可以直接调用。并且会自动返回ToolMessage，缺点是没有自定义异常处理
 # 此外还有其他定义tool方法，参考 https://www.langchain.com.cn/docs/how_to/custom_tools/
@@ -38,7 +38,8 @@ def split_screen(split_n: Annotated[int, "要切换的分屏数量,整数并且
 def play_video_record(camera_name: Annotated[str, "相机名称,例如：南卡口1号相机"],
                       start_time: Annotated[str, "录像开始时间,格式为yyyy-MM-dd hh:mm:ss，例 2025-03-16 01:00:00"],
                       end_time: Annotated[str, "录像结束时间,格式为yyyy-MM-dd hh:mm:ss，例 2025-03-16 02:09:31"]) -> str:
-    pass
+    log.info('【function】play_video_record 输入 [{},{},{}]', camera_name, start_time, end_time)
+    return f"正在打开{camera_name}的录像,请等待操作完成"
 
 
 @tool("switch_page", description="打开、跳转页面")
@@ -58,6 +59,7 @@ def view_flight_details(
             str, "航班编号,由字母+数字组成的完整编号,若编号包含多余字符（如标点符号），需过滤后保留有效部分"],
         flight_date: Annotated[str, "提取完整日期（年月日），自动补零至标准格式 yyyy-MM-dd, 例：2025-03-16"],
         ie_type: Annotated[str, "出入境类型，仅识别'入境'或'出境'两种类型"]) -> str:
+    log.info('【function】view_flight_details 输入 [{},{},{},{}]', airport_name, flight_code, flight_date,ie_type)
     return f"{airport_name}航班号{flight_code}{flight_date}{ie_type}数据,共100人乘机,起飞准时,晚点降落"
 
 

main.py

@@ -11,36 +11,18 @@ from llmagent.llm_agent import ChatAgent
 dsr = ChatAgent()
 
 if __name__ == '__main__':
-    ##########  测试 多轮对话 #########
-    dsr.multi_with_stream('你是什么模型', 1)
-    dsr.multi_with_stream('你能做什么', 1)
-    dsr.multi_with_stream('我的上一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)
-    dsr.multi_with_stream('我的第一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)
-
-    ##########  测试 多轮对话-相机选择 #########
-    dsr.multi_with_tool_call_stream('播放南卡口相机', 1)
-    dsr.multi_with_tool_call_stream('1', 1)
-
-
-    # print(dsr.invoke_with_tool_call('今天昆明天怎么样'))
-    ##########  测试 function call #########
-    # print(dsr.invoke_with_tool_call('播放南卡口相机'))
-    ## [{'name': 'play_video', 'args': {'camera_name': '北卡口1号道相机'}, 'id': 'call_apnb8fiqdkaz313katcs3tjf', 'type': 'tool_call'}]
+    # 1.参数提取
+    dsr.multi_with_tool_call_stream('播放北卡口入境1号道录像,从今天凌晨1点到2点', 1)
+    # dsr.multi_with_tool_call_stream('查看成都天府k00航班2004年1月1日入境预报航班人员明细', 1)
+    # 2.参数修正
     # dsr.multi_with_tool_call_stream('将大屏切换为-1分屏', 1)
-    ## [{'name': 'split_screen', 'args': {'split_n': 2}, 'id': 'call_2o7c94f591xag8p6lcyice9q', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('播放北卡口入境1号道录像,从今天到2025-03-16 02:09:31'))
-    ## 由于大模型没有联网，所以无法判断‘今天’
-    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号道', 'start_time': '2023-10-25 00:00:00', 'end_time': '2025-03-16 02:09:31'}, 'id': 'call_6gllr9wofeaoomgiz13bgnk6', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('播放北卡口入境1号道录像,从25年3月1号到2号'))
-    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号道', 'start_time': '2025-03-01 00:00:00', 'end_time': '2025-03-02 23:59:59'}, 'id': 'call_qj78jpep6k5wjwo90dkrz4qt', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('播放北卡口入境1号从25年3月1号到2号的录像'))
-    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号相机', 'start_time': '2025-03-01 00:00:00', 'end_time': '2025-03-02 23:59:59'}, 'id': 'call_ixpyke880tdhopv27c6moci0', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('跳转到首页'))
-    ## [{'name': 'switch_page', 'args': {'page_name': '首页'}, 'id': 'call_3ev11bjkyrqyzxaman3402az', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('打开日志'))
-    ## [{'name': 'switch_page', 'args': {'page_name': '系统日志'}, 'id': 'call_acwy2yk7xz3bgpt28suujca2', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool('放大地图到2级'))
-    ## [{'name': 'zoom_in', 'args': {'level_n': 2}, 'id': 'call_ip1kang0lbr63y8gya8nrpwc', 'type': 'tool_call'}]
-    # print(dsr.invoke_with_tool_call('查看成都天府k00航班2004年1月1日入境预报航班人员明细'))
-    ## [{'name': 'view_flight_details', 'args': {'airport_name': '成都天府机场', 'flight_code': 'K00', 'flight_date': '2004-01-01', 'ie_type': '入境'}, 'id': 'call_igummeorjq4r2pqjyr9tq6xq', 'type': 'tool_call'}]
 
+    # 3.多轮对话
+    # dsr.multi_with_stream('你是什么模型?请简要回答', 1)
+    # dsr.multi_with_stream('你能做什么?请简要回答', 1)
+    # dsr.multi_with_stream('我的上一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)
+    # dsr.multi_with_stream('我的第一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)
+
+    # 多轮对话-多个相机
+    # dsr.multi_with_tool_call_stream('播放南卡口相机', 1)
+    # dsr.multi_with_tool_call_stream('1', 1)