githubmirror/AgentPilot

src/utils/zzzapi.py

17 строк595 B

jbexta

models

08 мар 2024, 17:17

08 мар 2024, 17:174b762db

# import os
# from src.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", '')