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

use annotate and field discriminator

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-01 13:39:38 -07:00 • committed by Dinesh Yeduguru
parent ef64863616
commit 7696f31284
7 changed files with 56 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,7 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Dict, List, Optional, Protocol, runtime_checkable
from typing import Dict, List, Literal, Optional, Protocol, runtime_checkable
from llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, Field
@ -25,6 +25,7 @@ class ModelDef(BaseModel):
@json_schema_type
class ModelDefWithProvider(ModelDef):
type: Literal["model"] = "model"
provider_id: str = Field(
description="The provider ID for this model",
)