phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 00:49:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add validator for provider_resource_id

Browse source
This commit is contained in:
Matthew Farrellee 2025-07-16 11:58:07 -04:00
parent 7ba5a67224
commit 9205cd04c2
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/apis/models/models.py
View file

@ -7,7 +7,7 @@
from enum import StrEnum from enum import StrEnum
from typing import Any, Literal, Protocol, runtime_checkable from typing import Any, Literal, Protocol, runtime_checkable
from pydantic import BaseModel, ConfigDict, Field from pydantic import BaseModel, ConfigDict, Field, field_validator
from llama_stack.apis.resource import Resource, ResourceType from llama_stack.apis.resource import Resource, ResourceType
from llama_stack.providers.utils.telemetry.trace_protocol import trace_protocol from llama_stack.providers.utils.telemetry.trace_protocol import trace_protocol
@ -44,6 +44,13 @@ class Model(CommonModelFields, Resource):
model_type: ModelType = Field(default=ModelType.llm) model_type: ModelType = Field(default=ModelType.llm)
@field_validator("provider_resource_id")
@classmethod
def validate_provider_resource_id(cls, v):
if v is None:
raise ValueError("provider_resource_id cannot be None")
return v
class ModelInput(CommonModelFields): class ModelInput(CommonModelFields):
model_id: str model_id: str