diff options
Diffstat (limited to 'packages')
| -rw-r--r-- | packages/sdk/src/supabase-store.ts | 5 | ||||
| -rw-r--r-- | packages/sdk/src/types.ts | 2 |
2 files changed, 7 insertions, 0 deletions
diff --git a/packages/sdk/src/supabase-store.ts b/packages/sdk/src/supabase-store.ts index 7f69791..386c3cb 100644 --- a/packages/sdk/src/supabase-store.ts +++ b/packages/sdk/src/supabase-store.ts @@ -20,6 +20,7 @@ type MemoryRow = { tags: Tag[]; metadata: Record<string, unknown>; embedding: unknown; + embedding_dimensions: number | null; created_at: string; updated_at: string; }; @@ -88,6 +89,8 @@ export class SupabaseStore implements MemoryStore { if (input.embedding) { insertData.embedding = input.embedding; + insertData.embedding_dimensions = + input.embeddingDimensions ?? input.embedding.length; } const { data, error } = await this.client @@ -142,6 +145,8 @@ export class SupabaseStore implements MemoryStore { if (input.embedding !== undefined) { updates.embedding = input.embedding; + updates.embedding_dimensions = + input.embeddingDimensions ?? input.embedding.length; } const { data, error } = await this.client diff --git a/packages/sdk/src/types.ts b/packages/sdk/src/types.ts index 76a63de..4586ecd 100644 --- a/packages/sdk/src/types.ts +++ b/packages/sdk/src/types.ts @@ -42,6 +42,7 @@ export type MemoryCreateInput = { tags?: Tag[]; metadata?: MemoryMetadata; embedding?: number[]; + embeddingDimensions?: number; }; export type MemoryUpdateInput = { @@ -50,6 +51,7 @@ export type MemoryUpdateInput = { tags?: Tag[]; metadata?: MemoryMetadata; embedding?: number[]; + embeddingDimensions?: number; }; export type MemoryFilter = { |