mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-08-28 05:25:33 +00:00
add props interface export
This commit is contained in:
parent
8436229cbb
commit
5fb33743b3
3 changed files with 10 additions and 9 deletions
|
|
@ -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",
|
||||
|
|
|
|||
|
|
@ -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 }
|
||||
|
|
|
|||
|
|
@ -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")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue