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

change to llm type

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-10 16:57:00 -08:00
parent 62890b3171
commit 2b6aa71a21
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -22,7 +22,7 @@ class CommonModelFields(BaseModel):
class ModelType(Enum):
llm_model = "llm_model"
llm = "llm"
embedding_model = "embedding_model"
@ -40,14 +40,14 @@ class Model(CommonModelFields, Resource):
model_config = ConfigDict(protected_namespaces=())
model_type: ModelType = Field(default=ModelType.llm_model)
model_type: ModelType = Field(default=ModelType.llm)
class ModelInput(CommonModelFields):
model_id: str
provider_id: Optional[str] = None
provider_model_id: Optional[str] = None
model_type: Optional[ModelType] = ModelType.llm_model
model_type: Optional[ModelType] = ModelType.llm
model_config = ConfigDict(protected_namespaces=())