fix: Update embedder to modern model and fix Neo4j auth

- Update default OpenAI embedder from text-embedding-ada-002 to text-embedding-3-small
- Add environment variable support for Neo4j credentials (NEO4J_URI, NEO4J_USER, NEO4J_PASSWORD)
- Add environment variable support for FalkorDB credentials (FALKORDB_URI, FALKORDB_PASSWORD)
- Fixes CI/CD authentication issues where environment variables weren't being recognized

🤖 Generated with Claude Code

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Daniel Chalef 2025-08-26 20:43:30 -07:00
parent 0621aee982
commit e1765b2928
2 changed files with 19 additions and 6 deletions

View file

@ -145,7 +145,7 @@ class EmbedderConfig(BaseModel):
"""Embedder configuration.""" """Embedder configuration."""
provider: str = Field(default='openai', description='Embedder provider') provider: str = Field(default='openai', description='Embedder provider')
model: str = Field(default='text-embedding-ada-002', description='Model name') model: str = Field(default='text-embedding-3-small', description='Model name')
dimensions: int = Field(default=1536, description='Embedding dimensions') dimensions: int = Field(default=1536, description='Embedding dimensions')
providers: EmbedderProvidersConfig = Field(default_factory=EmbedderProvidersConfig) providers: EmbedderProvidersConfig = Field(default_factory=EmbedderProvidersConfig)

View file

@ -263,10 +263,17 @@ class DatabaseDriverFactory:
neo4j_config = Neo4jProviderConfig() neo4j_config = Neo4jProviderConfig()
# Check for environment variable overrides (for CI/CD compatibility)
import os
uri = os.environ.get('NEO4J_URI', neo4j_config.uri)
username = os.environ.get('NEO4J_USER', neo4j_config.username)
password = os.environ.get('NEO4J_PASSWORD', neo4j_config.password)
return { return {
'uri': neo4j_config.uri, 'uri': uri,
'user': neo4j_config.username, 'user': username,
'password': neo4j_config.password, 'password': password,
# Note: database and use_parallel_runtime would need to be passed # Note: database and use_parallel_runtime would need to be passed
# to the driver after initialization if supported # to the driver after initialization if supported
} }
@ -286,10 +293,16 @@ class DatabaseDriverFactory:
falkor_config = FalkorDBProviderConfig() falkor_config = FalkorDBProviderConfig()
# Check for environment variable overrides (for CI/CD compatibility)
import os
uri = os.environ.get('FALKORDB_URI', falkor_config.uri)
password = os.environ.get('FALKORDB_PASSWORD', falkor_config.password)
return { return {
'driver': 'falkordb', 'driver': 'falkordb',
'uri': falkor_config.uri, 'uri': uri,
'password': falkor_config.password, 'password': password,
'database': falkor_config.database, 'database': falkor_config.database,
} }