Pratik Narola
a228780146
- Make Ollama URL configurable via OLLAMA_BASE_URL env var - Add version: v1.1 to Mem0 config (required for latest features) - Make embedding model and dimensions configurable - Fix ownership check: O(1) lookup instead of fetching 10k records - Add tenacity retry logic for database operations
114 lines
3.7 KiB
Python
114 lines
3.7 KiB
Python
"""Configuration management for Mem0 Interface POC."""
|
|
|
|
import os
|
|
import json
|
|
from typing import List, Optional, Dict
|
|
from pydantic import Field, AliasChoices
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
"""Application settings loaded from environment variables."""
|
|
|
|
model_config = SettingsConfigDict(
|
|
env_file=".env", case_sensitive=False, extra="ignore"
|
|
)
|
|
|
|
# API Configuration
|
|
# Accept both OPENAI_API_KEY (from docker-compose) and OPENAI_COMPAT_API_KEY (from direct .env)
|
|
openai_api_key: str = Field(
|
|
validation_alias=AliasChoices(
|
|
"OPENAI_API_KEY", "OPENAI_COMPAT_API_KEY", "openai_api_key"
|
|
)
|
|
)
|
|
openai_base_url: str = Field(
|
|
validation_alias=AliasChoices(
|
|
"OPENAI_BASE_URL", "OPENAI_COMPAT_BASE_URL", "openai_base_url"
|
|
)
|
|
)
|
|
cohere_api_key: str = Field(
|
|
validation_alias=AliasChoices("COHERE_API_KEY", "cohere_api_key")
|
|
)
|
|
|
|
# Database Configuration
|
|
qdrant_host: str = Field(
|
|
default="localhost", validation_alias=AliasChoices("QDRANT_HOST", "qdrant_host")
|
|
)
|
|
qdrant_port: int = Field(
|
|
default=6333, validation_alias=AliasChoices("QDRANT_PORT", "qdrant_port")
|
|
)
|
|
qdrant_collection_name: str = Field(
|
|
default="mem0",
|
|
validation_alias=AliasChoices(
|
|
"QDRANT_COLLECTION_NAME", "qdrant_collection_name"
|
|
),
|
|
)
|
|
|
|
# Neo4j Configuration
|
|
neo4j_uri: str = Field(
|
|
default="bolt://localhost:7687",
|
|
validation_alias=AliasChoices("NEO4J_URI", "neo4j_uri"),
|
|
)
|
|
neo4j_username: str = Field(
|
|
default="neo4j",
|
|
validation_alias=AliasChoices("NEO4J_USERNAME", "neo4j_username"),
|
|
)
|
|
neo4j_password: str = Field(
|
|
default="mem0_neo4j_password",
|
|
validation_alias=AliasChoices("NEO4J_PASSWORD", "neo4j_password"),
|
|
)
|
|
|
|
# Application Configuration
|
|
log_level: str = Field(
|
|
default="INFO", validation_alias=AliasChoices("LOG_LEVEL", "log_level")
|
|
)
|
|
cors_origins: str = Field(
|
|
default="http://localhost:3000",
|
|
validation_alias=AliasChoices("CORS_ORIGINS", "cors_origins"),
|
|
)
|
|
|
|
# Model Configuration - Ultra-minimal (single model)
|
|
default_model: str = Field(
|
|
default="claude-sonnet-4",
|
|
validation_alias=AliasChoices("DEFAULT_MODEL", "default_model"),
|
|
)
|
|
|
|
# Embedder Configuration
|
|
ollama_base_url: str = Field(
|
|
default="http://host.docker.internal:11434",
|
|
validation_alias=AliasChoices("OLLAMA_BASE_URL", "ollama_base_url"),
|
|
)
|
|
embedding_model: str = Field(
|
|
default="qwen3-embedding:4b-q8_0",
|
|
validation_alias=AliasChoices("EMBEDDING_MODEL", "embedding_model"),
|
|
)
|
|
embedding_dims: int = Field(
|
|
default=2560, validation_alias=AliasChoices("EMBEDDING_DIMS", "embedding_dims")
|
|
)
|
|
|
|
# Authentication Configuration
|
|
# Format: JSON string mapping API keys to user IDs
|
|
# Example: {"api_key_123": "alice", "api_key_456": "bob"}
|
|
api_keys: str = Field(
|
|
default="{}", validation_alias=AliasChoices("API_KEYS", "api_keys")
|
|
)
|
|
|
|
@property
|
|
def cors_origins_list(self) -> List[str]:
|
|
"""Convert CORS origins string to list."""
|
|
return [origin.strip() for origin in self.cors_origins.split(",")]
|
|
|
|
@property
|
|
def api_key_mapping(self) -> Dict[str, str]:
|
|
"""Parse and return API key to user_id mapping."""
|
|
try:
|
|
mapping = json.loads(self.api_keys)
|
|
if not isinstance(mapping, dict):
|
|
raise ValueError("API_KEYS must be a JSON object/dict")
|
|
return mapping
|
|
except json.JSONDecodeError as e:
|
|
raise ValueError(f"Invalid API_KEYS JSON format: {e}")
|
|
|
|
|
|
# Global settings instance
|
|
settings = Settings()
|