mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-09-06 08:16:03 +00:00
196609650e
- Add withSupermemory wrapper section to OpenAI SDK docs - Create new Claude Memory Tool integration page - Replace gpt-4o with gpt-5 across documentation - Fix normalizePathToCustomId in claude-memory.ts (customId validation) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
100 lines
2.6 KiB
TypeScript
100 lines
2.6 KiB
TypeScript
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)
|