mlogclub
34051a4631
- Updated labels in the AI Agents dashboard for clarity, changing "流程状态" to "Playbook 状态" and "未发布流程" to "未发布 Playbook". - Introduced AI Agent rollout percentage management in channel editing, allowing users to set and rollback rollout percentages. - Added new API endpoints for rolling back AI Agent rollout and fetching agent run metrics. - Implemented new UI components for displaying agent run details, including status, duration, and input/output tokens. - Enhanced type definitions for AdminChannel and AIAgent to include rollout percentages and runtime modes. - Updated navigation to include a section for agent runs. - Added new translations for agent run features in both English and Chinese.
38 lines
1.4 KiB
Go
38 lines
1.4 KiB
Go
package einoexperiment
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"agent-desk/internal/ai/mcps"
|
|
runtimetooling "agent-desk/internal/ai/runtime/tooling"
|
|
aitooling "agent-desk/internal/ai/tooling"
|
|
)
|
|
|
|
// MCPToolExecutor is the narrow execution boundary used by the Eino
|
|
// experiment. The production MCP executor remains responsible for dynamic
|
|
// registry resolution, policy enforcement, timeout, and transport lifecycle.
|
|
type MCPToolExecutor interface {
|
|
Execute(context.Context, string, map[string]any, aitooling.Policy) (aitooling.Definition, *mcps.ToolCallResult, error)
|
|
}
|
|
|
|
// NewMCPToolHandler adapts a dynamically discovered MCP tool to GuardedTool.
|
|
// Callers must still configure GuardedTool.Definition and Policy so its
|
|
// pre-handler guard provides a deterministic rejection before MCP transport.
|
|
func NewMCPToolHandler(executor MCPToolExecutor, toolCode string, policy aitooling.Policy) ToolHandler {
|
|
return func(ctx context.Context, arguments map[string]any) (string, error) {
|
|
if executor == nil {
|
|
return "", fmt.Errorf("eino experiment MCP executor is required")
|
|
}
|
|
definition, result, err := executor.Execute(ctx, strings.TrimSpace(toolCode), arguments, policy)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if definition.Code == "" {
|
|
return "", fmt.Errorf("MCP executor returned an empty tool definition")
|
|
}
|
|
return runtimetooling.BuildReducedToolResultSummary(result), nil
|
|
}
|
|
}
|