mirror of
https://github.com/Hopiu/llm.git
synced 2026-03-19 14:10:25 +00:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from click.testing import CliRunner
|
|
from llm.cli import cli
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def mocked_models(requests_mock):
|
|
requests_mock.get(
|
|
"https://api.openai.com/v1/models",
|
|
json={
|
|
"data": [
|
|
{
|
|
"id": "ada:2020-05-03",
|
|
"object": "model",
|
|
"created": 1588537600,
|
|
"owned_by": "openai",
|
|
},
|
|
{
|
|
"id": "babbage:2020-05-03",
|
|
"object": "model",
|
|
"created": 1588537600,
|
|
"owned_by": "openai",
|
|
},
|
|
]
|
|
},
|
|
headers={"Content-Type": "application/json"},
|
|
)
|
|
return requests_mock
|
|
|
|
|
|
def test_openai_models(mocked_models):
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["openai", "models", "--key", "x"])
|
|
assert result.exit_code == 0
|
|
assert result.output == (
|
|
"id owned_by created \n"
|
|
"ada:2020-05-03 openai 2020-05-03T20:26:40\n"
|
|
"babbage:2020-05-03 openai 2020-05-03T20:26:40\n"
|
|
)
|