openai-python/tests/test_module_client.py at main · brunotech/openai-python · GitHub

History

182 lines (124 loc) · 5.28 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

# File generated from our OpenAPI spec by Stainless.

from __future__ import annotations

import os as _os

import httpx

import pytest

from httpx import URL

import openai

from openai import DEFAULT_TIMEOUT, DEFAULT_MAX_RETRIES

def reset_state() -> None:

openai._reset_client()

openai.api_key = None or "My API Key"

openai.organization = None

openai.base_url = None

openai.timeout = DEFAULT_TIMEOUT

openai.max_retries = DEFAULT_MAX_RETRIES

openai.default_headers = None

openai.default_query = None

openai.http_client = None

openai.api_type = _os.environ.get("OPENAI_API_TYPE") # type: ignore

openai.api_version = None

openai.azure_endpoint = None

openai.azure_ad_token = None

openai.azure_ad_token_provider = None

@pytest.fixture(autouse=True)

def reset_state_fixture() -> None:

reset_state()

def test_base_url_option() -> None:

assert openai.base_url is None

assert openai.completions._client.base_url == URL("https://api.openai.com/v1/")

openai.base_url = "http://foo.com"

assert openai.base_url == URL("http://foo.com")

assert openai.completions._client.base_url == URL("http://foo.com")

def test_timeout_option() -> None:

assert openai.timeout == openai.DEFAULT_TIMEOUT

assert openai.completions._client.timeout == openai.DEFAULT_TIMEOUT

openai.timeout = 3

assert openai.timeout == 3

assert openai.completions._client.timeout == 3

def test_max_retries_option() -> None:

assert openai.max_retries == openai.DEFAULT_MAX_RETRIES

assert openai.completions._client.max_retries == openai.DEFAULT_MAX_RETRIES

openai.max_retries = 1

assert openai.max_retries == 1

assert openai.completions._client.max_retries == 1

def test_default_headers_option() -> None:

assert openai.default_headers == None

openai.default_headers = {"Foo": "Bar"}

assert openai.default_headers["Foo"] == "Bar"

assert openai.completions._client.default_headers["Foo"] == "Bar"

def test_default_query_option() -> None:

assert openai.default_query is None

assert openai.completions._client._custom_query == {}

openai.default_query = {"Foo": {"nested": 1}}

assert openai.default_query["Foo"] == {"nested": 1}

assert openai.completions._client._custom_query["Foo"] == {"nested": 1}

def test_http_client_option() -> None:

assert openai.http_client is None

original_http_client = openai.completions._client._client

assert original_http_client is not None

new_client = httpx.Client()

openai.http_client = new_client

assert openai.completions._client._client is new_client

import contextlib

from typing import Iterator

from openai.lib.azure import AzureOpenAI

@contextlib.contextmanager

def fresh_env() -> Iterator[None]:

old = _os.environ.copy()

try:

_os.environ.clear()

yield

finally:

_os.environ.update(old)

def test_only_api_key_results_in_openai_api() -> None:

with fresh_env():

openai.api_type = None

openai.api_key = "example API key"

assert type(openai.completions._client).__name__ == "_ModuleClient"

def test_azure_api_key_env_without_api_version() -> None:

with fresh_env():

openai.api_type = None

_os.environ["AZURE_OPENAI_API_KEY"] = "example API key"

with pytest.raises(

ValueError,

match=r"Must provide either the `api_version` argument or the `OPENAI_API_VERSION` environment variable",

openai.completions._client

def test_azure_api_key_and_version_env() -> None:

with fresh_env():

openai.api_type = None

_os.environ["AZURE_OPENAI_API_KEY"] = "example API key"

_os.environ["OPENAI_API_VERSION"] = "example-version"

with pytest.raises(

ValueError,

match=r"Must provide one of the `base_url` or `azure_endpoint` arguments, or the `AZURE_OPENAI_ENDPOINT` environment variable",

openai.completions._client

def test_azure_api_key_version_and_endpoint_env() -> None:

with fresh_env():

openai.api_type = None

_os.environ["AZURE_OPENAI_API_KEY"] = "example API key"

_os.environ["OPENAI_API_VERSION"] = "example-version"

_os.environ["AZURE_OPENAI_ENDPOINT"] = "https://www.example"

openai.completions._client

assert openai.api_type == "azure"

def test_azure_azure_ad_token_version_and_endpoint_env() -> None:

with fresh_env():

openai.api_type = None

_os.environ["AZURE_OPENAI_AD_TOKEN"] = "example AD token"

_os.environ["OPENAI_API_VERSION"] = "example-version"

_os.environ["AZURE_OPENAI_ENDPOINT"] = "https://www.example"

client = openai.completions._client

assert isinstance(client, AzureOpenAI)

assert client._azure_ad_token == "example AD token"

def test_azure_azure_ad_token_provider_version_and_endpoint_env() -> None:

with fresh_env():

openai.api_type = None

_os.environ["OPENAI_API_VERSION"] = "example-version"

_os.environ["AZURE_OPENAI_ENDPOINT"] = "https://www.example"

openai.azure_ad_token_provider = lambda: "token"

client = openai.completions._client

assert isinstance(client, AzureOpenAI)

assert client._azure_ad_token_provider is not None

assert client._azure_ad_token_provider() == "token"

openai-python/tests/test_module_client.py at main · brunotech/openai-python · GitHub

Expand file tree

Breadcrumbs

test_module_client.py

Latest commit

History

Breadcrumbs

test_module_client.py

File metadata and controls