Function main

examples/ptc/file-processor/main.go:18–174

文件批处理示例 演示如何使用 PTC 进行复杂的文件处理任务

()

// 文件批处理示例
// 演示如何使用 PTC 进行复杂的文件处理任务
func main() {
	// 检查 API Key
	apiKey := os.Getenv("ANTHROPIC_API_KEY")
	if apiKey == "" {
		log.Fatal("请设置环境变量 ANTHROPIC_API_KEY")
	}

	fmt.Println("=== Aster PTC 文件处理示例 ===")

	// 1. 创建工具生态系统
	registry := tools.NewRegistry()
	builtin.RegisterAll(registry)

	toolBridge := bridge.NewToolBridge(registry)
	codeExecTool := builtin.NewCodeExecuteToolWithBridge(toolBridge)

	// 2. 准备工具列表(启用 PTC)
	toolSchemas := []provider.ToolSchema{
		{
			Name:           "CodeExecute",
			Description:    codeExecTool.Description(),
			InputSchema:    codeExecTool.InputSchema(),
			AllowedCallers: []string{"direct"},
		},
	}

	// 添加可在 Python 中调用的工具
	allowedTools := []string{"Read", "Write", "Glob", "Grep", "Bash"}
	for _, toolName := range allowedTools {
		tool, _ := registry.Create(toolName, nil)
		toolSchemas = append(toolSchemas, provider.ToolSchema{
			Name:           toolName,
			Description:    tool.Description(),
			InputSchema:    tool.InputSchema(),
			AllowedCallers: []string{"direct", "code_execution_20250825"},
		})
	}

	// 3. 创建 Anthropic Provider
	providerConfig := &types.ModelConfig{
		Provider: "anthropic",
		Model:    "claude-3-5-sonnet-20241022",
		APIKey:   apiKey,
	}

	anthropicProvider, err := provider.NewAnthropicProvider(providerConfig)
	if err != nil {
		log.Fatalf("创建 Provider 失败: %v", err)
	}

	// 4. 定义复杂任务
	task := `请编写 Python 代码完成以下文件处理任务:

1. 使用 Glob 查找当前目录下所有 .go 文件
2. 使用 Grep 在这些文件中搜索包含 "TODO" 或 "FIXME" 的注释
3. 统计每个文件中待办事项的数量
4. 生成一份 Markdown 格式的报告,包含:
   - 总待办事项数量