phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

No more model_id warnings

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-15 12:20:18 -08:00
parent e8112b31ab
commit 20bf2f50c2
2 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,7 +7,7 @@
from typing import Any, Dict, List, Literal, Optional, Protocol, runtime_checkable
from llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, Field
from pydantic import BaseModel, ConfigDict, Field
from llama_stack.apis.resource import Resource, ResourceType
@ -37,6 +37,8 @@ class ModelInput(CommonModelFields):
provider_id: Optional[str] = None
provider_model_id: Optional[str] = None
model_config = ConfigDict(protected_namespaces=())
@runtime_checkable
class Models(Protocol):