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."""
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')
providers: EmbedderProvidersConfig = Field(default_factory=EmbedderProvidersConfig)

View file

@ -263,10 +263,17 @@ class DatabaseDriverFactory:
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 {
'uri': neo4j_config.uri,
'user': neo4j_config.username,
'password': neo4j_config.password,
'uri': uri,
'user': username,
'password': password,
# Note: database and use_parallel_runtime would need to be passed
# to the driver after initialization if supported
}
@ -286,10 +293,16 @@ class DatabaseDriverFactory:
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 {
'driver': 'falkordb',
'uri': falkor_config.uri,
'password': falkor_config.password,
'uri': uri,
'password': password,
'database': falkor_config.database,
}