kopia lustrzana https://github.com/gd3kr/BlenderGPT
58 wiersze
1.9 KiB
Python
58 wiersze
1.9 KiB
Python
import time
|
|
|
|
from openai import util, error
|
|
from openai.api_resources.abstract.engine_api_resource import EngineAPIResource
|
|
from openai.error import TryAgain
|
|
|
|
|
|
class Edit(EngineAPIResource):
|
|
OBJECT_NAME = "edits"
|
|
|
|
@classmethod
|
|
def create(cls, *args, **kwargs):
|
|
"""
|
|
Creates a new edit for the provided input, instruction, and parameters.
|
|
"""
|
|
start = time.time()
|
|
timeout = kwargs.pop("timeout", None)
|
|
|
|
api_type = kwargs.pop("api_type", None)
|
|
typed_api_type = cls._get_api_type_and_version(api_type=api_type)[0]
|
|
if typed_api_type in (util.ApiType.AZURE, util.ApiType.AZURE_AD):
|
|
raise error.InvalidAPIType(
|
|
"This operation is not supported by the Azure OpenAI API yet."
|
|
)
|
|
|
|
while True:
|
|
try:
|
|
return super().create(*args, **kwargs)
|
|
except TryAgain as e:
|
|
if timeout is not None and time.time() > start + timeout:
|
|
raise
|
|
|
|
util.log_info("Waiting for model to warm up", error=e)
|
|
|
|
@classmethod
|
|
async def acreate(cls, *args, **kwargs):
|
|
"""
|
|
Creates a new edit for the provided input, instruction, and parameters.
|
|
"""
|
|
start = time.time()
|
|
timeout = kwargs.pop("timeout", None)
|
|
|
|
api_type = kwargs.pop("api_type", None)
|
|
typed_api_type = cls._get_api_type_and_version(api_type=api_type)[0]
|
|
if typed_api_type in (util.ApiType.AZURE, util.ApiType.AZURE_AD):
|
|
raise error.InvalidAPIType(
|
|
"This operation is not supported by the Azure OpenAI API yet."
|
|
)
|
|
|
|
while True:
|
|
try:
|
|
return await super().acreate(*args, **kwargs)
|
|
except TryAgain as e:
|
|
if timeout is not None and time.time() > start + timeout:
|
|
raise
|
|
|
|
util.log_info("Waiting for model to warm up", error=e)
|