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

Remove additional_pip_packages; move deps to providers

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-08 10:19:46 -07:00
parent 6de36b6a15
commit 8d7ecf0c47
5 changed files with 18 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_toolchain/distribution/distribution.py
View file

@ -22,6 +22,14 @@ from .datatypes import (
ProviderSpec,
)
# These are the dependencies needed by the distribution server.
# `llama-toolchain` is automatically installed by the installation script.
SERVER_DEPENDENCIES = [
"fastapi",
"python-dotenv",
"uvicorn",
]
def distribution_dependencies(distribution: DistributionSpec) -> List[str]:
# only consider InlineProviderSpecs when calculating dependencies
@ -30,7 +38,7 @@ def distribution_dependencies(distribution: DistributionSpec) -> List[str]:
for provider_spec in distribution.provider_specs.values()
if isinstance(provider_spec, InlineProviderSpec)
for dep in provider_spec.pip_packages
] + distribution.additional_pip_packages
] + SERVER_DEPENDENCIES
def api_endpoints() -> Dict[Api, List[ApiEndpoint]]: