From dde846a328930239a1f7f946e16c9fdc5c6cd54a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=20D=C3=B6rr?= <tom.doerr@tum.de>
Date: Tue, 19 Oct 2021 03:08:32 +0200
Subject: [PATCH] Add openairc

---
 openai/__init__.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/openai/__init__.py b/openai/__init__.py
index 054cec3d4d..d064d29784 100644
--- a/openai/__init__.py
+++ b/openai/__init__.py
@@ -1,4 +1,5 @@
 import os
+import configparser
 
 # OpenAI Python bindings.
 #
@@ -8,6 +9,21 @@
 
 api_key = os.environ.get("OPENAI_API_KEY")
 organization = os.environ.get("OPENAI_ORGANIZATION")
+
+home = os.path.expanduser("~")
+config_home = os.environ.get("XDG_CONFIG_HOME", os.path.join(home, ".config"))
+config_parser = configparser.RawConfigParser()
+config_location = os.path.join(config_home, "openairc")
+if os.path.exists(config_location):
+    config_parser.read(config_location)
+
+    if not api_key:
+        api_key = config_parser.get("api", "APIKey")
+
+    if not organization:
+        organization = config_parser.get("api", "Organization")
+
+
 client_id = None
 api_base = os.environ.get("OPENAI_API_BASE", "https://api.openai.com")
 file_api_base = None