From 18bfe312fa6ce7458b862ccff80b67e035f8da92 Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Wed, 24 Sep 2025 17:59:44 +0800 Subject: [PATCH] perf: open web ui --- apps/accounts/api/account/account.py | 10 +++++++ apps/assets/const/gpt.py | 39 ++++++++++++++++++++++++++++ 2 files changed, 49 insertions(+) diff --git a/apps/accounts/api/account/account.py b/apps/accounts/api/account/account.py index bf9a76afb..95cec3f31 100644 --- a/apps/accounts/api/account/account.py +++ b/apps/accounts/api/account/account.py @@ -11,6 +11,7 @@ from accounts.const import ChangeSecretRecordStatusChoice from accounts.filters import AccountFilterSet, NodeFilterBackend from accounts.mixins import AccountRecordViewLogMixin from accounts.models import Account, ChangeSecretRecord +from assets.const.gpt import create_or_update_chatx_resources from assets.models import Asset, Node from authentication.permissions import UserConfirmation, ConfirmType from common.api.mixin import ExtraFilterFieldsMixin @@ -18,6 +19,7 @@ from common.drf.filters import AttrRulesFilterBackend from common.permissions import IsValidUser from common.utils import lazyproperty, get_logger from orgs.mixins.api import OrgBulkModelViewSet +from orgs.utils import tmp_to_root_org from rbac.permissions import RBACPermission logger = get_logger(__file__) @@ -43,6 +45,7 @@ class AccountViewSet(OrgBulkModelViewSet): 'clear_secret': 'accounts.change_account', 'move_to_assets': 'accounts.delete_account', 'copy_to_assets': 'accounts.add_account', + 'chat': 'accounts.view_account', } export_as_zip = True @@ -152,6 +155,13 @@ class AccountViewSet(OrgBulkModelViewSet): def copy_to_assets(self, request, *args, **kwargs): return self._copy_or_move_to_assets(request, move=False) + @action(methods=['get'], detail=False, url_path='chat') + def chat(self, request, *args, **kwargs): + with tmp_to_root_org(): + __, account = create_or_update_chatx_resources() + serializer = self.get_serializer(account) + return Response(serializer.data) + class AccountSecretsViewSet(AccountRecordViewLogMixin, AccountViewSet): """ diff --git a/apps/assets/const/gpt.py b/apps/assets/const/gpt.py index b3b078031..58bfa2e64 100644 --- a/apps/assets/const/gpt.py +++ b/apps/assets/const/gpt.py @@ -1,5 +1,6 @@ from django.utils.translation import gettext_lazy as _ +from orgs.models import Organization from .base import BaseType @@ -52,3 +53,41 @@ class GPTTypes(BaseType): return [ cls.CHATGPT, ] + + +CHATX_NAME = 'ChatX' + + +def create_or_update_chatx_resources(chatx_name=CHATX_NAME, org_id=Organization.SYSTEM_ID): + from django.apps import apps + + platform_model = apps.get_model('assets', 'Platform') + asset_model = apps.get_model('assets', 'Asset') + account_model = apps.get_model('accounts', 'Account') + + platform, __ = platform_model.objects.get_or_create( + name=chatx_name, + defaults={ + 'internal': True, + 'type': chatx_name, + 'category': 'ai', + } + ) + asset, __ = asset_model.objects.get_or_create( + address=chatx_name, + defaults={ + 'name': chatx_name, + 'platform': platform, + 'org_id': org_id + } + ) + + account, __ = account_model.objects.get_or_create( + username=chatx_name, + defaults={ + 'name': chatx_name, + 'asset': asset, + 'org_id': org_id + } + ) + return asset, account