54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from matrix_gpt.config import global_config
|
||
|
|
from matrix_gpt.generate_clients.anthropic import AnthropicApiClient
|
||
|
|
from matrix_gpt.generate_clients.openai import OpenAIClient
|
||
|
|
|
||
|
|
"""
|
||
|
|
Global variable to sync importing and sharing the configured module.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
class ApiClientManager:
|
||
|
|
def __init__(self):
|
||
|
|
self._openai_api_key = None
|
||
|
|
self._openai_api_base = None
|
||
|
|
self._anth_api_key = None
|
||
|
|
self.logger = logging.getLogger('MatrixGPT').getChild('ApiClientManager')
|
||
|
|
|
||
|
|
def _set_from_config(self):
|
||
|
|
"""
|
||
|
|
Have to update the config because it may not be instantiated yet.
|
||
|
|
"""
|
||
|
|
self._openai_api_key = global_config['openai'].get('api_key', 'MatrixGPT')
|
||
|
|
self._anth_api_key = global_config['anthropic'].get('api_key')
|
||
|
|
|
||
|
|
def get_client(self, mode: str):
|
||
|
|
if mode == 'openai':
|
||
|
|
return self.openai_client()
|
||
|
|
elif mode == 'anth':
|
||
|
|
return self.anth_client()
|
||
|
|
else:
|
||
|
|
raise Exception
|
||
|
|
|
||
|
|
def openai_client(self):
|
||
|
|
self._set_from_config()
|
||
|
|
if not self._openai_api_key:
|
||
|
|
self.logger.error('Missing an OpenAI API key!')
|
||
|
|
return None
|
||
|
|
return OpenAIClient(
|
||
|
|
api_key=self._openai_api_key,
|
||
|
|
)
|
||
|
|
|
||
|
|
def anth_client(self):
|
||
|
|
self._set_from_config()
|
||
|
|
if not self._anth_api_key:
|
||
|
|
self.logger.error('Missing an Anthropic API key!')
|
||
|
|
return None
|
||
|
|
return AnthropicApiClient(
|
||
|
|
api_key=self._anth_api_key,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
api_client_helper = ApiClientManager()
|