| 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 |
|
|
