add support for openai base url

Author: npiv

README.md

@@ -235,6 +235,7 @@ positional arguments:
 options:
   -h, --help                       show this help message and exit
   --openai-api-key key             the OpenAI API key can also be set as env variable OPENAI_API_KEY
+  --openai-base-url key            A custom url to use the openAI against a local or custom model, eg ollama
   --temperature t                  temperature (openai setting)
   -c CHAT_GPT, --chat-gpt CHAT_GPT
                                    chat GPT model use either the fully qualified model name, or 3.5 (gpt-3.5-turbo-0613), 4 (gpt-4), 4t (gpt-4-1106-preview). Can also be set via env variable OPENAI_API_MODEL

chatblade/chat.py

@@ -112,7 +112,7 @@ def build_client(config):
             azure_deployment=os.environ.get("OPENAI_API_AZURE_ENGINE"),
         )
     else:
-        return openai.OpenAI(api_key=config["openai_api_key"])
+        return openai.OpenAI(api_key=config["openai_api_key"], base_url=config["openai_base_url"])
 
 
 def query_chat_gpt(messages, config):

chatblade/parser.py

@@ -110,6 +110,12 @@ def parse(args):
         type=str,
         help="the OpenAI API key can also be set as env variable OPENAI_API_KEY",
     )
+    parser.add_argument(
+        "--openai-base-url",
+        metavar="key",
+        type=str,
+        help="A custom url to use the openAI against a local or custom model, eg ollama",
+    )
     parser.add_argument(
         "--temperature",
         metavar="t",