Former-commit-id: a6200eb5e95c25618fa085e86aa6a8608e681bf9
@ -1,4 +1,4 @@
from swarms.models import OpenAIChat
from swarms.models import OpenAI
from swarms.structs import Flow
from langchain.schema.messages import ChatMessage, BaseMessage
import os
@ -10,11 +10,10 @@ message: BaseMessage = [ ChatMessage(role="user", content='Translate the followi
api_key = os.environ.get("OPENAI_API_KEY")
# Initialize the language model, this model can be swapped out with Anthropic, ETC, Huggingface Models like Mistral, ETC
llm = OpenAIChat(
llm = OpenAI(
# model_name="gpt-4"
openai_api_key=api_key,
temperature=0.5,
message = message
# max_tokens=100,
)
@ -105,7 +105,7 @@ def _create_retry_decorator(
openai.APIError,
openai.APIConnectionError,
openai.RateLimitError,
openai.error.ServiceUnavailableError,
openai.ServiceUnavailableError,
]
return create_base_retry_decorator(
error_types=errors, max_retries=llm.max_retries, run_manager=run_manager
@ -93,7 +93,7 @@ def _create_retry_decorator(
import openai
errors = [
openai.error.Timeout,
openai.Timeout,
openai.error.APIError,
openai.error.APIConnectionError,
openai.error.RateLimitError,
@ -594,7 +594,8 @@ class BaseOpenAI(BaseLLM):
if self.openai_proxy:
openai.proxy = {"http": self.openai_proxy, "https": self.openai_proxy} # type: ignore[assignment] # noqa: E501
# TODO: The 'openai.proxy' option isn't read in the client API. You will need to pass it when you instantiate the client, e.g. 'OpenAI(proxy={"http": self.openai_proxy, "https": self.openai_proxy})'
# openai.proxy = {"http": self.openai_proxy, "https": self.openai_proxy} # type: ignore[assignment] # noqa: E501
return {**openai_creds, **self._default_params}
@property