Function main

examples/ptc/basic/main.go:18–168

PTC 基础示例 演示如何使用 Programmatic Tool Calling 让 LLM 生成 Python 代码并调用工具

()

// PTC 基础示例
// 演示如何使用 Programmatic Tool Calling 让 LLM 生成 Python 代码并调用工具
func main() {
	// 检查 API Key
	apiKey := os.Getenv("ANTHROPIC_API_KEY")
	if apiKey == "" {
		log.Fatal("请设置环境变量 ANTHROPIC_API_KEY")
	}

	// 1. 创建工具注册表并注册内置工具
	registry := tools.NewRegistry()
	builtin.RegisterAll(registry)

	// 2. 创建 ToolBridge 用于程序化工具调用
	toolBridge := bridge.NewToolBridge(registry)

	// 3. 创建 CodeExecute 工具并启用 PTC 支持
	codeExecTool := builtin.NewCodeExecuteToolWithBridge(toolBridge)

	// 可选: 自定义桥接服务器地址
	// codeExecTool.SetBridgeURL("http://localhost:9000")

	// 4. 创建工具实例
	readTool, _ := registry.Create("Read", nil)
	writeTool, _ := registry.Create("Write", nil)
	globTool, _ := registry.Create("Glob", nil)
	bashTool, _ := registry.Create("Bash", nil)

	// 5. 转换为 Provider ToolSchema (添加 AllowedCallers 支持)
	toolSchemas := []provider.ToolSchema{
		{
			Name:        "CodeExecute",
			Description: codeExecTool.Description(),
			InputSchema: codeExecTool.InputSchema(),
			// CodeExecute 只能被 LLM 直接调用
			AllowedCallers: []string{"direct"},
		},
		{
			Name:        "Read",
			Description: readTool.Description(),
			InputSchema: readTool.InputSchema(),
			// Read 可以被 LLM 直接调用,也可以在 Python 代码中调用
			AllowedCallers: []string{"direct", "code_execution_20250825"},
		},
		{
			Name:           "Write",
			Description:    writeTool.Description(),
			InputSchema:    writeTool.InputSchema(),
			AllowedCallers: []string{"direct", "code_execution_20250825"},
		},
		{
			Name:           "Glob",
			Description:    globTool.Description(),
			InputSchema:    globTool.InputSchema(),
			AllowedCallers: []string{"direct", "code_execution_20250825"},
		},
		{
			Name:           "Bash",
			Description:    bashTool.Description(),
			InputSchema:    bashTool.InputSchema(),