Spaces:
Running
Running
File size: 2,868 Bytes
c47ec10 | 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 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | from __future__ import annotations
import json
import unittest
from unittest import mock
import requests
from services.protocol import openai_v1_models
AUTH_KEY = "chatgpt2api"
BASE_URL = "http://localhost:8000"
class ModelListTests(unittest.TestCase):
def test_list_models_only_returns_image_models_backed_by_account_types(self):
with (
mock.patch.object(
openai_v1_models.OpenAIBackendAPI,
"list_models",
return_value={"object": "list", "data": []},
),
mock.patch.object(
openai_v1_models.account_service,
"list_accounts",
return_value=[
{"access_token": "token-free", "type": "free"},
{"access_token": "token-web-team", "type": "Team", "source_type": "web"},
{"access_token": "token-codex-team", "type": "Team", "source_type": "codex"},
],
),
):
result = openai_v1_models.list_models()
ids = {item["id"] for item in result["data"]}
self.assertIn("gpt-image-2", ids)
self.assertIn("codex-gpt-image-2", ids)
self.assertIn("team-codex-gpt-image-2", ids)
self.assertNotIn("plus-codex-gpt-image-2", ids)
self.assertNotIn("pro-codex-gpt-image-2", ids)
def test_list_models_does_not_return_codex_models_for_web_plus_accounts(self):
with (
mock.patch.object(
openai_v1_models.OpenAIBackendAPI,
"list_models",
return_value={"object": "list", "data": []},
),
mock.patch.object(
openai_v1_models.account_service,
"list_accounts",
return_value=[
{"access_token": "token-web-plus", "type": "Plus", "source_type": "web"},
],
),
):
result = openai_v1_models.list_models()
ids = {item["id"] for item in result["data"]}
self.assertIn("gpt-image-2", ids)
self.assertNotIn("codex-gpt-image-2", ids)
self.assertNotIn("plus-codex-gpt-image-2", ids)
def test_list_models_function(self):
"""测试直接调用服务层获取模型列表。"""
result = openai_v1_models.list_models()
print("function result:")
print(json.dumps(result, ensure_ascii=False, indent=2))
def test_list_models_http(self):
"""测试通过 HTTP 接口获取模型列表。"""
response = requests.get(
f"{BASE_URL}/v1/models",
headers={"Authorization": f"Bearer {AUTH_KEY}"},
timeout=30,
)
print("http status:")
print(response.status_code)
print("http result:")
print(json.dumps(response.json(), ensure_ascii=False, indent=2))
|