| | from __future__ import annotations |
| |
|
| | import os |
| |
|
| | from trenches_env.agents import AGENT_IDS |
| | from trenches_env.models import EntityModelBinding, ModelProviderName |
| | from trenches_env.rl import AGENT_ALLOWED_ACTIONS |
| |
|
| | _OBSERVATION_TOOLS = [ |
| | "inspect_public_brief", |
| | "inspect_private_brief", |
| | "inspect_assets", |
| | "inspect_sources", |
| | "emit_action", |
| | ] |
| | _KNOWN_PROVIDERS: set[str] = {"none", "openai", "anthropic", "openrouter", "huggingface", "ollama", "vllm", "custom"} |
| |
|
| |
|
| | def _env_value(base_name: str, agent_id: str) -> str | None: |
| | suffix = agent_id.upper() |
| | return os.getenv(f"{base_name}_{suffix}") or os.getenv(base_name) |
| |
|
| |
|
| | def _parse_bool(value: str | None, default: bool = False) -> bool: |
| | if value is None: |
| | return default |
| | return value.strip().lower() in {"1", "true", "yes", "on"} |
| |
|
| |
|
| | def _provider_name(value: str | None) -> ModelProviderName: |
| | normalized = (value or "none").strip().lower() |
| | if normalized not in _KNOWN_PROVIDERS: |
| | normalized = "custom" |
| | return normalized |
| |
|
| |
|
| | def build_entity_model_bindings() -> dict[str, EntityModelBinding]: |
| | bindings: dict[str, EntityModelBinding] = {} |
| | for agent_id in AGENT_IDS: |
| | provider = _provider_name(_env_value("TRENCHES_MODEL_PROVIDER", agent_id)) |
| | model_name = (_env_value("TRENCHES_MODEL_NAME", agent_id) or "").strip() |
| | base_url = (_env_value("TRENCHES_MODEL_BASE_URL", agent_id) or "").strip() or None |
| | api_key_env = (_env_value("TRENCHES_MODEL_API_KEY_ENV", agent_id) or "").strip() or None |
| | if provider == "huggingface" and api_key_env is None: |
| | api_key_env = "HF_TOKEN" |
| | configured = provider != "none" and bool(model_name) |
| | supports_tool_calls = _parse_bool(_env_value("TRENCHES_MODEL_SUPPORTS_TOOL_CALLS", agent_id), default=configured) |
| | supports_structured_output = _parse_bool( |
| | _env_value("TRENCHES_MODEL_SUPPORTS_STRUCTURED_OUTPUT", agent_id), |
| | default=configured, |
| | ) |
| |
|
| | notes: list[str] = [] |
| | if not configured: |
| | notes.append("Provider binding is not configured; heuristic fallback remains active.") |
| | if configured and api_key_env is None and provider not in {"ollama", "vllm"}: |
| | notes.append("No API key environment variable declared for this provider binding.") |
| | if configured and provider == "huggingface" and api_key_env == "HF_TOKEN": |
| | notes.append("Hugging Face bindings default to HF_TOKEN unless overridden per entity.") |
| | if configured and not supports_tool_calls: |
| | notes.append("Provider is configured without tool-calling support; action selection must stay text-only.") |
| |
|
| | bindings[agent_id] = EntityModelBinding( |
| | agent_id=agent_id, |
| | provider=provider, |
| | model_name=model_name, |
| | base_url=base_url, |
| | api_key_env=api_key_env, |
| | configured=configured, |
| | ready_for_inference=configured, |
| | decision_mode="provider_ready" if configured else "heuristic_fallback", |
| | supports_tool_calls=supports_tool_calls, |
| | supports_structured_output=supports_structured_output, |
| | action_tools=list(AGENT_ALLOWED_ACTIONS.get(agent_id, ())), |
| | observation_tools=list(_OBSERVATION_TOOLS), |
| | notes=notes, |
| | ) |
| | return bindings |
| |
|
