githubmirror/AgentPilot

agentpilot/utils/api.py

17 строк573 B

jbexta

oai v1

14 дек 2023, 03:51

14 дек 2023, 03:518aca68d

import os
from agentpilot.utils import sql

apis = {}
llm_apis = {}


def load_api_keys():
    global apis
    api_table = sql.get_results("SELECT `name`, `client_key`, `priv_key` FROM apis")
    apis = {api[0].lower(): {'client_key': api[1], 'priv_key': api[2]} for api in api_table}

    if apis['openai']['priv_key'] == '$OPENAI_API_KEY':
        apis['openai']['priv_key'] = os.environ.get("OPENAI_API_KEY", '')

    if apis['perplexity ai']['priv_key'] == '$PERPLEXITYAI_API_KEY':
        apis['perplexity ai']['priv_key'] = os.environ.get("PERPLEXITYAI_API_KEY", '')