mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zooqueenandHanzo Dev
b144973bfc
Fix 35 ruff errors (E722, F841, B904, B905, B007, E741, S202, S310, S306, S110, TID251, UP042, F401) and run black on 526 files. Co-authored-by: Hanzo Dev <[email protected]>
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
import os
|
|
|
|
import openai
|
|
import pytest
|
|
|
|
from agents import (
|
|
set_default_openai_api,
|
|
set_default_openai_client,
|
|
set_default_openai_key,
|
|
)
|
|
from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
|
|
from agents.models.openai_provider import OpenAIProvider
|
|
from agents.models.openai_responses import OpenAIResponsesModel
|
|
|
|
|
|
def test_cc_no_default_key_errors(monkeypatch):
|
|
monkeypatch.delenv("OPENAI_API_KEY", raising=False)
|
|
with pytest.raises(openai.OpenAIError):
|
|
OpenAIProvider(use_responses=False).get_model("gpt-4")
|
|
|
|
|
|
def test_cc_set_default_openai_key():
|
|
set_default_openai_key("test_key")
|
|
chat_model = OpenAIProvider(use_responses=False).get_model("gpt-4")
|
|
assert chat_model._client.api_key == "test_key" # type: ignore
|
|
|
|
|
|
def test_cc_set_default_openai_client():
|
|
client = openai.AsyncOpenAI(api_key="test_key")
|
|
set_default_openai_client(client)
|
|
chat_model = OpenAIProvider(use_responses=False).get_model("gpt-4")
|
|
assert chat_model._client.api_key == "test_key" # type: ignore
|
|
|
|
|
|
def test_resp_no_default_key_errors(monkeypatch):
|
|
monkeypatch.delenv("OPENAI_API_KEY", raising=False)
|
|
assert os.getenv("OPENAI_API_KEY") is None
|
|
with pytest.raises(openai.OpenAIError):
|
|
OpenAIProvider(use_responses=True).get_model("gpt-4")
|
|
|
|
|
|
def test_resp_set_default_openai_key():
|
|
set_default_openai_key("test_key")
|
|
resp_model = OpenAIProvider(use_responses=True).get_model("gpt-4")
|
|
assert resp_model._client.api_key == "test_key" # type: ignore
|
|
|
|
|
|
def test_resp_set_default_openai_client():
|
|
client = openai.AsyncOpenAI(api_key="test_key")
|
|
set_default_openai_client(client)
|
|
resp_model = OpenAIProvider(use_responses=True).get_model("gpt-4")
|
|
assert resp_model._client.api_key == "test_key" # type: ignore
|
|
|
|
|
|
def test_set_default_openai_api():
|
|
assert isinstance(
|
|
OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel
|
|
), "Default should be responses"
|
|
|
|
set_default_openai_api("chat_completions")
|
|
assert isinstance(
|
|
OpenAIProvider().get_model("gpt-4"), OpenAIChatCompletionsModel
|
|
), "Should be chat completions model"
|
|
|
|
set_default_openai_api("responses")
|
|
assert isinstance(
|
|
OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel
|
|
), "Should be responses model"
|