diff --git a/llama_stack/distribution/distribution.py b/llama_stack/distribution/distribution.py index b641b6582..035febb80 100644 --- a/llama_stack/distribution/distribution.py +++ b/llama_stack/distribution/distribution.py @@ -8,6 +8,8 @@ import importlib import inspect from typing import Dict, List +from pydantic import BaseModel + from llama_stack.apis.agents import Agents from llama_stack.apis.inference import Inference from llama_stack.apis.memory import Memory @@ -17,8 +19,6 @@ from llama_stack.apis.safety import Safety from llama_stack.apis.shields import Shields from llama_stack.apis.telemetry import Telemetry -from pydantic import BaseModel - from .datatypes import Api, ApiEndpoint, ProviderSpec, remote_provider_spec # These are the dependencies needed by the distribution server. @@ -26,6 +26,7 @@ from .datatypes import Api, ApiEndpoint, ProviderSpec, remote_provider_spec SERVER_DEPENDENCIES = [ "fastapi", "fire", + "httpx", "uvicorn", ]