refactor: Add missing install info
This commit is contained in:
parent
d1724c710b
commit
8265ec0334
2 changed files with 2 additions and 2 deletions
|
|
@ -48,7 +48,7 @@ def create_relational_engine(
|
||||||
)
|
)
|
||||||
except ImportError:
|
except ImportError:
|
||||||
raise 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)
|
return SQLAlchemyAdapter(connection_string)
|
||||||
|
|
|
||||||
|
|
@ -79,7 +79,7 @@ class LLMConfig(BaseSettings):
|
||||||
if self.structured_output_framework.lower() == "baml" and ClientRegistry is None:
|
if self.structured_output_framework.lower() == "baml" and ClientRegistry is None:
|
||||||
raise ImportError(
|
raise ImportError(
|
||||||
"BAML is selected as structured output framework but not available. "
|
"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:
|
elif self.structured_output_framework.lower() == "baml" and ClientRegistry is not None:
|
||||||
self.baml_registry = ClientRegistry()
|
self.baml_registry = ClientRegistry()
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue