add openai proxy option

sakerhetspolis · May 30, 2023 · ecd8ca3 · ecd8ca3
1 parent d8e9e1e
commit ecd8ca3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/gepetto/config.ini b/gepetto/config.ini
@@ -2,8 +2,12 @@
 MODEL = gpt-3.5-turbo
 
 # Specify the program language. It can be "fr_FR", "zh_CN", or any folder in locales. Defaults to English.
-LANGUAGE =
+LANGUAGE = 
 
 [OpenAI]
 # Set your API key here, or put it in the OPENAI_API_KEY environment variable.
-API_KEY =
+API_KEY = 
+
+# Set your OpenAI Proxy here, or put it in the HTTPS_PROXY environment variable.
+# Such as: OPENAI_PROXY = http://127.0.0.1:7890
+OPENAI_PROXY = 
diff --git a/gepetto/config.py b/gepetto/config.py
@@ -33,6 +33,13 @@ def load_config():
         openai.api_key = config.get('OpenAI', 'API_KEY')
         print(f"Key set to {openai.api_key}")
 
+    # Get OPENAPI PROXY
+    if not config.get('OpenAI', 'OPENAI_PROXY'):
+        openai.proxy = None
+    else:
+        openai.proxy = config.get('OpenAI', 'OPENAI_PROXY')
+        print(f"OpenAI Proxy set to {openai.proxy}")
+
     # Select model
     requested_model = config.get('Gepetto', 'MODEL')
     model = get_model(requested_model)