Merge pull request #403 from webcoderz/main

fix pyproject bug

pyproject.toml

@@ -51,7 +51,7 @@ tests = [
     "langchain_community",
     "langchain_google_genai",
     "langchain_groq",
-    "langchain-ollama',
+    "langchain-ollama",
     "pytest-asyncio>=0.18.2,!=0.22.0,<0.23.0",
     "pytest-env>=0.8,<2.0",
     "pytest-rerunfailures>=10,<14",

src/controlflow/memory/providers/postgres.py

@@ -74,7 +74,7 @@ class PostgresMemory(MemoryProvider):
         description="Dimension of the embedding vectors. Match your model's output.",
     )
 
-    embedding_fn: Callable = Field(
+    embedding_fn: OpenAIEmbeddings = Field(
         default_factory=lambda: OpenAIEmbeddings(
             model="text-embedding-ada-002",
         ),
@@ -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.",
     )

tests/llm/test_models.py

@@ -46,11 +46,13 @@ def test_get_groq_model(monkeypatch):
     assert isinstance(model, ChatGroq)
     assert model.model_name == "mixtral-8x7b-32768"
 
+
 def test_get_ollama_model(monkeypatch):
     model = get_model("ollama/qwen2.5")
     assert isinstance(model, ChatOllama)
     assert model.model == "qwen2.5"
 
+
 def test_get_model_with_invalid_format():
     with pytest.raises(ValueError, match="The model `gpt-4o` is not valid."):
         get_model("gpt-4o")