package main

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

	"go-tiny-claw/internal/engine"
	"go-tiny-claw/internal/schema"
	"go-tiny-claw/internal/provider"
)


// 伪造的工具注册表 (用于测试 Provider 的工具提取能力)
type mockRegistry struct{}

func (m *mockRegistry) GetAvailableTools() []schema.ToolDefinition {
    return []schema.ToolDefinition{
        {
            Name:        "get_weather",
            Description: "获取指定城市的当前天气情况。",
            InputSchema: map[string]interface{}{
                "type": "object",
                "properties": map[string]interface{}{
                    "city": map[string]interface{}{
                        "type": "string",
                    },
                },
                "required": []string{"city"},
            },
        },
    }
}

func (m *mockRegistry) Execute(ctx context.Context, call schema.ToolCall) schema.ToolResult {
    log.Printf("  -> [Mock 工具执行] 获取 %s 的天气中...\n", call.Name)
    return schema.ToolResult{
        ToolCallID: call.ID,
        Output:     "API 返回：今天是晴天，气温 25 度。",
        IsError:    false,
    }
}

func main() {
 
	fmt.Printf("11111")
    workDir, _ := os.Getwd()

    // 1. 初始化真实的 Provider大脑 (指向智谱 GLM-4.5)
    // 这里你可以任意切换 NewZhipuClaudeProvider 或 NewZhipuOpenAIProvider，效果完全一致！
    llmProvider := provider.DeepseekOpenAIProvider("deepseek-v4-flash")
    // 2. 注入伪造的工具注册表
    registry := &mockRegistry{}

    // 3. 实例化并运行引擎，开启 EnableThinking = true (开启慢思考阶段！)
    eng := engine.NewAgentEngine(llmProvider, registry, workDir, true)

    // 设定测试任务
    prompt := "我想去北京跑步，帮我查查天气适合吗？"

    err := eng.Run(context.Background(), prompt)
    if err != nil {
        log.Fatalf("引擎运行崩溃: %v", err)
    }
}