Commit 6572ef4f

ZheNing Hu <adlternative@gmail.com>
2023-04-08 09:09:55
Add proxy argument support for openai client (#313)
Co-authored-by: huzhening.hzn <huzhening.hzn@alibaba-inc.com>
1 parent bd269ab
Changed files (2)
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 main():
         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)
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