set defaultEmbeddingDims if MigrateEmbeddingDims dimensions is 0 (#257)

getzep · Nov 1, 2023 · 88d7882 · 88d7882
1 parent 5dc5082
commit 88d7882
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/store/postgres/schema.go b/pkg/store/postgres/schema.go
@@ -25,6 +25,8 @@ import (
 	"github.com/uptrace/bun"
 )
 
+const defaultEmbeddingDims = 1536
+
 var maxOpenConns = 4 * runtime.GOMAXPROCS(0)
 
 type SessionSchema struct {
@@ -603,6 +605,11 @@ func MigrateEmbeddingDims(
 	tableName string,
 	dimensions int,
 ) error {
+	// we may be missing a config key, so use the default dimensions if none are provided
+	if dimensions == 0 {
+		dimensions = defaultEmbeddingDims
+	}
+
 	tx, err := db.BeginTx(ctx, nil)
 	if err != nil {
 		return fmt.Errorf("MigrateEmbeddingDims error starting transaction: %w", err)