administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-07-15 00:11:15 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Include aliases

Browse Source
This commit is contained in:
Nicolas Mowen 2026-04-02 17:24:34 -06:00
parent 95555e4efa
commit b7a4eb57e9
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
frigate/genai/llama_cpp.py
View File

@ -72,8 +72,13 @@ class LlamaCppClient(GenAIClient):
models_data = response.json() models_data = response.json()
model_found = False model_found = False
configured_model = self.genai_config.model
for model in models_data.get("data", []): for model in models_data.get("data", []):
if model.get("id") == self.genai_config.model: model_ids = {model.get("id")}
# llama.cpp models can have aliases (e.g. short names)
for alias in model.get("aliases", []):
model_ids.add(alias)
if configured_model in model_ids:
model_found = True model_found = True
# Extract context size from model metadata if available # Extract context size from model metadata if available
meta = model.get("meta", {}) meta = model.get("meta", {})
@ -82,12 +87,14 @@ class LlamaCppClient(GenAIClient):
break break
if not model_found: if not model_found:
available = [ available = []
m.get("id", "unknown") for m in models_data.get("data", []) for m in models_data.get("data", []):
] available.append(m.get("id", "unknown"))
for alias in m.get("aliases", []):
available.append(alias)
logger.error( logger.error(
"Model '%s' not found on llama.cpp server. Available models: %s", "Model '%s' not found on llama.cpp server. Available models: %s",
self.genai_config.model, configured_model,
available, available,
) )
return None return None