From 20bf2f50c28f7f22d8c83449dea9a697e16e5fe1 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Fri, 15 Nov 2024 12:20:18 -0800 Subject: [PATCH] No more model_id warnings --- llama_stack/apis/models/models.py | 4 +++- llama_stack/distribution/server/server.py | 14 +++++++++----- 2 files changed, 12 insertions(+), 6 deletions(-) diff --git a/llama_stack/apis/models/models.py b/llama_stack/apis/models/models.py index a1bfcac00..aabe78d85 100644 --- a/llama_stack/apis/models/models.py +++ b/llama_stack/apis/models/models.py @@ -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): diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 5796b6c68..0cfd11eda 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -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: