fix tests

This commit is contained in:
Xi Yan 2024-12-26 18:56:23 -08:00
parent 50764d76a7
commit 562ef41ff8
3 changed files with 20 additions and 13 deletions

View file

@ -8,10 +8,9 @@ from typing import Dict, List
from pydantic import BaseModel
from llama_stack.apis.inspect import Inspect
from llama_stack.distribution.datatypes import HealthInfo, ProviderInfo, RouteInfo
from llama_stack.apis.inspect import HealthInfo, Inspect, ProviderInfo, RouteInfo
from llama_stack.distribution.datatypes import StackRunConfig
from llama_stack.distribution.server.endpoints import get_all_api_endpoints
from llama_stack.providers.datatypes import StackRunConfig
class DistributionInspectConfig(BaseModel):