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

remove distribution types!

Browse source
This commit is contained in:
Xi Yan 2024-09-16 10:12:45 -07:00
parent e466ec389b
commit c0c5839361
17 changed files with 17 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_toolchain/core/datatypes.py
View file

@ -151,23 +151,13 @@ def remote_provider_spec(
@json_schema_type
class DistributionSpec(BaseModel):
distribution_type: str = Field(
default="local",
description="Name of the distribution type. This can used to identify the distribution",
)
description: Optional[str] = Field(
default="",
description="Description of the distribution",
)
docker_image: Optional[str] = None
providers: Dict[str, str] = Field(
default={
"inference": "meta-reference",
"memory": "meta-reference-faiss",
"safety": "meta-reference",
"agentic_system": "meta-reference",
"telemetry": "console",
},
default_factory=dict,
description="Provider Types for each of the APIs provided by this distribution",
)
@ -183,8 +173,6 @@ Reference to the distribution this package refers to. For unregistered (adhoc) p
this could be just a hash
""",
)
distribution_type: Optional[str] = None
docker_image: Optional[str] = Field(
default=None,
description="Reference to the docker image if this package refers to a container",