No more model_id warnings

meta-llama · Nov 15, 2024 · 20bf2f5 · 20bf2f5
1 parent e8112b3
commit 20bf2f5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git 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
@@ -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: