Skip to content
This repository was archived by the owner on Aug 14, 2025. It is now read-only.

Commit 748e6db

Browse files
feat(api): update via SDK Studio
1 parent b6fa2b1 commit 748e6db

File tree

11 files changed

+170
-76
lines changed

11 files changed

+170
-76
lines changed

.stats.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 106
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-ab9c9bf2527d3b4179e2bc3e6495c64d43c42b2ea8dc1a55d472986e1a1430a0.yml
3-
openapi_spec_hash: b93c85fb747e3c29134451d2f364ce8b
4-
config_hash: b105ed4704c2998287916138c03d211a
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-fe0ae8a478adcaa69c96b9ee501073818e499700ce2ec66dd7f6eaf4c3fcf69c.yml
3+
openapi_spec_hash: 418d4a00c4676bbb2973b64a93484965
4+
config_hash: e6c3e48e220b264936ee6df8b996ab12

api.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -458,12 +458,12 @@ Methods:
458458
Types:
459459

460460
```python
461-
from llama_stack_client.types import OpenAIModerationsResponse, RunShieldResponse
461+
from llama_stack_client.types import CreateResponse, RunShieldResponse
462462
```
463463

464464
Methods:
465465

466-
- <code title="post /v1/openai/v1/moderations">client.safety.<a href="./src/llama_stack_client/resources/safety.py">openai_moderations</a>(\*\*<a href="src/llama_stack_client/types/safety_openai_moderations_params.py">params</a>) -> <a href="./src/llama_stack_client/types/openai_moderations_response.py">OpenAIModerationsResponse</a></code>
466+
- <code title="post /v1/openai/v1/moderations">client.safety.<a href="./src/llama_stack_client/resources/safety.py">create</a>(\*\*<a href="src/llama_stack_client/types/safety_create_params.py">params</a>) -> <a href="./src/llama_stack_client/types/create_response.py">CreateResponse</a></code>
467467
- <code title="post /v1/safety/run-shield">client.safety.<a href="./src/llama_stack_client/resources/safety.py">run_shield</a>(\*\*<a href="src/llama_stack_client/types/safety_run_shield_params.py">params</a>) -> <a href="./src/llama_stack_client/types/run_shield_response.py">RunShieldResponse</a></code>
468468

469469
# Shields

src/llama_stack_client/resources/safety.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import httpx
88

9-
from ..types import safety_run_shield_params, safety_openai_moderations_params
9+
from ..types import safety_create_params, safety_run_shield_params
1010
from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
1111
from .._utils import maybe_transform, async_maybe_transform
1212
from .._compat import cached_property
@@ -18,9 +18,9 @@
1818
async_to_streamed_response_wrapper,
1919
)
2020
from .._base_client import make_request_options
21+
from ..types.create_response import CreateResponse
2122
from ..types.run_shield_response import RunShieldResponse
2223
from ..types.shared_params.message import Message
23-
from ..types.openai_moderations_response import OpenAIModerationsResponse
2424

2525
__all__ = ["SafetyResource", "AsyncSafetyResource"]
2626

@@ -45,18 +45,18 @@ def with_streaming_response(self) -> SafetyResourceWithStreamingResponse:
4545
"""
4646
return SafetyResourceWithStreamingResponse(self)
4747

48-
def openai_moderations(
48+
def create(
4949
self,
5050
*,
5151
input: Union[str, List[str]],
52-
model: str | NotGiven = NOT_GIVEN,
52+
model: str,
5353
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
5454
# The extra values given here take precedence over values defined on the client or passed to this method.
5555
extra_headers: Headers | None = None,
5656
extra_query: Query | None = None,
5757
extra_body: Body | None = None,
5858
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
59-
) -> OpenAIModerationsResponse:
59+
) -> CreateResponse:
6060
"""
6161
Classifies if text and/or image inputs are potentially harmful.
6262
@@ -81,12 +81,12 @@ def openai_moderations(
8181
"input": input,
8282
"model": model,
8383
},
84-
safety_openai_moderations_params.SafetyOpenAIModerationsParams,
84+
safety_create_params.SafetyCreateParams,
8585
),
8686
options=make_request_options(
8787
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
8888
),
89-
cast_to=OpenAIModerationsResponse,
89+
cast_to=CreateResponse,
9090
)
9191

9292
def run_shield(
@@ -157,18 +157,18 @@ def with_streaming_response(self) -> AsyncSafetyResourceWithStreamingResponse:
157157
"""
158158
return AsyncSafetyResourceWithStreamingResponse(self)
159159

160-
async def openai_moderations(
160+
async def create(
161161
self,
162162
*,
163163
input: Union[str, List[str]],
164-
model: str | NotGiven = NOT_GIVEN,
164+
model: str,
165165
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
166166
# The extra values given here take precedence over values defined on the client or passed to this method.
167167
extra_headers: Headers | None = None,
168168
extra_query: Query | None = None,
169169
extra_body: Body | None = None,
170170
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
171-
) -> OpenAIModerationsResponse:
171+
) -> CreateResponse:
172172
"""
173173
Classifies if text and/or image inputs are potentially harmful.
174174
@@ -193,12 +193,12 @@ async def openai_moderations(
193193
"input": input,
194194
"model": model,
195195
},
196-
safety_openai_moderations_params.SafetyOpenAIModerationsParams,
196+
safety_create_params.SafetyCreateParams,
197197
),
198198
options=make_request_options(
199199
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
200200
),
201-
cast_to=OpenAIModerationsResponse,
201+
cast_to=CreateResponse,
202202
)
203203

