githubmirror/AgentPilot

openagent/utils/api.py

31 строка765 B

jbexta

initial gui release, still some features not yet implemented

16 окт 2023, 20:19

16 окт 2023, 20:1949da414

import os

import openai
# import openagent.operations.src.

from utils import sql
import litellm

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 'openai' in apis:
        openai.api_key = apis['openai']['priv_key']
        os.environ["OPENAI_API_KEY"] = apis['openai']['priv_key']


# def set_llm_api_keys():
#     for api_name, api_config in apis.items():
#         if api_name == 'openai':
#             litellm.openai_key openai.api_key = api_config['priv_key']
#             litellm.openai_key = api_config['priv_key']


load_api_keys()
# set_llm_api_keys()