30 lines
880 B
Python
30 lines
880 B
Python
import logging
|
|
import logging.config
|
|
import time
|
|
|
|
import zai
|
|
from zai import ZaiClient
|
|
|
|
|
|
def test_completions_temp0(logging_conf):
|
|
logging.config.dictConfig(logging_conf) # type: ignore
|
|
# Skip this test if no valid API key is provided
|
|
import os
|
|
if not os.environ.get('ZAI_API_KEY') or os.environ.get('ZAI_API_KEY') == '{your apikey}':
|
|
import pytest
|
|
pytest.skip("No valid API key provided for integration test")
|
|
|
|
client = ZaiClient(disable_token_cache=False)
|
|
try:
|
|
# Generate request_id
|
|
request_id = time.time()
|
|
print(f'request_id:{request_id}')
|
|
response = client.moderations.create(model='moderation', input={'type': 'text', 'text': 'hello world '})
|
|
print(response)
|
|
|
|
except zai.core._errors.APIRequestFailedError as err:
|
|
print(err)
|
|
except zai.core._errors.APIInternalError as err:
|
|
print(err)
|
|
except zai.core._errors.APIStatusError as err:
|
|
print(err)
|