diff --git a/api/apps/sdk/session.py b/api/apps/sdk/session.py index bdc1260de9..8df1812b83 100644 --- a/api/apps/sdk/session.py +++ b/api/apps/sdk/session.py @@ -816,6 +816,29 @@ def chatbot_completions(dialog_id): return get_result(data=answer) +@manager.route("/chatbots//info", methods=["GET"]) # noqa: F821 +def chatbots_inputs(dialog_id): + token = request.headers.get("Authorization").split() + if len(token) != 2: + return get_error_data_result(message='Authorization is not valid!"') + token = token[1] + objs = APIToken.query(beta=token) + if not objs: + return get_error_data_result(message='Authentication error: API key is invalid!"') + + e, dialog = DialogService.get_by_id(dialog_id) + if not e: + return get_error_data_result(f"Can't find dialog by ID: {dialog_id}") + + return get_result( + data={ + "title": dialog.name, + "avatar": dialog.icon, + "prologue": dialog.prompt_config.get("prologue", ""), + } + ) + + @manager.route("/agentbots//completions", methods=["POST"]) # noqa: F821 def agent_bot_completions(agent_id): req = request.json