Added proxy support.

openAI.sublime-settings

@@ -46,11 +46,19 @@
     // Affects only `chat_completion` command.
     "markdown": false,
 
-     // Apply custom syntax highlight to OpenAI completion output panel text.
+    // Apply custom syntax highlight to OpenAI completion output panel text.
     // e.g.: "Packages/MarkdownEditing/syntaxes/MultiMarkdown.sublime-syntax"
     "syntax_path": "",
 
-     // Minimum amount of characters selected to perform completion.
+    // Minimum amount of characters selected to perform completion.
     // Does not affect completion command if the "output_panel" setting is true.
-    "minimum_selection_length": 20
+    "minimum_selection_length": 20,
+
+    // Proxy setting
+    "proxy": {
+        // Proxy address
+        "address": "",
+        // Proxy port
+        "port": 8080
+    }
 }

openai_worker.py

@@ -96,7 +96,13 @@ def exec_net_request(self, connect: http.client.HTTPSConnection):
 
     def chat_complete(self):
         cacher = Cacher()
-        conn = http.client.HTTPSConnection("api.openai.com")
+        proxy = self.settings.get('proxy')['address']
+        port = self.settings.get('proxy')['port']
+        if len(proxy) > 0:
+            conn = http.client.HTTPSConnection(host=proxy, port=port)
+            conn.set_tunnel("api.openai.com")
+        else:
+            conn = http.client.HTTPSConnection("api.openai.com")
 
         payload = {
             # Todo add uniq name for each output panel (e.g. each window)

outputpanel.py

@@ -26,7 +26,7 @@ def rewrite_output_panel(self, window, markdown: bool, syntax_path: str):
 
         output_panel.set_read_only(True)
         num_lines = get_number_of_lines(output_panel)
-        print(num_lines)
+        print(f'num_lines: {num_lines}')
 
         ## FIXME: To calibrate scroll.
         point = output_panel.text_point(num_lines, 0) # +8 is that much from last line of a past answer and the first line of a next one.
@@ -47,7 +47,7 @@ def on_activated(self, view):
                 window=window,
                 markdown=settings.get('markdown'),
                 syntax_path=settings.get('syntax_path')
-                )
+            )
 
     def show_panel(self, window):
         window.run_command("show_panel", {"panel": f"output.{self.OUTPUT_PANEL_NAME}"})