Function main

examples/router/main.go:21–123

()

)

func main() {
	ctx := context.Background()

	// 简单的JSON存储 + 本地沙箱
	memStore, err := store.NewJSONStore("./.aster-router")
	if err != nil {
		log.Fatalf("Failed to create store: %v", err)
	}
	sbFactory := sandbox.NewFactory()

	// Provider 工厂
	providerFactory := provider.NewMultiProviderFactory()

	// 模板注册：同一个模板 ID，但可以通过 RoutingProfile 决定实际模型
	tplRegistry := agent.NewTemplateRegistry()
	tplRegistry.Register(&types.AgentTemplateDefinition{
		ID:           "router-demo",
		SystemPrompt: "You are a helpful assistant.",
		Model:        "claude-3-5-sonnet-20241022",
		Tools:        []string{},
	})

	// 构造静态路由器：
	// - quality-first: 使用 anthropic 的较强模型（示例）
	// - cost-first: 使用 deepseek 的模型（示例）
	defaultModel := &types.ModelConfig{
		Provider: "anthropic",
		Model:    "claude-3-5-sonnet-20241022",
	}

	routes := []router.StaticRouteEntry{
		{
			Task:     "chat",
			Priority: router.PriorityQuality,
			Model: &types.ModelConfig{
				Provider: "anthropic",
				Model:    "claude-3-5-sonnet-20241022",
			},
		},
		{
			Task:     "chat",
			Priority: router.PriorityCost,
			Model: &types.ModelConfig{
				Provider: "deepseek",
				Model:    "deepseek-chat",
			},
		},
	}

	rt := router.NewStaticRouter(defaultModel, routes)

	deps := &agent.Dependencies{
		Store:            memStore,
		SandboxFactory:   sbFactory,
		ToolRegistry:     toolsRegistry(),
		ProviderFactory:  providerFactory,
		Router:           rt,
		TemplateRegistry: tplRegistry,