diff --git a/docs/changelog/update.md b/docs/changelog/update.md index 18505ac1..50d955ee 100644 --- a/docs/changelog/update.md +++ b/docs/changelog/update.md @@ -1,6 +1,7 @@ 目前已有的开发计划包括: 💭 **Features Todo** +- [ ] LangGraph 升级到 0.6+ 版本,并适配新特性,如 context 等。 - [ ] 使用其他的聊天记录管理方法,解决两个问题,一个是上下文长度过长,一个是上下文的类型变得更加丰富,比如多模态等等。(现在是基于 LangGraph 的 Memory 实现的,v0.2.3 版本实现),暂定使用 [mem0](github.com/mem0ai/mem0) 来实现。但是目前了解下来,还不是我想要的那种方案。可能会基于这个实现一个 ThreadConvManager 这个类。 - [ ] 添加对于上传文件的支持:这里的复杂的地方就在于如何和历史记录结合在一起(v0.2.3 版本实现,放在记忆管理后面) @@ -13,6 +14,8 @@ - [ ] 支持数据库查询功能 - [ ] 消息内容支持图片显示 +- [ ] 添加 SQL 读取工具 +- [ ] 添加绘图工具(这里的绘图是指绘制固定格式的图和表等,暂时没想好如何支持自定义绘图) - [ ] 添加测试脚本,覆盖最常见的功能 - [ ] 添加用户日志与用户反馈模块,可以在 AgentView 中查看信息 - [ ] 优化对文档信息的检索展示(检索结果页、详情页) diff --git a/src/agents/tools_factory.py b/src/agents/tools_factory.py index a402d927..29d28079 100644 --- a/src/agents/tools_factory.py +++ b/src/agents/tools_factory.py @@ -84,8 +84,8 @@ def get_kb_based_tools() -> dict[str, Any]: name=tool_id, description=description, args_schema=KnowledgeRetrieverModel, - metadata=retrieve_info | { - "tag": "knowledgebase" + metadata=retrieve_info["metadata"] | { + "tag": ["knowledgebase"] } ) diff --git a/src/knowledge/knowledge_base.py b/src/knowledge/knowledge_base.py index bf6857f1..5516aa63 100644 --- a/src/knowledge/knowledge_base.py +++ b/src/knowledge/knowledge_base.py @@ -439,6 +439,7 @@ class KnowledgeBase(ABC): "name": meta["name"], "description": meta["description"], "retriever": make_retriever(db_id), + "metadata": meta } return retrievers diff --git a/web/src/components/AgentConfigSidebar.vue b/web/src/components/AgentConfigSidebar.vue index 80e71241..f3f1903f 100644 --- a/web/src/components/AgentConfigSidebar.vue +++ b/web/src/components/AgentConfigSidebar.vue @@ -205,7 +205,7 @@