Update postgres.py

webcoderz · webcoderz · commit bf63a7f6f90a · 2025-01-28T12:43:16.000-05:00
diff --git a/src/controlflow/memory/providers/postgres.py b/src/controlflow/memory/providers/postgres.py
@@ -263,7 +263,7 @@ class AsyncPostgresMemory(AsyncMemoryProvider):
         description="Dimension of the embedding vectors. Must match your model output size.",
     )
 
-    embedding_fn: Callable = Field(
+    embedding_fn: OpenAIEmbeddings = Field(
         default_factory=lambda: OpenAIEmbeddings(model="text-embedding-ada-002"),
         description="Function that turns a string into a numeric vector.",
     )

Original file line number	Diff line number	Diff line change
`@@ -263,7 +263,7 @@ class AsyncPostgresMemory(AsyncMemoryProvider):`
`263`	`263`	`description="Dimension of the embedding vectors. Must match your model output size.",`
`264`	`264`	`)`
`265`	`265`
`266`		`- embedding_fn: Callable = Field(`
	`266`	`+ embedding_fn: OpenAIEmbeddings = Field(`
`267`	`267`	`default_factory=lambda: OpenAIEmbeddings(model="text-embedding-ada-002"),`
`268`	`268`	`description="Function that turns a string into a numeric vector.",`
`269`	`269`	`)`