diff --git a/mcp_server/src/config/schema.py b/mcp_server/src/config/schema.py index ff7609bd1..f054600c2 100644 --- a/mcp_server/src/config/schema.py +++ b/mcp_server/src/config/schema.py @@ -187,6 +187,7 @@ class FalkorDBProviderConfig(BaseModel): """FalkorDB provider configuration.""" uri: str = 'redis://localhost:6379' + username: str | None = None password: str | None = None database: str = 'default_db' diff --git a/mcp_server/src/graphiti_mcp_server.py b/mcp_server/src/graphiti_mcp_server.py index 833bc5d93..0b059da18 100644 --- a/mcp_server/src/graphiti_mcp_server.py +++ b/mcp_server/src/graphiti_mcp_server.py @@ -218,6 +218,7 @@ async def initialize(self) -> None: falkor_driver = FalkorDriver( host=db_config['host'], port=db_config['port'], + username=db_config.get('username'), password=db_config['password'], database=db_config['database'], ) diff --git a/mcp_server/src/services/factories.py b/mcp_server/src/services/factories.py index 02f19e35d..d377ab137 100644 --- a/mcp_server/src/services/factories.py +++ b/mcp_server/src/services/factories.py @@ -418,6 +418,7 @@ def create_config(config: DatabaseConfig) -> dict: from urllib.parse import urlparse uri = os.environ.get('FALKORDB_URI', falkor_config.uri) + username = os.environ.get('FALKORDB_USERNAME', falkor_config.username) password = os.environ.get('FALKORDB_PASSWORD', falkor_config.password) # Parse the URI to extract host and port @@ -429,6 +430,7 @@ def create_config(config: DatabaseConfig) -> dict: 'driver': 'falkordb', 'host': host, 'port': port, + 'username': username, 'password': password, 'database': falkor_config.database, }