From 6ab3410581d664dd602f6af68f5c8f78bfdb4806 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 13 二月 2025 13:53:28 +0800 Subject: [PATCH] 增加ds模型文档智能 --- app/task/sync_resources.py | 17 +++++++++++------ 1 files changed, 11 insertions(+), 6 deletions(-) diff --git a/app/task/sync_resources.py b/app/task/sync_resources.py index 5fddf33..fca2ae8 100644 --- a/app/task/sync_resources.py +++ b/app/task/sync_resources.py @@ -1,19 +1,22 @@ +import asyncio + from fastapi import Depends from Log import logger from app.config.config import settings from app.models import AgentType -from app.models.app_model import AppRegisterModel +from app.models.public_api_model import AppRegisterModel from app.models.base_model import get_db from app.service.bisheng import BishengService from app.service.ragflow import RagflowService -async def sync_resource(): - await sync_knowledge() - await sync_dialog() - await sync_agent() - await sync_llm() +def sync_resource(): + asyncio.run(sync_knowledge()) + asyncio.run(sync_dialog()) + asyncio.run(sync_agent()) + asyncio.run(sync_llm()) + async def sync_knowledge(db=Depends(get_db)): @@ -24,12 +27,14 @@ if rapp.app_type == AgentType.RAGFLOW: token = "" ragflow_service = RagflowService(settings.fwr_base_url) + ragflow_service.get_knowledge_list() elif rapp.app_type ==AgentType.BISHENG: token = "" bisheng_service = BishengService(settings.sgb_base_url) else: logger.error("娉ㄥ唽鏈煡搴旂敤锛歿}".format(rapp.app_type)) + async def sync_dialog(): ... -- Gitblit v1.8.0