feat: 企业内部文档检索工具

This commit is contained in:
martsforever
2025-09-11 00:45:19 +08:00
parent 1697324bf4
commit b9a79f7361
6 changed files with 28 additions and 4 deletions
+3 -1
View File
@@ -1,3 +1,5 @@
from typing import List, Any
from langchain_core.runnables import RunnableConfig
from langchain_core.tools import tool
@@ -11,7 +13,7 @@ from app.utils.db_utils import async_session
name_or_callable="tool_query_projects",
description="查询参与的项目信息"
)
async def tool_query_projects(config: RunnableConfig) -> float:
async def tool_query_projects(config: RunnableConfig) -> List[Any]:
user_id = config.get('configurable').get('user_id')
async with async_session() as session: