@ -204,10 +204,17 @@ class BaseOpenAI(BaseLLM):
openai_api_key = get_from_dict_or_env (
openai_api_key = get_from_dict_or_env (
values , " openai_api_key " , " OPENAI_API_KEY "
values , " openai_api_key " , " OPENAI_API_KEY "
)
)
openai_organization = get_from_dict_or_env (
values , " openai_organization " , " OPENAI_ORGANIZATION " , default = None
)
try :
try :
import openai
import openai
openai . api_key = openai_api_key
openai . api_key = openai_api_key
if openai_organization :
print ( " USING ORGANIZATION: " )
print ( openai_organization )
openai . organization = openai_organization
values [ " client " ] = openai . Completion
values [ " client " ] = openai . Completion
except ImportError :
except ImportError :
raise ValueError (
raise ValueError (
@ -588,10 +595,15 @@ class OpenAIChat(BaseLLM):
openai_api_key = get_from_dict_or_env (
openai_api_key = get_from_dict_or_env (
values , " openai_api_key " , " OPENAI_API_KEY "
values , " openai_api_key " , " OPENAI_API_KEY "
)
)
openai_organization = get_from_dict_or_env (
values , " openai_organization " , " OPENAI_ORGANIZATION " , default = None
)
try :
try :
import openai
import openai
openai . api_key = openai_api_key
openai . api_key = openai_api_key
if openai_organization :
openai . organization = openai_organization
except ImportError :
except ImportError :
raise ValueError (
raise ValueError (
" Could not import openai python package. "
" Could not import openai python package. "