1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from __future__ import annotations
import pytest
from openai import OpenAI, AsyncOpenAI
from openai._utils import assert_signatures_in_sync
@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
def test_create_and_run_poll_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
assert_signatures_in_sync(
checking_client.beta.threads.create_and_run, # pyright: ignore[reportDeprecated]
checking_client.beta.threads.create_and_run_poll,
exclude_params={"stream"},
)
@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
def test_create_and_run_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
assert_signatures_in_sync(
checking_client.beta.threads.create_and_run, # pyright: ignore[reportDeprecated]
checking_client.beta.threads.create_and_run_stream,
exclude_params={"stream"},
)
@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
def test_run_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
assert_signatures_in_sync(
checking_client.beta.threads.runs.create, # pyright: ignore[reportDeprecated]
checking_client.beta.threads.runs.stream, # pyright: ignore[reportDeprecated]
exclude_params={"stream"},
)
@pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"])
def test_create_and_poll_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None:
checking_client: OpenAI | AsyncOpenAI = client if sync else async_client
assert_signatures_in_sync(
checking_client.beta.threads.runs.create, # pyright: ignore[reportDeprecated]
checking_client.beta.threads.runs.create_and_poll, # pyright: ignore[reportDeprecated]
exclude_params={"stream"},
)
|
