From 3aa6f0a42b71d70725f724f6518d7659dad61c8b Mon Sep 17 00:00:00 2001
From: adhikjoshi <adhik3035@gmail.com>
Date: Sun, 1 Mar 2026 15:50:42 +0530
Subject: [PATCH] feat: add ModelsLab chat provider

Adds ModelsLab as an OpenAI-compatible chat provider using the
uncensored-chat endpoint at https://modelslab.com/api/uncensored-chat/v1.

Usage:
    export MODELSLAB_API_KEY=your_key

    import aisuite as ai
    client = ai.Client()
    response = client.chat.completions.create(
        model='modelslab:meta-llama/Meta-Llama-3-8B-Instruct',
        messages=[{'role': 'user', 'content': 'Hello!'}],
    )

API docs: https://docs.modelslab.com
---
 aisuite/providers/modelslab_provider.py | 59 +++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 aisuite/providers/modelslab_provider.py

diff --git a/aisuite/providers/modelslab_provider.py b/aisuite/providers/modelslab_provider.py
new file mode 100644
index 00000000..de5c0dec
--- /dev/null
+++ b/aisuite/providers/modelslab_provider.py
@@ -0,0 +1,59 @@
+"""ModelsLab provider for aisuite.
+
+ModelsLab provides uncensored chat via an OpenAI-compatible endpoint.
+API docs: https://docs.modelslab.com
+API key: https://modelslab.com/account/api-key
+"""
+
+import os
+
+import openai
+
+from aisuite.provider import LLMError, Provider
+
+
+class ModelsLabProvider(Provider):
+    """Provider for ModelsLab's OpenAI-compatible chat API."""
+
+    BASE_URL = "https://modelslab.com/api/uncensored-chat/v1"
+
+    def __init__(self, **config):
+        """Initialize the ModelsLab provider.
+
+        Args:
+            **config: Configuration kwargs. Accepts `api_key` and any other
+                kwargs supported by `openai.OpenAI`. If `api_key` is not
+                provided, reads from the ``MODELSLAB_API_KEY`` environment
+                variable.
+        """
+        config.setdefault("api_key", os.getenv("MODELSLAB_API_KEY"))
+        if not config["api_key"]:
+            raise ValueError(
+                "ModelsLab API key is missing. Please provide it in the config "
+                "or set the MODELSLAB_API_KEY environment variable."
+            )
+        config["base_url"] = self.BASE_URL
+        self.client = openai.OpenAI(**config)
+
+    def chat_completions_create(self, model, messages, **kwargs):
+        """Create a chat completion using ModelsLab's API.
+
+        Args:
+            model: The model ID (e.g. "meta-llama/Meta-Llama-3-8B-Instruct").
+            messages: List of message dicts in OpenAI format.
+            **kwargs: Additional kwargs forwarded to the API.
+
+        Returns:
+            The API response object.
+
+        Raises:
+            LLMError: If the API call fails.
+        """
+        try:
+            return self.client.chat.completions.create(
+                model=model,
+                messages=messages,
+                **kwargs,
+            )
+        except Exception as exc:
+            raise LLMError(f"ModelsLab API error: {exc}") from exc