Add proxy argument support for openai client (#313)

Co-authored-by: huzhening.hzn <[email protected]>

Author: adlternative

README.md

@@ -138,6 +138,9 @@ openai api chat_completions.create -m gpt-3.5-turbo -g user "Hello world"
 
 # generate images via DALL·E API
 openai api image.create -p "two dogs playing chess, cartoon" -n 1
+
+# using openai through a proxy
+openai --proxy=http://proxy.com api models.list
 ```
 
 ## Example code

openai/_openai_scripts.py

@@ -32,6 +32,7 @@ def main():
     )
     parser.add_argument("-b", "--api-base", help="What API base url to use.")
     parser.add_argument("-k", "--api-key", help="What API key to use.")
+    parser.add_argument("-p", "--proxy", nargs='+', help="What proxy to use.")
     parser.add_argument(
         "-o",
         "--organization",
@@ -65,6 +66,13 @@ def help(args):
         openai.api_base = args.api_base
     if args.organization is not None:
         openai.organization = args.organization
+    if args.proxy is not None:
+        openai.proxy = {}
+        for proxy in args.proxy:
+            if proxy.startswith('https'):
+                openai.proxy['https'] = proxy
+            elif proxy.startswith('http'):
+                openai.proxy['http'] = proxy
 
     try:
         args.func(args)