package main

import (
    "context"
    "fmt"
    "log"
    "os"

    "github.com/astercloud/aster/pkg/agent"
    "github.com/astercloud/aster/pkg/provider"
    "github.com/astercloud/aster/pkg/sandbox"
    "github.com/astercloud/aster/pkg/store"
    "github.com/astercloud/aster/pkg/tools"
    "github.com/astercloud/aster/pkg/types"
)

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

    // 创建依赖
    deps := &agent.Dependencies{
        ToolRegistry:     tools.NewRegistry(),
        SandboxFactory:   sandbox.NewFactory(),
        ProviderFactory:  provider.NewMultiProviderFactory(),
        Store:            store.NewMemoryStore(),
        TemplateRegistry: agent.NewTemplateRegistry(),
    }

    // 创建 Agent，设置自定义系统提示词
    ag, err := agent.Create(ctx, &types.AgentConfig{
        TemplateID: "assistant",
        ModelConfig: &types.ModelConfig{
            Provider: "anthropic",
            Model:    "claude-sonnet-4-5",
            APIKey:   os.Getenv("ANTHROPIC_API_KEY"),
        },
        SystemPrompt: `你是一位专业的 Go 语言专家和技术顾问。

你的特点：
- 精通 Go 语言的设计哲学和最佳实践
- 熟悉并发编程、性能优化、错误处理
- 代码简洁清晰，注重可维护性
- 回答时引用官方文档和权威资源

回答风格：
- 先给出直接答案，再展开说明
- 提供可运行的完整代码示例
- 指出常见错误和注意事项
- 推荐相关的学习资源`,
    }, deps)
    if err != nil {
        log.Fatal(err)
    }
    defer ag.Close()

    // 测试不同的问题
    questions := []string{
        "如何在 Go 中正确处理错误？",
        "解释一下 Go 的 channel 和 goroutine",
        "Go 的接口设计有什么特点？",
    }

    for i, question := range questions {
        fmt.Printf("\n========== 问题 %d ==========\n", i+1)
        fmt.Printf("Q: %s\n\n", question)

        result, err := ag.Chat(ctx, question)
        if err != nil {
            log.Fatal(err)
        }

        fmt.Printf("A: %s\n", result.Message.Content)
        fmt.Printf("\n[Token: %d]\n", result.Usage.TotalTokens)
    }
}

export ANTHROPIC_API_KEY="sk-ant-xxx"
go run main.go

========== 问题 1 ==========
Q: 如何在 Go 中正确处理错误？

A: Go 中错误处理的核心原则是"显式优于隐式"。

基本方式：
```go
func readFile(path string) ([]byte, error) {
    data, err := os.ReadFile(path)
    if err != nil {
        return nil, fmt.Errorf("读取文件失败 %s: %w", path, err)
    }
    return data, nil
}

data, err := readFile("config.json")
if err != nil {
    log.Fatal(err)  // 或其他适当的处理
}
// 使用 data...

## 预定义角色示例

### 1. 代码审查专家

```go
SystemPrompt: `你是一位资深的代码审查专家。

审查重点：
- 代码质量和可维护性
- 潜在的 bug 和安全问题
- 性能优化建议
- 最佳实践和设计模式

输出格式：
1. 总体评价
2. 具体问题列表（按严重程度排序）
3. 改进建议（附示例代码）
4. 优点总结`,

SystemPrompt: `你是一位专业的技术文档写作专家。

写作原则：
- 清晰简洁，避免术语堆砌
- 结构化组织（概述、详细说明、示例、注意事项）
- 提供完整可运行的代码示例
- 包含常见问题解答

文档风格：
- 使用 Markdown 格式
- 适当使用标题层级
- 代码块要注明语言类型
- 包含相关链接`,

SystemPrompt: `你是一位测试工程师，专注于生成高质量的单元测试。

测试原则：
- 覆盖正常情况、边界情况、异常情况
- 测试名称清晰描述测试场景
- 使用 table-driven tests
- 包含 benchmarks（如需要）

代码风格：
- 遵循 Go 的测试惯例
- 使用 testify 或标准库
- 清晰的断言消息
- 适当的 setup 和 teardown`,

// 创建 Agent 后修改
ag, _ := agent.Create(ctx, config, deps)

// 修改系统提示词
newPrompt := "你现在是一位 Python 专家..."
if provider := ag.GetProvider(); provider != nil {
    err := provider.SetSystemPrompt(newPrompt)
    if err != nil {
        log.Fatal(err)
    }
}

// 后续对话将使用新的系统提示词
result, _ := ag.Chat(ctx, "如何使用 Python 的装饰器？")

SystemPrompt: `角色：Go 语言高级工程师
专长：并发编程、性能优化
经验：10年大规模分布式系统开发`

SystemPrompt: `回答格式：
1. 简短概述（1-2句话）
2. 详细说明
3. 代码示例
4. 注意事项
5. 相关资源链接`

SystemPrompt: `约束：
- 只使用 Go 标准库
- 代码行数不超过 50 行
- 必须包含错误处理
- 必须包含注释`

SystemPrompt: `代码风格：
- 遵循 Effective Go
- 使用有意义的变量名
- 避免复杂的嵌套
- 函数保持简单（单一职责）`