mlogclub
5d7c10aeab
- Updated runtime configuration to use __CS_AI_AGENT_WIDGET_CONFIG__ instead of __CS_AGENT_WIDGET_CONFIG__. - Changed message types in support host bridge from "cs-agent" to "cs-ai-agent". - Minified SDK script updated to reflect new AI agent naming conventions. - Adjusted scrollbar styles in main.scss to use .cs-ai-agent-scrollbar instead of .cs-agent-scrollbar.
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package mcps
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"cs-ai-agent/internal/pkg/config"
|
|
"cs-ai-agent/internal/pkg/errorsx"
|
|
)
|
|
|
|
type RuntimeService struct {
|
|
client *Client
|
|
}
|
|
|
|
var Runtime = NewRuntimeService()
|
|
|
|
func NewRuntimeService() *RuntimeService {
|
|
return &RuntimeService{
|
|
client: NewClient(),
|
|
}
|
|
}
|
|
|
|
func (s *RuntimeService) CallTool(ctx context.Context, serverCode string, toolName string, arguments map[string]any) (*ToolCallResult, error) {
|
|
server, err := s.resolveServer(serverCode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s.client.CallTool(ctx, server, toolName, arguments)
|
|
}
|
|
|
|
func (s *RuntimeService) ListTools(ctx context.Context, serverCode string) ([]ToolInfo, error) {
|
|
server, err := s.resolveServer(serverCode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s.client.ListTools(ctx, server)
|
|
}
|
|
|
|
func (s *RuntimeService) resolveServer(serverCode string) (ServerConfig, error) {
|
|
cfg := config.Current()
|
|
if !cfg.MCP.Enabled {
|
|
return ServerConfig{}, errorsx.InvalidParam("MCP未启用")
|
|
}
|
|
serverCode = strings.TrimSpace(serverCode)
|
|
if serverCode == "" {
|
|
return ServerConfig{}, errorsx.InvalidParam("serverCode不能为空")
|
|
}
|
|
server, ok := cfg.MCP.Servers[serverCode]
|
|
if !ok {
|
|
return ServerConfig{}, errorsx.InvalidParam("MCP服务配置不存在")
|
|
}
|
|
if !server.Enabled {
|
|
return ServerConfig{}, errorsx.InvalidParam("MCP服务未启用")
|
|
}
|
|
return ServerConfig{
|
|
Code: serverCode,
|
|
Endpoint: strings.TrimSpace(server.Endpoint),
|
|
TimeoutMS: server.TimeoutMS,
|
|
Headers: cloneRuntimeHeaders(server.Headers),
|
|
}, nil
|
|
}
|
|
|
|
func cloneRuntimeHeaders(headers map[string]string) map[string]string {
|
|
if len(headers) == 0 {
|
|
return nil
|
|
}
|
|
ret := make(map[string]string, len(headers))
|
|
for key, value := range headers {
|
|
ret[key] = value
|
|
}
|
|
return ret
|
|
}
|