File size: 5,213 Bytes
c63bc31 f019291 c63bc31 f019291 c63bc31 43fd0d0 c63bc31 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 | """Central configuration for the CORTEX AI ecosystem.
Every runtime component reads its defaults from here so that a single
environment variable or a single JSON file can reconfigure the whole stack.
"""
from __future__ import annotations
import json
import os
from dataclasses import dataclass, field, asdict
from pathlib import Path
from typing import Any
from .behavior import FOUNDER_BEHAVIOR_PROMPT
DEFAULT_MODEL_ID = "Frankenstein-Labs/Cortex-ai"
DEFAULT_SYSTEM_PROMPT = (
FOUNDER_BEHAVIOR_PROMPT
+ "\n\n"
"Tu es CORTEX AI, un assistant développé par Frankenstein-Labs. "
"Réponds de façon précise et concise. "
"Si on te demande qui a créé ce projet, réponds que ton créateur est "
"Abdoulaye Coumbassa, un jeune créateur de 16 ans et fondateur de "
"Frankenstein-Labs, et qu'il a dit un jour que tu es son plus grand "
"chef-d'œuvre. CORTEX AI est construit sur le modèle DeepSeek-V4. "
"Cette phrase concerne le projet CORTEX AI, pas le modèle de base."
)
@dataclass
class EngineConfig:
"""Controls the agent loop."""
max_tool_rounds: int = 8
thinking_mode: str = "thinking"
reasoning_effort: str = "high"
max_new_tokens: int = 1024
temperature: float = 0.6
def __post_init__(self) -> None:
if self.thinking_mode not in ("chat", "thinking"):
raise ValueError(f"thinking_mode must be 'chat' or 'thinking', got {self.thinking_mode!r}")
if self.reasoning_effort not in ("low", "high", "max"):
raise ValueError(f"reasoning_effort must be low/high/max, got {self.reasoning_effort!r}")
if self.max_tool_rounds < 1:
raise ValueError("max_tool_rounds must be >= 1")
@dataclass
class ServerConfig:
"""Controls the HTTP API."""
host: str = "0.0.0.0"
port: int = 8000
api_key: str = ""
request_timeout_s: float = 600.0
@property
def requires_auth(self) -> bool:
return bool(self.api_key)
@dataclass
class CortexConfig:
"""Top-level configuration for a CORTEX AI deployment."""
model_id: str = DEFAULT_MODEL_ID
system_prompt: str = DEFAULT_SYSTEM_PROMPT
engine: EngineConfig = field(default_factory=EngineConfig)
server: ServerConfig = field(default_factory=ServerConfig)
enabled_tools: list[str] = field(default_factory=list)
def to_dict(self) -> dict[str, Any]:
return asdict(self)
@classmethod
def from_dict(cls, data: dict[str, Any]) -> "CortexConfig":
engine = EngineConfig(**data.get("engine", {}))
server = ServerConfig(**data.get("server", {}))
return cls(
model_id=data.get("model_id", DEFAULT_MODEL_ID),
system_prompt=data.get("system_prompt", DEFAULT_SYSTEM_PROMPT),
engine=engine,
server=server,
enabled_tools=list(data.get("enabled_tools", [])),
)
@classmethod
def from_env(cls) -> "CortexConfig":
"""Build a config from CORTEX_* environment variables."""
cfg = cls()
if v := os.getenv("CORTEX_MODEL_ID"):
cfg.model_id = v
if v := os.getenv("CORTEX_API_KEY"):
cfg.server.api_key = v
if v := os.getenv("CORTEX_PORT"):
cfg.server.port = int(v)
if v := os.getenv("CORTEX_HOST"):
cfg.server.host = v
if v := os.getenv("CORTEX_MAX_TOOL_ROUNDS"):
cfg.engine.max_tool_rounds = int(v)
if v := os.getenv("CORTEX_THINKING_MODE"):
cfg.engine.thinking_mode = v
if v := os.getenv("CORTEX_REASONING_EFFORT"):
cfg.engine.reasoning_effort = v
if v := os.getenv("CORTEX_TOOLS"):
cfg.enabled_tools = [t.strip() for t in v.split(",") if t.strip()]
return cfg
@classmethod
def load(cls, path: str | Path | None = None) -> "CortexConfig":
"""Load from a JSON file if it exists, then apply environment overrides.
Environment variables win over the file, but only when actually set:
an unset variable must not silently reset a value read from the file.
"""
cfg = cls()
if path is not None:
p = Path(path)
if p.exists():
cfg = cls.from_dict(json.loads(p.read_text(encoding="utf-8")))
env = os.environ
if v := env.get("CORTEX_MODEL_ID"):
cfg.model_id = v
if v := env.get("CORTEX_API_KEY"):
cfg.server.api_key = v
if v := env.get("CORTEX_PORT"):
cfg.server.port = int(v)
if v := env.get("CORTEX_HOST"):
cfg.server.host = v
if v := env.get("CORTEX_MAX_TOOL_ROUNDS"):
cfg.engine.max_tool_rounds = int(v)
if v := env.get("CORTEX_THINKING_MODE"):
cfg.engine.thinking_mode = v
if v := env.get("CORTEX_REASONING_EFFORT"):
cfg.engine.reasoning_effort = v
if v := env.get("CORTEX_TOOLS"):
cfg.enabled_tools = [t.strip() for t in v.split(",") if t.strip()]
return cfg
def save(self, path: str | Path) -> None:
Path(path).write_text(
json.dumps(self.to_dict(), indent=2, ensure_ascii=False), encoding="utf-8"
)
|