Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions src/openai/resources/containers/containers.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,16 @@

from __future__ import annotations

from typing import Iterable, Union

from typing_extensions import Literal

import httpx

from ... import _legacy_response
from ...types import container_list_params, container_create_params
from ...types.responses.inline_skill_param import InlineSkillParam
from ...types.responses.skill_reference_param import SkillReferenceParam
from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, SequenceNotStr, omit, not_given
from ..._utils import maybe_transform, async_maybe_transform
from ..._compat import cached_property
Expand Down Expand Up @@ -61,6 +65,7 @@ def create(
expires_after: container_create_params.ExpiresAfter | Omit = omit,
file_ids: SequenceNotStr[str] | Omit = omit,
memory_limit: Literal["1g", "4g", "16g", "64g"] | Omit = omit,
skills: Iterable[Union[SkillReferenceParam, InlineSkillParam]] | Omit = omit,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
extra_headers: Headers | None = None,
Expand All @@ -80,6 +85,8 @@ def create(

memory_limit: Optional memory limit for the container. Defaults to "1g".

skills: Optional list of skills referenced by id or inline data.

extra_headers: Send extra headers

extra_query: Add additional query parameters to the request
Expand All @@ -96,6 +103,7 @@ def create(
"expires_after": expires_after,
"file_ids": file_ids,
"memory_limit": memory_limit,
"skills": skills,
},
container_create_params.ContainerCreateParams,
),
Expand Down Expand Up @@ -261,6 +269,7 @@ async def create(
expires_after: container_create_params.ExpiresAfter | Omit = omit,
file_ids: SequenceNotStr[str] | Omit = omit,
memory_limit: Literal["1g", "4g", "16g", "64g"] | Omit = omit,
skills: Iterable[Union[SkillReferenceParam, InlineSkillParam]] | Omit = omit,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
extra_headers: Headers | None = None,
Expand All @@ -280,6 +289,8 @@ async def create(

memory_limit: Optional memory limit for the container. Defaults to "1g".

skills: Optional list of skills referenced by id or inline data.

extra_headers: Send extra headers

extra_query: Add additional query parameters to the request
Expand All @@ -296,6 +307,7 @@ async def create(
"expires_after": expires_after,
"file_ids": file_ids,
"memory_limit": memory_limit,
"skills": skills,
},
container_create_params.ContainerCreateParams,
),
Expand Down
7 changes: 7 additions & 0 deletions src/openai/types/container_create_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,13 @@

from __future__ import annotations

from typing import Iterable, Union

from typing_extensions import Literal, Required, TypedDict

from .._types import SequenceNotStr
from .responses.inline_skill_param import InlineSkillParam
from .responses.skill_reference_param import SkillReferenceParam

__all__ = ["ContainerCreateParams", "ExpiresAfter"]

Expand All @@ -22,6 +26,9 @@ class ContainerCreateParams(TypedDict, total=False):
memory_limit: Literal["1g", "4g", "16g", "64g"]
"""Optional memory limit for the container. Defaults to "1g"."""

skills: Iterable[Union[SkillReferenceParam, InlineSkillParam]]
"""Optional list of skills referenced by id or inline data."""


class ExpiresAfter(TypedDict, total=False):
"""Container expiration time in seconds relative to the 'anchor' time."""
Expand Down
8 changes: 8 additions & 0 deletions tests/api_resources/test_containers.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,10 @@ def test_method_create_with_all_params(self, client: OpenAI) -> None:
},
file_ids=["string"],
memory_limit="1g",
skills=[
{"type": "skill_reference", "skill_id": "skill_4db6f1a2c9e73508b41f9da06e2c7b5f"},
{"type": "skill_reference", "skill_id": "openai-spreadsheets", "version": "latest"},
],
)
assert_matches_type(ContainerCreateResponse, container, path=["response"])

Expand Down Expand Up @@ -199,6 +203,10 @@ async def test_method_create_with_all_params(self, async_client: AsyncOpenAI) ->
},
file_ids=["string"],
memory_limit="1g",
skills=[
{"type": "skill_reference", "skill_id": "skill_4db6f1a2c9e73508b41f9da06e2c7b5f"},
{"type": "skill_reference", "skill_id": "openai-spreadsheets", "version": "latest"},
],
)
assert_matches_type(ContainerCreateResponse, container, path=["response"])

Expand Down