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 128ab11763
commit 637e8379b6
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
scripts/preflight.py
View file

@ -49,18 +49,21 @@ _SERVICES: dict[str, tuple[str, int, str, bool, bool]] = {
"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")],
"vllm": [("vllm", "/v1"), ("vllm_research", "/v1")],
"vision": [("vision_service", "")], "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),
"ollama_research": ("ollama_research", 11434), # container-internal port is always 11434
"vllm": ("vllm", 8000), "vllm": ("vllm", 8000),
"vision": ("vision", 8002), "vision": ("vision", 8002),
"searxng": ("searxng", 8080), # searxng internal port differs from host port "searxng": ("searxng", 8080), # searxng internal port differs from host port