From 2bbaab20d63a62c1e844101397330a4ad709e35c Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Mon, 28 Oct 2024 06:42:01 -0400
Subject: [PATCH] test

---
 .vscode/settings.json             |  6 ++++
 simplemind/integrations/base.py   | 48 ++++++++++++++++++++++---------
 simplemind/integrations/openai.py | 12 ++++++++
 t.py                              |  0
 4 files changed, 53 insertions(+), 13 deletions(-)
 create mode 100644 .vscode/settings.json
 create mode 100644 t.py

diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..fc446d7
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,6 @@
+{
+    "workbench.colorCustomizations": {
+        "sash.hoverBorder": "#d5f4a4"
+    },
+    "peacock.color": "#c0ef76"
+}
diff --git a/simplemind/integrations/base.py b/simplemind/integrations/base.py
index 2b6cf83..569892e 100644
--- a/simplemind/integrations/base.py
+++ b/simplemind/integrations/base.py
@@ -1,32 +1,54 @@
 import os
 import logging
 
+
 class BaseClientProvider:
 
-    def __init__(self, *, api_key=None, api_key_environ_name=None):
+    def __init__(self, *, api_key_environ_key=None, api_key=None):
         self.logger = logging.getLogger(self.__class__.__name__)
-
-        self._api_key_environ_name = api_key_environ_name
+        self.client = None
 
         # Load API key from environment if not provided
-        self._api_key = api_key or self._load_from_environ()
+        self._api_key = api_key or self._load_from_environ(self._api_key_environ_name)
 
-    def _load_from_environ(self):
-        if self._api_key_environ_name:
-            return os.environ.get(self._api_key_environ_name)
-        return None
+    @classmethod
+    def from_environ(cls, environ_key):
+        """Loads the API key from the environment (recommended)."""
+        return cls(api_key=os.environ.get(environ_key))
+
+    def initialize(self):
+        """Initializes the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+        raise NotImplementedError(msg)
 
     def test_connection(self):
-        raise NotImplementedError("This method must be implemented by the AI provider client.")
+        """Tests the connection to the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+        raise NotImplementedError(msg)
 
     def generate_response(self, request):
-        raise NotImplementedError("This method must be implemented by the AI provider client.")
+        """Generates a response from the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+        raise NotImplementedError(msg)
 
     def health_check(self):
-        raise NotImplementedError("This method must be implemented by the AI provider client.")
+        """Checks the health of the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+        raise NotImplementedError(msg)
 
     def available_models(self):
-        raise NotImplementedError("This method must be implemented by the AI provider client.")
+        """Returns the available models from the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+
+        raise NotImplementedError(msg)
 
     def features(self):
-        raise NotImplementedError("This method must be implemented by the AI provider client.")
+        """Returns the features of the AI provider client."""
+
+        msg = "This method must be implemented by the AI provider client."
+        raise NotImplementedError(msg)
diff --git a/simplemind/integrations/openai.py b/simplemind/integrations/openai.py
index 85e0203..7d9a92e 100644
--- a/simplemind/integrations/openai.py
+++ b/simplemind/integrations/openai.py
@@ -4,5 +4,17 @@ from .base import BaseClientProvider
 class OpenAI(BaseClientProvider):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        self.initialize()
+
+    def initialize(self):
+        assert self._api_key, "API key is required for OpenAI client"
+        assert self._api_key.startswith("sk-"), "OpenAI API key must start with 'sk-'"
+
+        self.logger.info("Initializing OpenAI client")
+        self.logger.debug(f"API key: {self._api_key}")
+
+        self.test_connection()
+
+
 
     def test_connection(self):
diff --git a/t.py b/t.py
new file mode 100644
index 0000000..e69de29