15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: add AppApiException for missing application key in chat steps

This commit is contained in:
CaptainB 2025-12-25 10:50:54 +08:00
parent 71c63adb0b
commit c067479643
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -26,6 +26,7 @@ from application.chat_pipeline.pipeline_manage import PipelineManage
from application.chat_pipeline.step.chat_step.i_chat_step import IChatStep, PostResponseHandler from application.chat_pipeline.step.chat_step.i_chat_step import IChatStep, PostResponseHandler
from application.flow.tools import Reasoning, mcp_response_generator from application.flow.tools import Reasoning, mcp_response_generator
from application.models import ApplicationChatUserStats, ChatUserType, Application, ApplicationApiKey from application.models import ApplicationChatUserStats, ChatUserType, Application, ApplicationApiKey
from common.exception.app_exception import AppApiException
from common.utils.logger import maxkb_logger from common.utils.logger import maxkb_logger
from common.utils.rsa_util import rsa_long_decrypt from common.utils.rsa_util import rsa_long_decrypt
from common.utils.tool_code import ToolExecutor from common.utils.tool_code import ToolExecutor
@ -273,7 +274,10 @@ class BaseChatStep(IChatStep):
if app_key is not None: if app_key is not None:
api_key = app_key.secret_key api_key = app_key.secret_key
else: else:
continue raise AppApiException(
500,
_('Application Key is required for application tool 【{name}】').format(name=app.name)
)
executor = ToolExecutor() executor = ToolExecutor()
app_config = executor.get_app_mcp_config(api_key) app_config = executor.get_app_mcp_config(api_key)
mcp_servers_config[app.name] = app_config mcp_servers_config[app.name] = app_config

7
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -13,6 +13,7 @@ from functools import reduce
from typing import List, Dict from typing import List, Dict
from django.db.models import QuerySet from django.db.models import QuerySet
from django.utils.translation import gettext as _
from langchain.schema import HumanMessage, SystemMessage from langchain.schema import HumanMessage, SystemMessage
from langchain_core.messages import BaseMessage, AIMessage from langchain_core.messages import BaseMessage, AIMessage
@ -20,6 +21,7 @@ from application.flow.i_step_node import NodeResult, INode
from application.flow.step_node.ai_chat_step_node.i_chat_node import IChatNode from application.flow.step_node.ai_chat_step_node.i_chat_node import IChatNode
from application.flow.tools import Reasoning, mcp_response_generator from application.flow.tools import Reasoning, mcp_response_generator
from application.models import Application, ApplicationApiKey from application.models import Application, ApplicationApiKey
from common.exception.app_exception import AppApiException
from common.utils.rsa_util import rsa_long_decrypt from common.utils.rsa_util import rsa_long_decrypt
from common.utils.tool_code import ToolExecutor from common.utils.tool_code import ToolExecutor
from models_provider.models import Model from models_provider.models import Model
@ -259,7 +261,10 @@ class BaseChatNode(IChatNode):
if app_key is not None: if app_key is not None:
api_key = app_key.secret_key api_key = app_key.secret_key
else: else:
continue raise AppApiException(
500,
_('Application Key is required for application tool 【{name}】').format(name=app.name)
)
executor = ToolExecutor() executor = ToolExecutor()
app_config = executor.get_app_mcp_config(api_key) app_config = executor.get_app_mcp_config(api_key)
mcp_servers_config[app.name] = app_config mcp_servers_config[app.name] = app_config