示例#1
0
def test_completions_url_composition_azure() -> None:
    url = Completion.class_url("test_engine", "azure", '2021-11-01-preview')
    assert url == '/openai/deployments/test_engine/completions?api-version=2021-11-01-preview'
示例#2
0
def test_completions_url_composition_instance_url_timeout_azure() -> None:
    completion = Completion(id="test_id", engine="test_engine", api_type="azure", api_version='2021-11-01-preview')
    completion["timeout"] = 12
    url = completion.instance_url()
    assert url == "/openai/deployments/test_engine/completions/test_id?api-version=2021-11-01-preview&timeout=12"
示例#3
0
def test_completions_url_composition_instance_url_timeout_openai() -> None:
    completion = Completion(id="test_id", engine="test_engine", api_type="open_ai" )
    completion["timeout"] = 12
    url = completion.instance_url()
    assert url == "/engines/test_engine/completions/test_id?timeout=12"
示例#4
0
def test_completions_url_composition_instance_url_open_ai() -> None:
    completion = Completion(id="test_id", engine="test_engine", api_type="open_ai", api_version='2021-11-01-preview')
    url = completion.instance_url()
    assert url == "/engines/test_engine/completions/test_id"
示例#5
0
def test_completions_url_composition_instance_url_invalid() -> None:
    completion = Completion(id="test_id", engine="test_engine", api_type="invalid")
    with pytest.raises(Exception):
        url = completion.instance_url()
示例#6
0
def test_completions_url_composition_instance_url_default() -> None:
    completion = Completion(id="test_id", engine="test_engine")
    url = completion.instance_url()
    assert url == "/engines/test_engine/completions/test_id"
示例#7
0
def test_completions_url_composition_instance_url_azure_no_version() -> None:
    completion = Completion(id="test_id", engine="test_engine", api_type="azure", api_version=None)
    with pytest.raises(Exception):
        completion.instance_url()
示例#8
0
def test_completions_url_composition_invalid_type() -> None:
    with pytest.raises(Exception):
        url = Completion.class_url("test_engine", "invalid")
示例#9
0
def test_completions_url_composition_open_ai() -> None:
    url = Completion.class_url("test_engine", "open_ai")
    assert url == '/engines/test_engine/completions'
示例#10
0
def test_completions_url_composition_default() -> None:
    url = Completion.class_url("test_engine")
    assert url == '/engines/test_engine/completions'