gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Change vector_db_url usage to its appropriate vector_db_host usage

Browse source
This commit is contained in:
Igor Ilic 2026-01-20 16:09:33 +01:00
parent 05084e6779
commit 0903092fc9
5 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py
View file

@ -102,7 +102,7 @@ class SQLAlchemyAdapter:
"postgresql+asyncpg",
username=vector_config.vector_db_username,
password=vector_config.vector_db_password,
host=vector_config.vector_db_url,
host=vector_config.vector_db_host,
port=int(vector_config.vector_db_port),
database=maintenance_db_name,
)

2
cognee/infrastructure/databases/vector/config.py
View file

@ -31,6 +31,7 @@ class VectorConfig(BaseSettings):
vector_dataset_database_handler: str = "lancedb"
vector_db_username: str = ""
vector_db_password: str = ""
vector_db_host: str = ""
model_config = SettingsConfigDict(env_file=".env", extra="allow")
@ -69,6 +70,7 @@ class VectorConfig(BaseSettings):
"vector_dataset_database_handler": self.vector_dataset_database_handler,
"vector_db_username": self.vector_db_username,
"vector_db_password": self.vector_db_password,
"vector_db_host": self.vector_db_host,
}

19
cognee/infrastructure/databases/vector/create_vector_engine.py
View file

@ -16,6 +16,7 @@ def create_vector_engine(
vector_dataset_database_handler: str = "",
vector_db_username: str = "",
vector_db_password: str = "",
vector_db_host: str = "",
):
"""
Wrapper function to call create vector engine with caching.
@ -28,6 +29,9 @@ def create_vector_engine(
vector_db_port,
vector_db_key,
vector_dataset_database_handler,
vector_db_username,
vector_db_password,
vector_db_host,
)
@ -36,9 +40,12 @@ def _create_vector_engine(
vector_db_provider: str,
vector_db_url: str,
vector_db_name: str,
vector_db_port: str = "",
vector_db_key: str = "",
vector_dataset_database_handler: str = "",
vector_db_port: str,
vector_db_key: str,
vector_dataset_database_handler: str,
vector_db_username: str,
vector_db_password: str,
vector_db_host: str,
):
"""
Create a vector database engine based on the specified provider.
@ -84,19 +91,19 @@ def _create_vector_engine(
if backend_access_control_enabled():
connection_string: str = (
f"postgresql+asyncpg://{vector_db_username}:{vector_db_password}"
f"@{vector_db_url}:{vector_db_port}/{vector_db_name}"
f"@{vector_db_host}:{vector_db_port}/{vector_db_name}"
)
else:
if (
vector_db_port
and vector_db_username
and vector_db_password
and vector_db_url
and vector_db_host
and vector_db_name
):
connection_string: str = (
f"postgresql+asyncpg://{vector_db_username}:{vector_db_password}"
f"@{vector_db_url}:{vector_db_port}/{vector_db_name}"
f"@{vector_db_host}:{vector_db_port}/{vector_db_name}"
)
else:
from cognee.infrastructure.databases.relational import get_relational_config

2
cognee/infrastructure/databases/vector/pgvector/PGVectorDatasetDatabaseHandler.py
View file

@ -32,6 +32,7 @@ class PGVectorDatasetDatabaseHandler(DatasetDatabaseHandlerInterface):
"port": vector_config.vector_db_port,
"username": vector_config.vector_db_username,
"password": vector_config.vector_db_password,
"host": vector_config.vector_db_host,
},
"vector_dataset_database_handler": "pgvector_local",
}
@ -51,6 +52,7 @@ class PGVectorDatasetDatabaseHandler(DatasetDatabaseHandlerInterface):
"vector_db_password": new_vector_config["vector_database_connection_info"][
"password"
],
"vector_db_host": new_vector_config["vector_database_connection_info"]["host"],
"vector_dataset_database_handler": "pgvector",
}
)

2
cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py
View file

@ -29,7 +29,7 @@ async def create_pg_database(vector_config):
pg_relational_engine = create_relational_engine(
db_path="",
db_host=vector_config["vector_db_url"],
db_host=vector_config["vector_db_host"],
db_name=vector_config["vector_db_name"],
db_port=vector_config["vector_db_port"],
db_username=vector_config["vector_db_username"],