204204
async def run_shield(
@@ -253,8 +253,8 @@ class SafetyResourceWithRawResponse:
253253
def __init__(self, safety: SafetyResource) -> None:
254254
self._safety = safety
255255

256-
self.openai_moderations = to_raw_response_wrapper(
257-
safety.openai_moderations,
256+
self.create = to_raw_response_wrapper(
257+
safety.create,
258258
)
259259
self.run_shield = to_raw_response_wrapper(
260260
safety.run_shield,
@@ -265,8 +265,8 @@ class AsyncSafetyResourceWithRawResponse:
265265
def __init__(self, safety: AsyncSafetyResource) -> None:
266266
self._safety = safety
267267

268-
self.openai_moderations = async_to_raw_response_wrapper(
269-
safety.openai_moderations,
268+
self.create = async_to_raw_response_wrapper(
269+
safety.create,
270270
)
271271
self.run_shield = async_to_raw_response_wrapper(
272272
safety.run_shield,
@@ -277,8 +277,8 @@ class SafetyResourceWithStreamingResponse:
277277
def __init__(self, safety: SafetyResource) -> None:
278278
self._safety = safety
279279

280-
self.openai_moderations = to_streamed_response_wrapper(
281-
safety.openai_moderations,
280+
self.create = to_streamed_response_wrapper(
281+
safety.create,
282282
)
283283
self.run_shield = to_streamed_response_wrapper(
284284
safety.run_shield,
@@ -289,8 +289,8 @@ class AsyncSafetyResourceWithStreamingResponse:
289289
def __init__(self, safety: AsyncSafetyResource) -> None:
290290
self._safety = safety
291291

292-
self.openai_moderations = async_to_streamed_response_wrapper(
293-
safety.openai_moderations,
292+
self.create = async_to_streamed_response_wrapper(
293+
safety.create,
294294
)
295295
self.run_shield = async_to_streamed_response_wrapper(
296296
safety.run_shield,

src/llama_stack_client/types/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
from .tool_response import ToolResponse as ToolResponse
4848
from .inference_step import InferenceStep as InferenceStep
4949
from .tool_def_param import ToolDefParam as ToolDefParam
50+
from .create_response import CreateResponse as CreateResponse
5051
from .response_object import ResponseObject as ResponseObject
5152
from .token_log_probs import TokenLogProbs as TokenLogProbs
5253
from .file_list_params import FileListParams as FileListParams
@@ -77,6 +78,7 @@
7778
from .list_routes_response import ListRoutesResponse as ListRoutesResponse
7879
from .query_spans_response import QuerySpansResponse as QuerySpansResponse
7980
from .response_list_params import ResponseListParams as ResponseListParams
81+
from .safety_create_params import SafetyCreateParams as SafetyCreateParams
8082
from .scoring_score_params import ScoringScoreParams as ScoringScoreParams
8183
from .shield_list_response import ShieldListResponse as ShieldListResponse
8284
from .agent_create_response import AgentCreateResponse as AgentCreateResponse
@@ -133,7 +135,6 @@
133135
from .inference_completion_params import InferenceCompletionParams as InferenceCompletionParams
134136
from .inference_embeddings_params import InferenceEmbeddingsParams as InferenceEmbeddingsParams
135137
from .list_vector_stores_response import ListVectorStoresResponse as ListVectorStoresResponse
136-
from .openai_moderations_response import OpenAIModerationsResponse as OpenAIModerationsResponse
137138
from .telemetry_get_span_response import TelemetryGetSpanResponse as TelemetryGetSpanResponse
138139
from .vector_db_register_response import VectorDBRegisterResponse as VectorDBRegisterResponse
139140
from .vector_db_retrieve_response import VectorDBRetrieveResponse as VectorDBRetrieveResponse
@@ -152,7 +153,6 @@
152153
from .tool_runtime_invoke_tool_params import ToolRuntimeInvokeToolParams as ToolRuntimeInvokeToolParams
153154
from .inference_chat_completion_params import InferenceChatCompletionParams as InferenceChatCompletionParams
154155
from .list_post_training_jobs_response import ListPostTrainingJobsResponse as ListPostTrainingJobsResponse
155-
from .safety_openai_moderations_params import SafetyOpenAIModerationsParams as SafetyOpenAIModerationsParams
156156
from .scoring_function_register_params import ScoringFunctionRegisterParams as ScoringFunctionRegisterParams
157157
from .telemetry_get_span_tree_response import TelemetryGetSpanTreeResponse as TelemetryGetSpanTreeResponse
158158
from .tool_runtime_list_tools_response import ToolRuntimeListToolsResponse as ToolRuntimeListToolsResponse

src/llama_stack_client/types/chat/completion_create_params.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
"MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam",
1414
"MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam",
1515
"MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParamImageURL",
16+
"MessageOpenAIUserMessageParamContentUnionMember1OpenAIFile",
17+
"MessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile",
1618
"MessageOpenAISystemMessageParam",
1719
"MessageOpenAISystemMessageParamContentUnionMember1",
1820
"MessageOpenAIAssistantMessageParam",
@@ -127,9 +129,24 @@ class MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionConten
127129
type: Required[Literal["image_url"]]
128130

129131

132+
class MessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile(TypedDict, total=False):
133+
file_data: str
134+
135+
file_id: str
136+
137+
filename: str
138+
139+
140+
class MessageOpenAIUserMessageParamContentUnionMember1OpenAIFile(TypedDict, total=False):
141+
file: Required[MessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile]
142+
143+
type: Required[Literal["file"]]
144+
145+
130146
MessageOpenAIUserMessageParamContentUnionMember1: TypeAlias = Union[
131147
MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam,
132148
MessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam,
149+
MessageOpenAIUserMessageParamContentUnionMember1OpenAIFile,
133150
]
134151

135152

src/llama_stack_client/types/chat/completion_create_response.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
"OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam",
1818
"OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam",
1919
"OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParamImageURL",
20+
"OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile",
21+
"OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile",
2022
"OpenAIChatCompletionChoiceMessageOpenAISystemMessageParam",
2123
"OpenAIChatCompletionChoiceMessageOpenAISystemMessageParamContentUnionMember1",
2224
"OpenAIChatCompletionChoiceMessageOpenAIAssistantMessageParam",
@@ -59,10 +61,25 @@ class OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1
5961
type: Literal["image_url"]
6062

6163

64+
class OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile(BaseModel):
65+
file_data: Optional[str] = None
66+
67+
file_id: Optional[str] = None
68+
69+
filename: Optional[str] = None
70+
71+
72+
class OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile(BaseModel):
73+
file: OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile
74+
75+
type: Literal["file"]
76+
77+
6278
OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1: TypeAlias = Annotated[
6379
Union[
6480
OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam,
6581
OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam,
82+
OpenAIChatCompletionChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile,
6683
],
6784
PropertyInfo(discriminator="type"),
6885
]

src/llama_stack_client/types/chat/completion_list_response.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
"DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam",
1717
"DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam",
1818
"DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParamImageURL",
19+
"DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile",
20+
"DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile",
1921
"DataChoiceMessageOpenAISystemMessageParam",
2022
"DataChoiceMessageOpenAISystemMessageParamContentUnionMember1",
2123
"DataChoiceMessageOpenAIAssistantMessageParam",
@@ -37,6 +39,8 @@
3739
"DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam",
3840
"DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam",
3941
"DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParamImageURL",
42+
"DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile",
43+
"DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile",
4044
"DataInputMessageOpenAISystemMessageParam",
4145
"DataInputMessageOpenAISystemMessageParamContentUnionMember1",
4246
"DataInputMessageOpenAIAssistantMessageParam",
@@ -72,10 +76,25 @@ class DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatComple
7276
type: Literal["image_url"]
7377

7478

79+
class DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile(BaseModel):
80+
file_data: Optional[str] = None
81+
82+
file_id: Optional[str] = None
83+
84+
filename: Optional[str] = None
85+
86+
87+
class DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile(BaseModel):
88+
file: DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile
89+
90+
type: Literal["file"]
91+
92+
7593
DataChoiceMessageOpenAIUserMessageParamContentUnionMember1: TypeAlias = Annotated[
7694
Union[
7795
DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam,
7896
DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam,
97+
DataChoiceMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile,
7998
],
8099
PropertyInfo(discriminator="type"),
81100
]
@@ -276,10 +295,25 @@ class DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatComplet
276295
type: Literal["image_url"]
277296

278297

298+
class DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile(BaseModel):
299+
file_data: Optional[str] = None
300+
301+
file_id: Optional[str] = None
302+
303+
filename: Optional[str] = None
304+
305+
306+
class DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile(BaseModel):
307+
file: DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFileFile
308+
309+
type: Literal["file"]
310+
311+
279312
DataInputMessageOpenAIUserMessageParamContentUnionMember1: TypeAlias = Annotated[
280313
Union[
281314
DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartTextParam,
282315
DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIChatCompletionContentPartImageParam,
316+
DataInputMessageOpenAIUserMessageParamContentUnionMember1OpenAIFile,
283317
],
284318
PropertyInfo(discriminator="type"),
285319
]

0 commit comments

Comments
 (0)