allow configuration of OPENAI_API_BASE, and use gpt-4o-mini by default; fix command line example

deep_translator/chatgpt.py

@@ -4,7 +4,11 @@ import os
 from typing import List, Optional
 
 from deep_translator.base import BaseTranslator
-from deep_translator.constants import OPEN_AI_ENV_VAR
+from deep_translator.constants import (
+    OPEN_AI_ENV_VAR,
+    OPEN_AI_BASE_URL_ENV_VAR,
+    OPEN_AI_MODEL_ENV_VAR,
+)
 from deep_translator.exceptions import ApiKeyException
 
 
@@ -19,19 +23,23 @@ class ChatGptTranslator(BaseTranslator):
         source: str = "auto",
         target: str = "english",
         api_key: Optional[str] = os.getenv(OPEN_AI_ENV_VAR, None),
-        model: Optional[str] = "gpt-3.5-turbo",
+        model: Optional[str] = os.getenv(OPEN_AI_MODEL_ENV_VAR, "gpt-4o-mini"),
+        base_url: Optional[str] = os.getenv(OPEN_AI_BASE_URL_ENV_VAR, None),
         **kwargs,
     ):
         """
         @param api_key: your openai api key.
         @param source: source language
         @param target: target language
+        @param model: OpenAI model to use
+        @param base_url: custom OpenAI API base URL
         """
         if not api_key:
             raise ApiKeyException(env_var=OPEN_AI_ENV_VAR)
 
         self.api_key = api_key
         self.model = model
+        self.base_url = base_url
 
         super().__init__(source=source, target=target, **kwargs)
 
@@ -42,12 +50,15 @@ class ChatGptTranslator(BaseTranslator):
         """
         import openai
 
-        openai.api_key = self.api_key
+        client = openai.OpenAI(
+            api_key=self.api_key,
+            base_url=self.base_url if self.base_url else None
+        )
 
         prompt = f"Translate the text below into {self.target}.\n"
         prompt += f'Text: "{text}"'
 
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model=self.model,
             messages=[
                 {

deep_translator/constants.py

@@ -1,7 +1,9 @@
 __copyright__ = "Copyright (C) 2020 Nidhal Baccouri"
 
 
-OPEN_AI_ENV_VAR = "OPEN_API_KEY"
+OPEN_AI_ENV_VAR = "OPENAI_API_KEY"
+OPEN_AI_BASE_URL_ENV_VAR = "OPENAI_API_BASE"
+OPEN_AI_MODEL_ENV_VAR = "OPENAI_MODEL"
 DEEPL_ENV_VAR = "DEEPL_API_KEY"
 LIBRE_ENV_VAR = "LIBRE_API_KEY"
 MSFT_ENV_VAR = "MICROSOFT_API_KEY"

docs/README.rst

@@ -592,8 +592,11 @@ ChatGpt Translator
     There are two required attributes, namely "api_key" (string) and "target" (string or list).
     Attribute "source" is optional.
 
-    You can provide your api key as an argument or you can export it as an env var
-    e.g. `export OPENAI_API_KEY="your_key"`
+    You can provide your api key, api base as an argument or you can export it as an env var
+    e.g. 
+    `export OPENAI_API_KEY="your_key"
+     export OPENAI_API_BASE=https://api.openai.com/v1
+    `
 
 .. code-block:: python
 
@@ -824,13 +827,15 @@ To translate a string or line of text:
 
 .. code-block:: console
 
-    $ deep_translator google --source "english" --target "german" --text "happy coding"
+    $ deep_translator --translator google --source "english" --target "german" --text "happy coding"
+    $ 
 
 Alternate short option names, along with using language abbreviations:
 
 .. code-block:: console
 
-    $ deep_translator google -src "en" -tgt "de" -txt "happy coding"
+    $ dt -trans google -src "en" -tg "de" -txt "happy coding"
+    $ dt -trans chatgpt -src "en" -tg "ja" -txt "happy coding"
 
 
 Finally, to retrieve a list of available languages for a given translator: