diff --git a/cognee/infrastructure/databases/relational/create_relational_engine.py b/cognee/infrastructure/databases/relational/create_relational_engine.py index 4f117bf4c..deaeaa2da 100644 --- a/cognee/infrastructure/databases/relational/create_relational_engine.py +++ b/cognee/infrastructure/databases/relational/create_relational_engine.py @@ -48,7 +48,7 @@ def create_relational_engine( ) except ImportError: raise ImportError( - "PostgreSQL dependencies are not installed. Please install with 'pip install cognee[postgres]' or 'pip install cognee[postgres-binary]' to use PostgreSQL functionality." + "PostgreSQL dependencies are not installed. Please install with 'pip install cognee\"[postgres]\"' or 'pip install cognee\"[postgres-binary]\"' to use PostgreSQL functionality." ) return SQLAlchemyAdapter(connection_string) diff --git a/cognee/infrastructure/llm/config.py b/cognee/infrastructure/llm/config.py index 3c3ff00b0..6658a6251 100644 --- a/cognee/infrastructure/llm/config.py +++ b/cognee/infrastructure/llm/config.py @@ -79,7 +79,7 @@ class LLMConfig(BaseSettings): if self.structured_output_framework.lower() == "baml" and ClientRegistry is None: raise ImportError( "BAML is selected as structured output framework but not available. " - "Please install with 'pip install cognee[baml]' to use BAML extraction features." + "Please install with 'pip install cognee\"[baml]\"' to use BAML extraction features." ) elif self.structured_output_framework.lower() == "baml" and ClientRegistry is not None: self.baml_registry = ClientRegistry()