From cdb5f8bf45f117831959291c89e0694606ebb479 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 04 十二月 2024 17:28:58 +0800 Subject: [PATCH] 增加公开api --- app/api/files.py | 8 +++++++- 1 files changed, 7 insertions(+), 1 deletions(-) diff --git a/app/api/files.py b/app/api/files.py index 4be3304..dca7d3c 100644 --- a/app/api/files.py +++ b/app/api/files.py @@ -15,6 +15,7 @@ from app.models.user_model import UserModel from app.service.basic import BasicService from app.service.bisheng import BishengService +from app.service.common.api_token import DfTokenDao from app.service.difyService import DifyService from app.service.ragflow import RagflowService from app.service.service_token import get_ragflow_token, get_bisheng_token @@ -96,9 +97,11 @@ # result = await service.paper_file_upload(chat_id, file.filename, file_content) elif agent.agent_type == AgentType.DIFY: - token = settings.dify_api_token dify_service = DifyService(base_url=settings.dify_base_url) if agent.type == "imageTalk": + token = DfTokenDao(db).get_token_by_id("image_and_text_conversion") + if not token: + raise HTTPException(status_code=500, detail="鑾峰彇token澶辫触锛宨mage_and_text_conversion锛�") file = file[0] # 璇诲彇涓婁紶鐨勬枃浠跺唴瀹� try: @@ -110,6 +113,9 @@ except Exception as e: raise HTTPException(status_code=500, detail=str(e)) elif agent.type == "reportWorkflow": + token = DfTokenDao(db).get_token_by_id("document_to_report") + if not token: + raise HTTPException(status_code=500, detail="鑾峰彇token澶辫触锛宒ocument_to_report锛�") result = [] for f in file: try: -- Gitblit v1.8.0