Skip to content

Commit

Permalink
fix(api): correct types for attachments (#1342)
Browse files Browse the repository at this point in the history
  • Loading branch information
stainless-bot committed Apr 18, 2024
1 parent 9e0ea64 commit 542d30c
Show file tree
Hide file tree
Showing 8 changed files with 106 additions and 106 deletions.
4 changes: 2 additions & 2 deletions src/openai/types/beta/thread_create_and_run_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,11 +171,11 @@ class ThreadCreateAndRunParamsBase(TypedDict, total=False):


class ThreadMessageAttachment(TypedDict, total=False):
add_to: List[Literal["file_search", "code_interpreter"]]

file_id: str
"""The ID of the file to attach to the message."""

tools: List[Literal["file_search", "code_interpreter"]]


class ThreadMessage(TypedDict, total=False):
content: Required[str]
Expand Down
4 changes: 2 additions & 2 deletions src/openai/types/beta/thread_create_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,11 +41,11 @@ class ThreadCreateParams(TypedDict, total=False):


class MessageAttachment(TypedDict, total=False):
add_to: List[Literal["file_search", "code_interpreter"]]

file_id: str
"""The ID of the file to attach to the message."""

tools: List[Literal["file_search", "code_interpreter"]]


class Message(TypedDict, total=False):
content: Required[str]
Expand Down
4 changes: 2 additions & 2 deletions src/openai/types/beta/threads/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,11 @@


class Attachment(BaseModel):
add_to: Optional[List[Literal["file_search", "code_interpreter"]]] = None

file_id: Optional[str] = None
"""The ID of the file to attach to the message."""

tools: Optional[List[Literal["file_search", "code_interpreter"]]] = None


class IncompleteDetails(BaseModel):
reason: Literal["content_filter", "max_tokens", "run_cancelled", "run_expired", "run_failed"]
Expand Down
4 changes: 2 additions & 2 deletions src/openai/types/beta/threads/message_create_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ class MessageCreateParams(TypedDict, total=False):


class Attachment(TypedDict, total=False):
add_to: List[Literal["file_search", "code_interpreter"]]

file_id: str
"""The ID of the file to attach to the message."""

tools: List[Literal["file_search", "code_interpreter"]]
4 changes: 2 additions & 2 deletions src/openai/types/beta/threads/run_create_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,11 +160,11 @@ class RunCreateParamsBase(TypedDict, total=False):


class AdditionalMessageAttachment(TypedDict, total=False):
add_to: List[Literal["file_search", "code_interpreter"]]

file_id: str
"""The ID of the file to attach to the message."""

tools: List[Literal["file_search", "code_interpreter"]]


class AdditionalMessage(TypedDict, total=False):
content: Required[str]
Expand Down
108 changes: 54 additions & 54 deletions tests/api_resources/beta/test_threads.py

Large diffs are not rendered by default.

12 changes: 6 additions & 6 deletions tests/api_resources/beta/threads/test_messages.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,15 +36,15 @@ def test_method_create_with_all_params(self, client: OpenAI) -> None:
attachments=[
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
metadata={},
Expand Down Expand Up @@ -265,15 +265,15 @@ async def test_method_create_with_all_params(self, async_client: AsyncOpenAI) ->
attachments=[
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
metadata={},
Expand Down
72 changes: 36 additions & 36 deletions tests/api_resources/beta/threads/test_runs.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,15 +43,15 @@ def test_method_create_with_all_params_overload_1(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -62,15 +62,15 @@ def test_method_create_with_all_params_overload_1(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -81,15 +81,15 @@ def test_method_create_with_all_params_overload_1(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand Down Expand Up @@ -170,15 +170,15 @@ def test_method_create_with_all_params_overload_2(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -189,15 +189,15 @@ def test_method_create_with_all_params_overload_2(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -208,15 +208,15 @@ def test_method_create_with_all_params_overload_2(self, client: OpenAI) -> None:
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand Down Expand Up @@ -635,15 +635,15 @@ async def test_method_create_with_all_params_overload_1(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -654,15 +654,15 @@ async def test_method_create_with_all_params_overload_1(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -673,15 +673,15 @@ async def test_method_create_with_all_params_overload_1(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand Down Expand Up @@ -762,15 +762,15 @@ async def test_method_create_with_all_params_overload_2(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -781,15 +781,15 @@ async def test_method_create_with_all_params_overload_2(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand All @@ -800,15 +800,15 @@ async def test_method_create_with_all_params_overload_2(self, async_client: Asyn
"attachments": [
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
{
"file_id": "string",
"add_to": ["file_search", "code_interpreter"],
"tools": ["file_search", "code_interpreter"],
},
],
"metadata": {},
Expand Down

0 comments on commit 542d30c

Please sign in to comment.