Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix: fix langchain error handling to avoid crash on openai>1.0.0 #2307

Merged
merged 3 commits into from
Nov 12, 2024
Merged

fix: fix langchain error handling to avoid crash on openai>1.0.0 #2307

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
3c21203
make langchain error handling support openai>1.0
mslhrotk Nov 5, 2024
da95b6d
fix lint
mslhrotk Nov 5, 2024
709cb9d
Merge branch 'master' into cl/langchain-1.0
mhamilton723 Nov 8, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 16 additions & 9 deletions cognitive/src/main/python/synapse/ml/services/langchain/LangchainTransform.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -233,6 +233,7 @@ def _transform(self, dataset):
@udf(schema)
def udfFunction(x):
import openai
from packaging import version

if self.running_on_synapse_internal and not self.isSet(self.url):
from synapse.ml.fabric.prerun.openai_prerun import OpenAIPrerun
Expand All @@ -244,15 +245,21 @@ def udfFunction(x):
openai.api_base = self.getUrl()
openai.api_version = self.getApiVersion()

error_messages = {
openai.error.Timeout: "OpenAI API request timed out, please retry your request after a brief wait and contact us if the issue persists: {}",
openai.error.APIError: "OpenAI API returned an API Error: {}",
openai.error.APIConnectionError: "OpenAI API request failed to connect, check your network settings, proxy configuration, SSL certificates, or firewall rules: {}",
openai.error.InvalidRequestError: "OpenAI API request was invalid: {}",
openai.error.AuthenticationError: "OpenAI API request was not authorized, please check your API key or token and make sure it is correct and active. You may need to generate a new one from your account dashboard: {}",
openai.error.PermissionError: "OpenAI API request was not permitted, make sure your API key has the appropriate permissions for the action or model accessed: {}",
openai.error.RateLimitError: "OpenAI API request exceeded rate limit: {}",
}
error_messages = {}
if version.parse(openai.__version__) < version.parse("1.0.0"):
error_messages = {
openai.error.Timeout: "OpenAI API request timed out, please retry your request after a brief wait and contact us if the issue persists: {}",
openai.error.APIError: "OpenAI API returned an API Error: {}",
openai.error.APIConnectionError: "OpenAI API request failed to connect, check your network settings, proxy configuration, SSL certificates, or firewall rules: {}",
openai.error.InvalidRequestError: "OpenAI API request was invalid: {}",
openai.error.AuthenticationError: "OpenAI API request was not authorized, please check your API key or token and make sure it is correct and active. You may need to generate a new one from your account dashboard: {}",
openai.error.PermissionError: "OpenAI API request was not permitted, make sure your API key has the appropriate permissions for the action or model accessed: {}",
openai.error.RateLimitError: "OpenAI API request exceeded rate limit: {}",
}
else:
{
openai.OpenAIError: "OpenAI API returned an API Error: {}",
}

try:
result = self.getChain().run(x)
Expand Down
Loading