feat: Added openai api key setting remote API

Upsonic · Aug 6, 2024 · 94b5e67 · 94b5e67
1 parent 9346dbd
commit 94b5e67
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/gpt_computer_assistant/api.py b/gpt_computer_assistant/api.py
@@ -367,6 +367,16 @@ def expand():
     return jsonify({"response": "Expanded."})
 
 
+@app.route("/save_openai_api_key", methods=["POST"])
+def save_openai_api_key():
+    """
+    This api saves the OpenAI API key
+    """
+    data = request.json
+    openai_api_key = data["openai_api_key"]
+    from .utils.db import save_api_key
+    save_api_key(openai_api_key)
+    return jsonify({"response": "OpenAI API key saved."})
 
 class ServerThread(threading.Thread):
     def __init__(self, app, host, port):

diff --git a/gpt_computer_assistant/remote.py b/gpt_computer_assistant/remote.py
@@ -194,6 +194,11 @@ def expand(self):
         return response["response"]
 
 
+    def save_openai_api_key(self, openai_api_key):
+        data = {"openai_api_key": openai_api_key}
+        response = self.send_request("/save_openai_api_key", data)
+        return response["response"]
+
 
     def wait(self, second):
         time.sleep(second)