add props interface export

This commit is contained in:
Mahesh Sanikommmu 2025-10-22 12:31:54 -07:00
parent 8436229cbb
commit 5fb33743b3
3 changed files with 10 additions and 9 deletions

View file

@ -1,7 +1,7 @@
{
"name": "@supermemory/tools",
"type": "module",
"version": "1.2.15",
"version": "1.2.16",
"description": "Memory tools for AI SDK and OpenAI function calling with supermemory",
"scripts": {
"build": "tsdown",

View file

@ -2,6 +2,13 @@ import type { LanguageModelV2 } from "@ai-sdk/provider"
import { wrapLanguageModel } from "ai"
import { createSupermemoryMiddleware } from "./middleware"
interface WrapVercelLanguageModelOptions {
conversationId?: string;
verbose?: boolean;
mode?: "profile" | "query" | "full";
addMemory?: "always" | "never";
}
/**
* Wraps a language model with supermemory middleware to automatically inject relevant memories
* into the system prompt based on the user's message content.
@ -43,12 +50,7 @@ import { createSupermemoryMiddleware } from "./middleware"
const wrapVercelLanguageModel = (
model: LanguageModelV2,
containerTag: string,
options?: {
conversationId?: string;
verbose?: boolean;
mode?: "profile" | "query" | "full";
addMemory?: "always" | "never";
},
options?: WrapVercelLanguageModelOptions,
): LanguageModelV2 => {
const SUPERMEMORY_API_KEY = process.env.SUPERMEMORY_API_KEY
@ -69,4 +71,4 @@ const wrapVercelLanguageModel = (
return wrappedModel
}
export { wrapVercelLanguageModel as withSupermemory }
export { wrapVercelLanguageModel as withSupermemory, type WrapVercelLanguageModelOptions as WithSupermemoryOptions }

View file

@ -70,7 +70,6 @@ const addSystemPrompt = async (
const queryText =
mode !== "profile"
? params.prompt
// Create a shallow copy before reversing to avoid mutating the original array
.slice().reverse()
.find((prompt) => prompt.role === "user")
?.content?.filter((content) => content.type === "text")