Circuit-Forge/peregrine
3
0
Fork 0
Code Issues 17 Pull requests 2 Projects Releases 6 Packages Wiki Activity Actions

feat: add ollama_research to preflight service table and LLM backend map

Browse source
This commit is contained in:
pyr0ball 2026-02-27 00:14:04 -08:00
parent dd40a84174
commit c9d7b810f6
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
scripts/preflight.py
View file

@ -44,26 +44,29 @@ OVERRIDE_YML = ROOT / "compose.override.yml"
# adoptable — True if an existing process on this port should be used instead # adoptable — True if an existing process on this port should be used instead
# of starting a Docker container (and the Docker service disabled) # of starting a Docker container (and the Docker service disabled)
_SERVICES: dict[str, tuple[str, int, str, bool, bool]] = { _SERVICES: dict[str, tuple[str, int, str, bool, bool]] = {
"streamlit": ("streamlit_port", 8501, "STREAMLIT_PORT", True, False), "streamlit": ("streamlit_port", 8501, "STREAMLIT_PORT", True, False),
"searxng": ("searxng_port", 8888, "SEARXNG_PORT", True, True), "searxng": ("searxng_port", 8888, "SEARXNG_PORT", True, True),
"vllm": ("vllm_port", 8000, "VLLM_PORT", True, True), "vllm": ("vllm_port", 8000, "VLLM_PORT", True, True),
"vision": ("vision_port", 8002, "VISION_PORT", True, True), "vision": ("vision_port", 8002, "VISION_PORT", True, True),
"ollama": ("ollama_port", 11434, "OLLAMA_PORT", True, True), "ollama": ("ollama_port", 11434, "OLLAMA_PORT", True, True),
"ollama_research": ("ollama_research_port", 11435, "OLLAMA_RESEARCH_PORT", True, True),
} }
# LLM yaml backend keys → url suffix, keyed by service name # LLM yaml backend keys → url suffix, keyed by service name
_LLM_BACKENDS: dict[str, list[tuple[str, str]]] = { _LLM_BACKENDS: dict[str, list[tuple[str, str]]] = {
"ollama": [("ollama", "/v1"), ("ollama_research", "/v1")], "ollama": [("ollama", "/v1")],
"vllm": [("vllm", "/v1")], "ollama_research": [("ollama_research", "/v1")],
"vision": [("vision_service", "")], "vllm": [("vllm", "/v1"), ("vllm_research", "/v1")],
"vision": [("vision_service", "")],
} }
# Docker-internal hostname:port for each service (when running in Docker) # Docker-internal hostname:port for each service (when running in Docker)
_DOCKER_INTERNAL: dict[str, tuple[str, int]] = { _DOCKER_INTERNAL: dict[str, tuple[str, int]] = {
"ollama": ("ollama", 11434), "ollama": ("ollama", 11434),
"vllm": ("vllm", 8000), "ollama_research": ("ollama_research", 11434), # container-internal port is always 11434
"vision": ("vision", 8002), "vllm": ("vllm", 8000),
"searxng": ("searxng", 8080), # searxng internal port differs from host port "vision": ("vision", 8002),
"searxng": ("searxng", 8080), # searxng internal port differs from host port
} }