phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki 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):

14
llama_stack/distribution/server/server.py
View file

@ -369,12 +369,16 @@ def main(
impl_method = getattr(impl, endpoint.name)
getattr(app, endpoint.method)(endpoint.route, response_model=None)(
create_dynamic_typed_route(
impl_method,
endpoint.method,
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", category=UserWarning, module="pydantic._internal._fields"
)
getattr(app, endpoint.method)(endpoint.route, response_model=None)(
create_dynamic_typed_route(
impl_method,
endpoint.method,
)
)
)
cprint(f"Serving API {api_str}", "white", attrs=["bold"])
for endpoint in endpoints: