@mastra/memory 0.0.2-alpha.2 → 0.0.2-alpha.3

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/memory
 
+## 0.0.2-alpha.3
+
+### Patch Changes
+
+- 836f4e3: Fixed some issues with memory, added Upstash as a memory provider. Silenced dev logs in core
+- Updated dependencies [836f4e3]
+  - @mastra/core@0.1.27-alpha.24
+
 ## 0.0.2-alpha.2
 
 ### Patch Changes

package/dist/index.d.ts

@@ -1 +1,2 @@
 export * from './postgres';
+export * from './kv/upstash';

package/dist/kv/upstash.d.ts

@@ -0,0 +1,71 @@
+import { MastraMemory, MessageType as BaseMastraMessageType, ThreadType, MessageResponse } from '@mastra/core';
+import { Redis } from '@upstash/redis';
+import { ToolResultPart, Message as AiMessage } from 'ai';
+interface MessageType extends BaseMastraMessageType {
+    tokens?: number;
+}
+interface SerializedThreadType extends Omit<ThreadType, 'createdAt' | 'updatedAt'> {
+    createdAt: string;
+    updatedAt: string;
+}
+export declare class UpstashKVMemory extends MastraMemory {
+    private prefix;
+    kv: Redis;
+    constructor(config: {
+        url: string;
+        token: string;
+        prefix?: string;
+        maxTokens?: number;
+    });
+    private getThreadKey;
+    private getMessagesKey;
+    private getToolCacheKey;
+    getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<ThreadType | null>;
+    getThreadsByResourceId({ resourceid }: {
+        resourceid: string;
+    }): Promise<ThreadType[]>;
+    saveThread({ thread }: {
+        thread: ThreadType;
+    }): Promise<ThreadType>;
+    updateThread(id: string, title: string, metadata: Record<string, unknown>): Promise<ThreadType>;
+    deleteThread(id: string): Promise<void>;
+    /**
+     * Tool Cache
+     */
+    validateToolCallArgs({ hashedArgs }: {
+        hashedArgs: string;
+    }): Promise<boolean>;
+    getToolResult({ threadId, toolArgs, toolName, }: {
+        threadId: string;
+        toolArgs: Record<string, unknown>;
+        toolName: string;
+    }): Promise<ToolResultPart['result'] | null>;
+    getContextWindow<T extends 'raw' | 'core_message'>({ threadId, startDate, endDate, format, }: {
+        format?: T;
+        threadId: string;
+        startDate?: Date;
+        endDate?: Date;
+    }): Promise<MessageResponse<T>>;
+    /**
+     * Messages
+     */
+    getMessages({ threadId }: {
+        threadId: string;
+    }): Promise<{
+        messages: MessageType[];
+        uiMessages: AiMessage[];
+    }>;
+    saveMessages({ messages }: {
+        messages: MessageType[];
+    }): Promise<MessageType[]>;
+    deleteMessage(id: string): Promise<void>;
+    /**
+     * Cleanup
+     */
+    drop(): Promise<void>;
+    parseThread(thread: SerializedThreadType): ThreadType;
+    parseMessages(messages: MessageType[]): MessageType[];
+}
+export {};