import "dotenv/config"
import { openai } from "@ai-sdk/openai"
import { createAnthropic } from "@ai-sdk/anthropic"
import {
  withSupermemory,
  supermemoryTools,
  searchMemoriesTool,
  addMemoryTool,
  type MemoryPromptData,
} from "@supermemory/tools/ai-sdk"

async function testMiddleware() {
  console.log("=== Middleware ===")

  // Basic wrapper
  const model = withSupermemory(openai("gpt-4"), "user-123")
  console.log("✓ withSupermemory basic")

  // With addMemory option
  const modelWithAdd = withSupermemory(openai("gpt-4"), "user-123", {
    addMemory: "always",
  })
  console.log("✓ withSupermemory with addMemory")

  // With verbose logging
  const modelVerbose = withSupermemory(openai("gpt-4"), "user-123", {
    verbose: true,
  })
  console.log("✓ withSupermemory with verbose")
}

async function testSearchModes() {
  console.log("\n=== Search Modes ===")

  const profileModel = withSupermemory(openai("gpt-4"), "user-123", { mode: "profile" })
  console.log("✓ mode: profile")

  const queryModel = withSupermemory(openai("gpt-4"), "user-123", { mode: "query" })
  console.log("✓ mode: query")

  const fullModel = withSupermemory(openai("gpt-4"), "user-123", { mode: "full" })
  console.log("✓ mode: full")
}

async function testCustomPrompt() {
  console.log("\n=== Custom Prompt Template ===")

  const anthropic = createAnthropic({ apiKey: "test-key" })

  const claudePrompt = (data: MemoryPromptData) =>
    `
<context>
  <user_profile>${data.userMemories}</user_profile>
  <relevant_memories>${data.generalSearchMemories}</relevant_memories>
</context>
`.trim()

  const model = withSupermemory(anthropic("claude-3-sonnet-20240229"), "user-123", {
    mode: "full",
    promptTemplate: claudePrompt,
  })
  console.log("✓ Custom prompt template")
}

async function testTools() {
  console.log("\n=== Memory Tools ===")

  // All tools
  const tools = supermemoryTools("YOUR_API_KEY")
  console.log("✓ supermemoryTools")

  // Individual tools
  const searchTool = searchMemoriesTool("API_KEY", { projectId: "personal" })
  console.log("✓ searchMemoriesTool")

  const addTool = addMemoryTool("API_KEY")
  console.log("✓ addMemoryTool")

  // Combined
  const toolsObj = {
    searchMemories: searchTool,
    addMemory: addTool,
  }
  console.log("✓ Combined tools object")
}

async function main() {
  console.log("AI SDK Integration Tests")
  console.log("========================\n")

  await testMiddleware()
  await testSearchModes()
  await testCustomPrompt()
  await testTools()

  console.log("\n========================")
  console.log("✅ All AI SDK tests passed!")
}

main().catch(console.error)