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

working end to end client sdk tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-23 11:04:24 -08:00
parent 2ad67529ef
commit 0155700ea6
5 changed files with 947 additions and 1039 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -74,12 +74,14 @@ ToolGroupDef = register_schema(
)
@json_schema_type
class ToolGroupInput(BaseModel):
tool_group_id: str
tool_group: ToolGroupDef
provider_id: Optional[str] = None
@json_schema_type
class ToolGroup(Resource):
type: Literal[ResourceType.tool_group.value] = ResourceType.tool_group.value

3
llama_stack/distribution/stack.py
View file

@ -33,6 +33,7 @@ from llama_stack.apis.scoring_functions import ScoringFunctions
from llama_stack.apis.shields import Shields
from llama_stack.apis.synthetic_data_generation import SyntheticDataGeneration
from llama_stack.apis.telemetry import Telemetry
from llama_stack.apis.tools import ToolGroups, ToolRuntime
from llama_stack.distribution.datatypes import StackRunConfig
from llama_stack.distribution.distribution import get_provider_registry
from llama_stack.distribution.resolver import ProviderRegistry, resolve_impls
@ -63,6 +64,8 @@ class LlamaStack(
Models,
Shields,
Inspect,
ToolGroups,
ToolRuntime,
):
pass