phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 00:34:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove pydantic namespace warnings using model_config

Browse source
This commit is contained in:
Matthew Farrellee 2024-11-18 09:51:37 -05:00
parent f1b9578f8d
commit 0649cf0af7
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -31,6 +31,8 @@ class Model(CommonModelFields, Resource):
def provider_model_id(self) -> str: def provider_model_id(self) -> str:
return self.provider_resource_id return self.provider_resource_id
model_config = ConfigDict(protected_namespaces=())
class ModelInput(CommonModelFields): class ModelInput(CommonModelFields):
model_id: str model_id: str

9
llama_stack/cli/download.py
View file

@ -19,7 +19,7 @@ import httpx
from llama_models.datatypes import Model from llama_models.datatypes import Model
from llama_models.sku_list import LlamaDownloadInfo from llama_models.sku_list import LlamaDownloadInfo
from pydantic import BaseModel from pydantic import BaseModel, ConfigDict
from rich.console import Console from rich.console import Console
from rich.progress import ( from rich.progress import (
@ -293,8 +293,8 @@ class ParallelDownloader:
if free_space < required_space: if free_space < required_space:
self.console.print( self.console.print(
f"[red]Not enough disk space. Required: {required_space // (1024*1024)} MB, " f"[red]Not enough disk space. Required: {required_space // (1024 * 1024)} MB, "
f"Available: {free_space // (1024*1024)} MB[/red]" f"Available: {free_space // (1024 * 1024)} MB[/red]"
) )
return False return False
return True return True
@ -413,8 +413,7 @@ class ModelEntry(BaseModel):
model_id: str model_id: str
files: Dict[str, str] files: Dict[str, str]
class Config: model_config = ConfigDict(protected_namespaces=())
protected_namespaces = ()
class Manifest(BaseModel): class Manifest(BaseModel):