Skip to content
Snippets Groups Projects
Commit 7c6bf69c authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

raise openai errors

parent a4504514
No related branches found
No related tags found
No related merge requests found
...@@ -123,11 +123,13 @@ class OpenAIEncoder(BaseEncoder): ...@@ -123,11 +123,13 @@ class OpenAIEncoder(BaseEncoder):
break break
except OpenAIError as e: except OpenAIError as e:
logger.error("Exception occurred", exc_info=True) logger.error("Exception occurred", exc_info=True)
if self.max_retries != 0: if self.max_retries != 0 and j < self.max_retries:
sleep(2**j) sleep(2**j)
logger.warning( logger.warning(
f"Retrying in {2**j} seconds due to OpenAIError: {e}" f"Retrying in {2**j} seconds due to OpenAIError: {e}"
) )
else:
raise
except Exception as e: except Exception as e:
logger.error(f"OpenAI API call failed. Error: {e}") logger.error(f"OpenAI API call failed. Error: {e}")
...@@ -178,11 +180,14 @@ class OpenAIEncoder(BaseEncoder): ...@@ -178,11 +180,14 @@ class OpenAIEncoder(BaseEncoder):
break break
except OpenAIError as e: except OpenAIError as e:
logger.error("Exception occurred", exc_info=True) logger.error("Exception occurred", exc_info=True)
if self.max_retries != 0: if self.max_retries != 0 and j < self.max_retries:
await asleep(2**j) await asleep(2**j)
logger.warning( logger.warning(
f"Retrying in {2**j} seconds due to OpenAIError: {e}" f"Retrying in {2**j} seconds due to OpenAIError: {e}"
) )
else:
raise
except Exception as e: except Exception as e:
logger.error(f"OpenAI API call failed. Error: {e}") logger.error(f"OpenAI API call failed. Error: {e}")
raise ValueError(f"OpenAI API call failed. Error: {e}") from e raise ValueError(f"OpenAI API call failed. Error: {e}") from e
......
...@@ -114,11 +114,13 @@ class AzureOpenAIEncoder(BaseEncoder): ...@@ -114,11 +114,13 @@ class AzureOpenAIEncoder(BaseEncoder):
break break
except OpenAIError as e: except OpenAIError as e:
logger.error("Exception occurred", exc_info=True) logger.error("Exception occurred", exc_info=True)
if self.max_retries != 0: if self.max_retries != 0 and j < self.max_retries:
sleep(2**j) sleep(2**j)
logger.warning( logger.warning(
f"Retrying in {2**j} seconds due to OpenAIError: {e}" f"Retrying in {2**j} seconds due to OpenAIError: {e}"
) )
else:
raise
except Exception as e: except Exception as e:
logger.error(f"Azure OpenAI API call failed. Error: {e}") logger.error(f"Azure OpenAI API call failed. Error: {e}")
raise ValueError(f"Azure OpenAI API call failed. Error: {e}") from e raise ValueError(f"Azure OpenAI API call failed. Error: {e}") from e
...@@ -151,11 +153,13 @@ class AzureOpenAIEncoder(BaseEncoder): ...@@ -151,11 +153,13 @@ class AzureOpenAIEncoder(BaseEncoder):
except OpenAIError as e: except OpenAIError as e:
logger.error("Exception occurred", exc_info=True) logger.error("Exception occurred", exc_info=True)
if self.max_retries != 0: if self.max_retries != 0 and j < self.max_retries:
await asleep(2**j) await asleep(2**j)
logger.warning( logger.warning(
f"Retrying in {2**j} seconds due to OpenAIError: {e}" f"Retrying in {2**j} seconds due to OpenAIError: {e}"
) )
else:
raise
except Exception as e: except Exception as e:
logger.error(f"Azure OpenAI API call failed. Error: {e}") logger.error(f"Azure OpenAI API call failed. Error: {e}")
raise ValueError(f"Azure OpenAI API call failed. Error: {e}") from e raise ValueError(f"Azure OpenAI API call failed. Error: {e}") from e
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment