address feedback

This commit is contained in:
Dinesh Yeduguru 2024-11-08 16:11:53 -08:00
parent 772e23e29e
commit d6a9a17828
4 changed files with 2 additions and 19 deletions

View file

@ -15,9 +15,6 @@ from llama_stack.apis.resource import Resource, ResourceType
@json_schema_type
class Model(Resource):
type: Literal[ResourceType.model.value] = ResourceType.model.value
llama_model: str = Field(
description="Pointer to the underlying core Llama family model. Each model served by Llama Stack must have a core Llama model.",
)
metadata: Dict[str, Any] = Field(
default_factory=dict,
description="Any additional metadata for this model",
@ -38,6 +35,5 @@ class Models(Protocol):
model_id: str,
provider_model_id: Optional[str] = None,
provider_id: Optional[str] = None,
llama_model: Optional[str] = None,
metadata: Optional[Dict[str, Any]] = None,
) -> Model: ...