openai/openai-python

Public

mirrored from https://github.com/openai/openai-pythonAvailable

CodeCommitsIssuesPull requestsActionsInsightsSecurity
v2.35.1

Branches

Tags

  • No tags available.
0Branches0Tags
Go to file
Add file
Code

Clone

HTTPS

Download ZIP

tests/lib/test_assistants.py

50lines · modeblame

a1203812Robert Craigie2 years ago1from __future__ import annotations
2
3import pytest
4
5from openai import OpenAI, AsyncOpenAI
1d228856Robert Craigie2 years ago6from openai._utils import assert_signatures_in_sync
a1203812Robert Craigie2 years ago7
8
9@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
10def test_create_and_run_poll_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
203f094bRobert Craigie2 years ago11checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
a1203812Robert Craigie2 years ago12
13assert_signatures_in_sync(
cca09707stainless-app[bot]1 years ago14checking_client.beta.threads.create_and_run, # pyright: ignore[reportDeprecated]
a1203812Robert Craigie2 years ago15checking_client.beta.threads.create_and_run_poll,
16exclude_params={"stream"},
17)
18
f14f8593Robert Craigie2 years ago19
a1203812Robert Craigie2 years ago20@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
21def test_create_and_run_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
203f094bRobert Craigie2 years ago22checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
a1203812Robert Craigie2 years ago23
24assert_signatures_in_sync(
cca09707stainless-app[bot]1 years ago25checking_client.beta.threads.create_and_run, # pyright: ignore[reportDeprecated]
a1203812Robert Craigie2 years ago26checking_client.beta.threads.create_and_run_stream,
27exclude_params={"stream"},
28)
1d228856Robert Craigie2 years ago29
30
d50550bdRobert Craigie1 years ago31@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
32def test_run_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
33checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
34
35assert_signatures_in_sync(
cca09707stainless-app[bot]1 years ago36checking_client.beta.threads.runs.create, # pyright: ignore[reportDeprecated]
37checking_client.beta.threads.runs.stream, # pyright: ignore[reportDeprecated]
d50550bdRobert Craigie1 years ago38exclude_params={"stream"},
39)
40
41
1d228856Robert Craigie2 years ago42@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
43def test_create_and_poll_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
44checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
45
46assert_signatures_in_sync(
cca09707stainless-app[bot]1 years ago47checking_client.beta.threads.runs.create, # pyright: ignore[reportDeprecated]
48checking_client.beta.threads.runs.create_and_poll, # pyright: ignore[reportDeprecated]
1d228856Robert Craigie2 years ago49exclude_params={"stream"},
50)