Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

read access code from params

Browse Source
This commit is contained in:
Abi Raja 2023-11-29 11:35:04 -05:00
parent 78a10c9571
commit b46d08b416
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/main.py
View File

@ -79,13 +79,17 @@ async def stream_code(websocket: WebSocket):
# Get the OpenAI API key from the request. Fall back to environment variable if not provided. # Get the OpenAI API key from the request. Fall back to environment variable if not provided.
# If neither is provided, we throw an error. # If neither is provided, we throw an error.
if params["openAiApiKey"]: openai_api_key = None
openai_api_key = params["openAiApiKey"] if "accessCode" in params and params["accessCode"]:
print("Using OpenAI API key from client-side settings dialog") print("Using access code")
else: else:
openai_api_key = os.environ.get("OPENAI_API_KEY") if params["openAiApiKey"]:
if openai_api_key: openai_api_key = params["openAiApiKey"]
print("Using OpenAI API key from environment variable") print("Using OpenAI API key from client-side settings dialog")
else:
openai_api_key = os.environ.get("OPENAI_API_KEY")
if openai_api_key:
print("Using OpenAI API key from environment variable")
if not openai_api_key: if not openai_api_key:
print("OpenAI API key not found") print("OpenAI API key not found")