multi agent
This commit is contained in:
@@ -0,0 +1,41 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
from collections.abc import Mapping
|
||||
from dataclasses import dataclass
|
||||
|
||||
ENABLED_ENV = "MULTI_AGENT_COLLABORATION_ENABLED"
|
||||
MAX_RECOVERY_ENV = "MULTI_AGENT_MAX_RECOVERY_ATTEMPTS"
|
||||
DEFAULT_MAX_RECOVERY_ATTEMPTS = 3
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class CollaborationConfig:
|
||||
enabled: bool = False
|
||||
max_recovery_attempts: int = DEFAULT_MAX_RECOVERY_ATTEMPTS
|
||||
|
||||
|
||||
def load_config(env: Mapping[str, str] | None = None) -> CollaborationConfig:
|
||||
values = env or os.environ
|
||||
return CollaborationConfig(
|
||||
enabled=_parse_bool(values.get(ENABLED_ENV), default=False),
|
||||
max_recovery_attempts=_parse_int(
|
||||
values.get(MAX_RECOVERY_ENV), default=DEFAULT_MAX_RECOVERY_ATTEMPTS
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
def _parse_bool(value: str | None, *, default: bool) -> bool:
|
||||
if value is None:
|
||||
return default
|
||||
return value.strip().lower() in {"1", "true", "yes", "on", "enabled"}
|
||||
|
||||
|
||||
def _parse_int(value: str | None, *, default: int) -> int:
|
||||
if value is None:
|
||||
return default
|
||||
try:
|
||||
result = int(value)
|
||||
except ValueError:
|
||||
return default
|
||||
return result if result > 0 else default
|
||||
Reference in New Issue
Block a user