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."""
|
"""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)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue