Skip to content

Commit

Permalink
updating import messages and reverting pyproject.toml
Browse files Browse the repository at this point in the history
  • Loading branch information
webcoderz committed Jan 10, 2025
1 parent 95a22a9 commit 4be056b
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 11 deletions.
9 changes: 0 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,6 @@ dependencies = [
"tiktoken>=0.7.0",
"typer>=0.10",
"ipython>=8.18.1",
"sqlalchemy>=2.0.36",
"pgvector>=0.3.6",
"lancedb>=0.17.0",
"python-dotenv>=1.0.1",
"sqlalchemy-utils>=0.41.2",
"psycopg-pool>=3.2.4",
"psycopg-binary>=3.2.3",
"psycopg>=3.2.3",
"psycopg2-binary>=2.9.10",
]
readme = "README.md"
requires-python = ">= 3.9"
Expand Down
3 changes: 2 additions & 1 deletion src/controlflow/memory/async_memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,8 @@ def get_memory_provider(provider: str) -> AsyncMemoryProvider:
import sqlalchemy
except ImportError:
raise ImportError(
"To use Postgres as a memory provider, please install the `sqlalchemy` package."
"""To use async Postgres as a memory provider, please install the `sqlalchemy, `psycopg-pool`,
`psycopg-binary`, and `psycopg` packages."""
)

import controlflow.memory.providers.postgres as postgres_providers
Expand Down
3 changes: 2 additions & 1 deletion src/controlflow/memory/memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,8 @@ def get_memory_provider(provider: str) -> MemoryProvider:
import sqlalchemy
except ImportError:
raise ImportError(
"To use Postgres as a memory provider, please install the `sqlalchemy` package."
"""To use Postgres as a memory provider, please install the `sqlalchemy, `psycopg-pool`,
`psycopg-binary`, and `psycopg` `psycopg2-binary` packages."""
)

import controlflow.memory.providers.postgres as postgres_providers
Expand Down

0 comments on commit 4be056b

Please sign in to comment.