phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 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

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: