File size: 3,380 Bytes
1794757 | 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 | 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 # type: ignore[return-value]
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
|