githubmirror/AgentPilot

src/system/apis.py

34 строки945 B

jbexta

0.3.0

10 июн 2024, 17:56

10 июн 2024, 17:566e8ca71

import json
import os
from src.utils import sql


class APIManager:
    def __init__(self):
        self.apis = {}

    def load(self):
        self.apis = {}
        apis = sql.get_results("""
            SELECT
                name,
                client_key,
                api_key AS api_key,
                config,
                provider_plugin
            FROM apis""")
        for api_name, client_key, api_key, api_config, provider_plugin in apis:
            if api_key.startswith('$'):
                api_key = os.environ.get(api_key[1:], '')
            if client_key.startswith('$'):
                client_key = os.environ.get(client_key[1:], '')

            self.apis[api_name] = {
                'client_key': client_key,
                'api_key': api_key,
                'provider_plugin': provider_plugin,
                'config': json.loads(api_config)
            }

    def to_dict(self):
        return self.apis