t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
"sync"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/contract"
|
|
)
|
|
|
|
type platformAIService struct {
|
|
mu sync.RWMutex
|
|
provider contract.PlatformAIProvider
|
|
}
|
|
|
|
var PlatformAIService = &platformAIService{}
|
|
|
|
func SetPlatformAIProvider(provider contract.PlatformAIProvider) {
|
|
PlatformAIService.mu.Lock()
|
|
defer PlatformAIService.mu.Unlock()
|
|
PlatformAIService.provider = provider
|
|
}
|
|
|
|
func (s *platformAIService) ModelSource(ctx context.Context) (string, error) {
|
|
provider := s.current()
|
|
if provider == nil {
|
|
return contract.ModelSourceCustom, nil
|
|
}
|
|
source, err := provider.ModelSource(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if strings.EqualFold(strings.TrimSpace(source), contract.ModelSourcePlatform) {
|
|
return contract.ModelSourcePlatform, nil
|
|
}
|
|
return contract.ModelSourceCustom, nil
|
|
}
|
|
|
|
func (s *platformAIService) IsPlatform(ctx context.Context) (bool, error) {
|
|
source, err := s.ModelSource(ctx)
|
|
return source == contract.ModelSourcePlatform, err
|
|
}
|
|
|
|
func (s *platformAIService) Config(ctx context.Context) (*contract.PlatformAIConfig, error) {
|
|
provider := s.current()
|
|
if provider == nil {
|
|
return nil, errors.New("platform AI provider is not initialized")
|
|
}
|
|
return provider.Config(ctx)
|
|
}
|
|
|
|
func (s *platformAIService) Status(ctx context.Context) (*contract.PlatformAIStatus, error) {
|
|
provider := s.current()
|
|
if provider == nil {
|
|
return nil, errors.New("platform AI provider is not initialized")
|
|
}
|
|
return provider.Status(ctx)
|
|
}
|
|
|
|
func (s *platformAIService) current() contract.PlatformAIProvider {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.provider
|
|
}
|