fix config.get_provider_for for provider inherit issue

src/ell/configurator.py

@@ -16,9 +16,9 @@ class _Model:
     name: str
     default_client: Optional[Union[openai.Client, Any]] = None
     #XXX: Deprecation in 0.1.0
-    #XXX: We will depreciate this when streaming is implemented. 
+    #XXX: We will depreciate this when streaming is implemented.
     # Currently we stream by default for the verbose renderer,
-    # but in the future we will not support streaming by default 
+    # but in the future we will not support streaming by default
     # and stream=True must be passed which will then make API providers the
     # single source of truth for whether or not a model supports an api parameter.
     # This makes our implementation extremely light, only requiring us to provide
@@ -44,9 +44,9 @@ def __init__(self, **data):
         self._lock = threading.Lock()
         self._local = threading.local()
 
-    
+
     def register_model(
-        self, 
+        self,
         name: str,
         default_client: Optional[Union[openai.Client, Any]] = None,
         supports_streaming: Optional[bool] = None
@@ -74,12 +74,12 @@ def model_registry_override(self, overrides: Dict[str, _Model]):
         """
         if not hasattr(self._local, 'stack'):
             self._local.stack = []
-        
+
         with self._lock:
             current_registry = self._local.stack[-1] if self._local.stack else self.registry
             new_registry = current_registry.copy()
             new_registry.update(overrides)
-        
+
         self._local.stack.append(new_registry)
         try:
             yield
@@ -133,11 +133,25 @@ def get_provider_for(self, client: Union[Type[Any], Any]) -> Optional[Provider]:
         """
 
         client_type = type(client) if not isinstance(client, type) else client
-        for provider_type, provider in self.providers.items():
-            if issubclass(client_type, provider_type) or client_type == provider_type:
-                return provider
+        # First, try to find an exact match
+        if client_type in self.providers:
+            return self.providers[client_type]
+
+        # If no exact match, look for the most specific subclass
+        matching_providers = [
+            (provider_type, provider)
+            for provider_type, provider in self.providers.items()
+            if issubclass(client_type, provider_type)
+        ]
+
+        if matching_providers:
+            # Sort by inheritance depth (most derived class first)
+            matching_providers.sort(key=lambda x: len(x[0].mro()), reverse=True)
+            return matching_providers[0][1]
+
         return None
 
+
 # Single* instance
 # XXX: Make a singleton
 config = Config()
@@ -187,7 +201,7 @@ def init(
 def get_store() -> Union[Store, None]:
     return config.store
 
-# Will be deprecated at 0.1.0 
+# Will be deprecated at 0.1.0
 
 # You can add more helper functions here if needed
 def register_provider(provider: Provider, client_type: Type[Any]) -> None:

src/ell/models/openai.py

@@ -30,6 +30,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 def register(client: openai.Client):
     """
     Register OpenAI models with the provided client.
@@ -92,4 +93,4 @@ def register(client: openai.Client):
     pass
 
 register(default_client)
-config.default_client = default_client
+config.default_client = default_client