AntSK/src/AntSK/Pages/AppPage/AddApp.razor.cs

using AntDesign;
using AntSK.Domain.Domain.Model.Constant;
using AntSK.Domain.Domain.Model.Enum;
using AntSK.Domain.Domain.Service;
using AntSK.Domain.Repositories;
using AntSK.Domain.Utils;
using Microsoft.AspNetCore.Components;
using Microsoft.SemanticKernel;

namespace AntSK.Pages.AppPage
{
    public partial class AddApp
    {
        [Parameter]
        public string AppId { get; set; }

        [Inject]
        protected IApps_Repositories _apps_Repositories { get; set; }

        [Inject]
        protected IKmss_Repositories _kmss_Repositories { get; set; }

        [Inject]
        protected IApis_Repositories _apis_Repositories { get; set; }
        [Inject]
        protected NavigationManager NavigationManager { get; set; }
        [Inject]
        protected MessageService? Message { get; set; }
        [Inject]
        protected IAIModels_Repositories _aimodels_Repositories { get; set; }
        [Inject]
        protected FunctionService _functionService { get; set; }

        private Apps _appModel = new Apps();

        IEnumerable<string> kmsIds;

        private List<Kmss> _kmsList = new List<Kmss>();

        IEnumerable<string> apiIds = [];

        private List<Apis> _apiList = new List<Apis>();

        IEnumerable<string> funIds = [];

        public Dictionary<string, string> _funList = new Dictionary<string, string>();

        private List<AIModels> _chatList;
        private List<AIModels> _embedingList;
        private List<AIModels> _rerankList;
        protected override async Task OnInitializedAsync()
        {
            await base.OnInitializedAsync();
            _kmsList = _kmss_Repositories.GetList();
            _apiList = _apis_Repositories.GetList();
            var models=_aimodels_Repositories.GetList();
            _chatList = models.Where(p => p.AIModelType == AIModelType.Chat).ToList();
            _embedingList = models.Where(p => p.AIModelType == AIModelType.Embedding).ToList();
            _rerankList = models.Where(p => p.AIModelType == AIModelType.Rerank).ToList();


            _functionService.SearchMarkedMethods();
            foreach (var func in _functionService.Functions)
            {
                var methodInfo = _functionService.MethodInfos[func.Key];
                _funList.Add(func.Key, methodInfo.Description);
            }


            if (!string.IsNullOrEmpty(AppId))
            {
                //查看
                _appModel = _apps_Repositories.GetFirst(p => p.Id == AppId);
                kmsIds = _appModel.KmsIdList?.Split(",");
                apiIds = _appModel.ApiFunctionList?.Split(",");
                funIds = _appModel.NativeFunctionList?.Split(",");
            }


        }
        private void HandleSubmit()
        {
            if (kmsIds != null && kmsIds.Count() > 0)
            {
                var kmsList = _kmss_Repositories.GetList(p => kmsIds.Contains(p.Id));
                bool allSameEmbeddingModelID = kmsList.Select(k => k.EmbeddingModelID).Distinct().Count() == 1;
                if (!allSameEmbeddingModelID)
                {
                    _ = Message.Error("同一个应用的知识库的Embedding模型必须相同！", 2);
                    return;
                }
                _appModel.KmsIdList = string.Join(",", kmsIds);
            }
            if (_appModel.Type == AppType.kms.ToString())
            {
                if (string.IsNullOrEmpty(_appModel.Prompt)|| !_appModel.Prompt.Contains("{{$doc}}") || !_appModel.Prompt.Contains("{{$input}}"))
                {
                    _ = Message.Error("知识库提示词必须包含 {{$doc}} 和 {{$input}}", 2);
                    return;
                }
            }
            if (apiIds.IsNotNull())
            {
                _appModel.ApiFunctionList = string.Join(",", apiIds);
            }
            if (funIds.IsNotNull())
            {

                _appModel.NativeFunctionList = string.Join(",", funIds);
            }
 
            if (string.IsNullOrEmpty(AppId))
            {
                //新增
                _appModel.Id = Guid.NewGuid().ToString();
                //秘钥
                _appModel.SecretKey = "sk-" + Guid.NewGuid().ToString();
                if (_apps_Repositories.IsAny(p => p.Name == _appModel.Name))
                {
                    _ = Message.Error("名称已存在！", 2);
                    return;
                }


                _apps_Repositories.Insert(_appModel);
            }
            else
            {
                //修改  
                _apps_Repositories.Update(_appModel);
            }

            //NavigationManager.NavigateTo($"/app/detail/{_appModel.Id}");
            NavigationManager.NavigateTo($"/applist");
        }


        private void Back()
        {
            NavigationManager.NavigateTo("/applist");
        }

        private void NavigateModelList()
        {
            NavigationManager.NavigateTo("/modelmanager/modellist");
        }

        private void NavigateKmsList()
        {
            NavigationManager.NavigateTo("/KmsList");
        }


        private void OnAppTypeChange(string value)
        {
            if (value == AppType.kms.ToString() && string.IsNullOrEmpty( _appModel.Prompt))
            {
                _appModel.Prompt = KmsConstantcs.KmsPrompt;
            }

            if (value == AppType.chat.ToString())
            {
                _appModel.Prompt = "";
            }
        }
    }

}