From 351cd801ebf0da665c521448c978da100b4aaf03 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 05 十二月 2024 15:21:08 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/rag-gateway
---
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