From c69efb9e5f5a4077523759142d33fb6ebf2492cc Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Mon, 28 Oct 2024 06:27:49 -0400 Subject: [PATCH] test --- simplemind/integrations/base.py | 4 +++- simplemind/integrations/claude.py | 4 ++++ simplemind/integrations/openai.py | 5 +++++ 3 files changed, 12 insertions(+), 1 deletion(-) diff --git a/simplemind/integrations/base.py b/simplemind/integrations/base.py index c98f459..0f533f7 100644 --- a/simplemind/integrations/base.py +++ b/simplemind/integrations/base.py @@ -1,6 +1,6 @@ import os -class AIProviderClient: +class BaseClientProvider: def __init__(self, *, api_key=None, environ_name=None): self._environ_name = environ_name @@ -22,6 +22,8 @@ class AIProviderClient: def health_check(self): raise NotImplementedError("This method must be implemented by the AI provider client.") + def available_models(self): + raise NotImplementedError("This method must be implemented by the AI provider client.") # TODO: logging provider. # diff --git a/simplemind/integrations/claude.py b/simplemind/integrations/claude.py index e69de29..8e0c3e2 100644 --- a/simplemind/integrations/claude.py +++ b/simplemind/integrations/claude.py @@ -0,0 +1,4 @@ +from .base import BaseClientProvider + +class OpenAI(BaseClientProvider): + pass diff --git a/simplemind/integrations/openai.py b/simplemind/integrations/openai.py index e69de29..5f6892d 100644 --- a/simplemind/integrations/openai.py +++ b/simplemind/integrations/openai.py @@ -0,0 +1,5 @@ +from .base import BaseClientProvider + + +class OpenAI(BaseClientProvider): + pass