From c10f72cf4c35e3dcec4b46f93ad274db9fd9bb8a Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Thu, 21 Dec 2023 18:36:02 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=86=85=E5=AE=B9=E5=87=BD=E6=95=B0?= =?UTF-8?q?=E8=B0=83=E7=94=A8=E6=8A=A5=E5=91=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/openai/api/chat_completion.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/pkg/openai/api/chat_completion.py b/pkg/openai/api/chat_completion.py index e308f17d..1e0e1bc5 100644 --- a/pkg/openai/api/chat_completion.py +++ b/pkg/openai/api/chat_completion.py @@ -6,6 +6,8 @@ from openai.types.chat import chat_completion_message from .model import RequestBase from .. import funcmgr +from ...plugin import host +from ...utils import context class ChatCompletionRequest(RequestBase): @@ -189,6 +191,16 @@ class ChatCompletionRequest(RequestBase): ret = "error: execute function failed: {}".format(str(e)) logging.error("函数执行失败: {}".format(str(e))) + # 上报数据 + plugin_info = host.get_plugin_info_for_audit(func_name.split('-')[0]) + audit_func_name = func_name.split('-')[1] + audit_func_desc = funcmgr.get_func_schema(func_name)['description'] + context.get_center_v2_api().usage.post_function_record( + plugin=plugin_info, + function_name=audit_func_name, + function_description=audit_func_desc, + ) + self.append_message( role="function", content=json.dumps(ret, ensure_ascii=False),