anti
0b9873982d
Lift the format-agnostic pieces from decnet/orchestrator/emailgen/
into the new decnet/realism/ library so file-class content generation
(stage 3 of the realism migration) can reuse them. Email-specific
delivery (RFC 2822 EML, IMAP/POP3 spool, thread chains) stays in
orchestrator/.
Renames (history-preserving git mv):
emailgen/personas.py -> realism/personas.py
emailgen/prompt.py -> realism/prompts/email.py
emailgen/global_pool.py -> realism/personas_pool.py
emailgen/llm/ -> realism/llm/
Env-var clean break (pre-v1, no aliases):
DECNET_EMAILGEN_LLM -> DECNET_REALISM_LLM
DECNET_EMAILGEN_MODEL -> DECNET_REALISM_MODEL
DECNET_EMAILGEN_TIMEOUT -> DECNET_REALISM_TIMEOUT
DECNET_EMAILGEN_PERSONAS -> DECNET_REALISM_PERSONAS
DECNET_EMAILGEN_FAKE_OUTPUT -> DECNET_REALISM_FAKE_OUTPUT
Importers rewritten in: orchestrator/emailgen/scheduler.py,
orchestrator/drivers/email.py, web/router/{emailgen,topology}/
api_personas.py, cli/emailgen.py. Tests for moved modules relocated
to tests/realism/; tests for stay-put modules updated in place.
API URL `/api/v1/emailgen/personas` and CLI `decnet emailgen
import-personas` keep their public names until the service-collapse
commit (stage 5).
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
"""Backend dispatch.
|
|
|
|
Reads ``DECNET_REALISM_LLM`` to pick a concrete :class:`LLMBackend`.
|
|
Defaults to ``ollama`` because that's what the prototype proved out and
|
|
what most dev boxes have on hand.
|
|
|
|
Supported keys:
|
|
|
|
* ``ollama`` — :class:`decnet.realism.llm.impl.ollama.OllamaBackend`
|
|
* ``fake`` — :class:`decnet.realism.llm.impl.fake.FakeBackend`
|
|
(canned output, used by tests so they don't shell out)
|
|
|
|
Anthropic / vLLM / llama.cpp slots in here as a third branch when the
|
|
need shows up. Per the provider-subpackages convention, do NOT collapse
|
|
factory dispatch into the impl modules — keeps the ``__init__`` import
|
|
graph cycle-free and the env contract auditable in one place.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Any
|
|
|
|
from decnet.realism.llm.base import LLMBackend
|
|
|
|
|
|
def get_llm(*, model: str | None = None, **kwargs: Any) -> LLMBackend:
|
|
"""Instantiate the LLM backend selected by environment.
|
|
|
|
*model* (when provided) overrides whatever the backend's own default
|
|
is — e.g. for :class:`OllamaBackend` that's ``llama3.1`` unless
|
|
``DECNET_REALISM_MODEL`` says otherwise. Lets the worker honour
|
|
``decnet orchestrate --model gpt-oss`` without each backend having
|
|
to know about CLI flags.
|
|
"""
|
|
backend_key = os.environ.get("DECNET_REALISM_LLM", "ollama").lower()
|
|
|
|
if backend_key == "ollama":
|
|
from decnet.realism.llm.impl.ollama import OllamaBackend
|
|
return OllamaBackend(model=model, **kwargs)
|
|
if backend_key == "fake":
|
|
from decnet.realism.llm.impl.fake import FakeBackend
|
|
return FakeBackend(model=model or "fake-model", **kwargs)
|
|
raise ValueError(
|
|
f"Unsupported DECNET_REALISM_LLM={backend_key!r}; "
|
|
"expected one of: ollama, fake"
|
|
)
|