allow setting extra headers for API calls via environment variables

Author: dasmy

.env.azure-example

@@ -2,6 +2,7 @@ OPENAI_API_KEY=XXXX
 OPENAI_API_TYPE=azure
 OPENAI_BASE_URL=https://your-resource-name.openai.azure.com
 OPENAI_API_VERSION=2023-03-15-preview
+# OPENAI_EXTRA_HEADERS={"key": "value"}
 AZURE_OPENAI_DEPLOYMENT=your-model-deployment-name
 API_PORT=5010
 WEB_PORT=8080

.env.example

@@ -1,5 +1,8 @@
 OPENAI_API_KEY=sk-XXXX
+OPENAI_API_TYPE=openai
 OPENAI_BASE_URL=https://api.openai.com
+OPENAI_API_VERSION=2023-03-15-preview
+# OPENAI_EXTRA_HEADERS={"key": "value"}
 API_PORT=5010
 WEB_PORT=8080
 SNAKEMQ_PORT=8765

gpt_code_ui/webapp/main.py

@@ -25,6 +25,7 @@
 openai.api_key = os.environ.get("OPENAI_API_KEY", "")
 openai.log = os.getenv("OPENAI_API_LOGLEVEL", "")
 AZURE_OPENAI_DEPLOYMENT = os.environ.get("AZURE_OPENAI_DEPLOYMENT", "")
+OPENAI_EXTRA_HEADERS = json.loads(os.environ.get("OPENAI_EXTRA_HEADERS", "{}"))
 
 UPLOAD_FOLDER = 'workspace/'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@@ -121,7 +122,7 @@ async def get_code(user_prompt, user_openai_key=None, model="gpt-3.5-turbo"):
 
     arguments = dict(
         temperature=0.7,
-        headers={'x-api-key': openai.api_key},
+        headers=OPENAI_EXTRA_HEADERS,
         messages=[
             # {"role": "system", "content": system},
             {"role": "user", "content": prompt},