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:
parent
0621aee982
commit
e1765b2928
2 changed files with 19 additions and 6 deletions
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue