python-sdk/workflowai/env.py at bf8a84d77911b7331dbacb16d7c68508234ac56a · WorkflowAI/python-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
"""A file to describe all environment variables"""

import os

WORKFLOWAI_DEFAULT_MODEL = os.getenv("WORKFLOWAI_DEFAULT_MODEL", "gemini-1.5-pro-latest")

WORKFLOWAI_API_URL = os.getenv("WORKFLOWAI_API_URL", "https://run.workflowai.com")


def _default_app_url():
    if not WORKFLOWAI_API_URL:
        return "https://workflowai.com"
    if WORKFLOWAI_API_URL.startswith("https://run."):
        return "https://" + WORKFLOWAI_API_URL.removeprefix("https://run.")
    if WORKFLOWAI_API_URL.startswith("https://api."):
        return "https://" + WORKFLOWAI_API_URL.removeprefix("https://api.")
    return "https://workflowai.com"


WORKFLOWAI_APP_URL = os.getenv("WORKFLOWAI_APP_URL", _default_app_url())

WORKFLOWAI_API_KEY = os.getenv("WORKFLOWAI_API_KEY", "")