phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-03 19:57:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pre-commit

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-02 14:57:58 -07:00
parent 31f211f0ad
commit be6819fa92
8 changed files with 14 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,7 +7,7 @@
from enum import Enum
from typing import Any, Literal, Protocol
from pydantic import BaseModel, Field
from pydantic import BaseModel
from typing_extensions import runtime_checkable
from llama_stack.apis.common.content_types import URL, InterleavedContent

3
llama_stack/models/llama/llama3_1/prompts.py
View file

@ -11,6 +11,7 @@
# top-level folder for each specific model found within the models/ directory at
# the top-level of this source tree.
import json
import textwrap
from llama_stack.models.llama.datatypes import (
@ -184,7 +185,7 @@ def usecases() -> list[UseCase | str]:
ToolCall(
call_id="tool_call_id",
tool_name=BuiltinTool.wolfram_alpha,
arguments={"query": "100th decimal of pi"},
arguments=json.dumps({"query": "100th decimal of pi"}),
)
],
),

3
llama_stack/models/llama/llama3_3/prompts.py
View file

@ -11,6 +11,7 @@
# top-level folder for each specific model found within the models/ directory at
# the top-level of this source tree.
import json
import textwrap
from llama_stack.models.llama.datatypes import (
@ -185,7 +186,7 @@ def usecases() -> list[UseCase | str]:
ToolCall(
call_id="tool_call_id",
tool_name=BuiltinTool.wolfram_alpha,
arguments={"query": "100th decimal of pi"},
arguments=json.dumps({"query": "100th decimal of pi"}),
)
],
),

1
llama_stack/providers/utils/inference/openai_compat.py
View file

@ -1171,7 +1171,6 @@ async def convert_openai_chat_completion_stream(
)
try:
arguments = json.loads(buffer["arguments"])
tool_call = ToolCall(
call_id=buffer["call_id"],
tool_name=buffer["name"],

7
tests/unit/distribution/routers/test_routing_tables.py
View file

@ -139,12 +139,7 @@ class ToolGroupsImpl(Impl):
description="Test tool",
input_schema={
"type": "object",
"properties": {
"test-param": {
"type": "string",
"description": "Test param"
}
}
"properties": {"test-param": {"type": "string", "description": "Test param"}},
},
)
]

13
tests/unit/providers/agent/test_meta_reference_agent.py
View file

@ -16,7 +16,6 @@ from llama_stack.apis.agents import (
)
from llama_stack.apis.common.responses import PaginatedResponse
from llama_stack.apis.inference import Inference
from llama_stack.apis.resource import ResourceType
from llama_stack.apis.safety import Safety
from llama_stack.apis.tools import ListToolDefsResponse, ToolDef, ToolGroups, ToolRuntime
from llama_stack.apis.vector_io import VectorIO
@ -241,20 +240,16 @@ async def test__initialize_tools(agents_impl, sample_agent_config):
input_schema={
"type": "object",
"properties": {
"story_title": {
"type": "string",
"description": "Title of the story",
"title": "Story Title"
},
"story_title": {"type": "string", "description": "Title of the story", "title": "Story Title"},
"input_words": {
"type": "array",
"description": "Input words",
"items": {"type": "string"},
"title": "Input Words",
"default": []
}
"default": [],
},
},
"required": ["story_title"]
"required": ["story_title"],
},
)
]

9
tests/unit/providers/agents/meta_reference/test_openai_responses.py
View file

@ -192,13 +192,8 @@ async def test_create_openai_response_with_string_input_with_tools(openai_respon
description="Search the web for information",
input_schema={
"type": "object",
"properties": {
"query": {
"type": "string",
"description": "The query to search for"
}
},
"required": ["query"]
"properties": {"query": {"type": "string", "description": "The query to search for"}},
"required": ["query"],
},
)

10
tests/unit/providers/inline/agents/meta_reference/responses/test_streaming.py
View file

@ -22,14 +22,8 @@ def test_convert_tooldef_to_chat_tool_preserves_items_field():
description="A test tool with array parameter",
input_schema={
"type": "object",
"properties": {
"tags": {
"type": "array",
"description": "List of tags",
"items": {"type": "string"}
}
},
"required": ["tags"]
"properties": {"tags": {"type": "array", "description": "List of tags", "items": {"type": "string"}}},
"required": ["tags"],
},
)