Re-added http based Fireworks & Together; Minor fixes. (#38)

Support for Fireworks provider using http.
Fixed few things based on the doc.
Support for Together using http requests.
Removing interface.py files
---------

Co-authored-by: Kevin Solorio <[email protected]>

aisuite/provider.py

@@ -21,12 +21,14 @@ class ProviderNames(str, Enum):
     ANTHROPIC = "anthropic"
     AWS = "aws"
     AZURE = "azure"
+    FIREWORKS = "fireworks"
     GROQ = "groq"
     GOOGLE = "google"
     HUGGINGFACE = "huggingface"
     MISTRAL = "mistral"
     OLLAMA = "ollama"
     OPENAI = "openai"
+    TOGETHER = "together"
 
 
 class ProviderFactory:
@@ -54,6 +56,14 @@ class ProviderFactory:
         ),
         ProviderNames.OLLAMA: ("aisuite.providers.ollama_provider", "OllamaProvider"),
         ProviderNames.OPENAI: ("aisuite.providers.openai_provider", "OpenAIProvider"),
+        ProviderNames.FIREWORKS: (
+            "aisuite.providers.fireworks_provider",
+            "FireworksProvider",
+        ),
+        ProviderNames.TOGETHER: (
+            "aisuite.providers.together_provider",
+            "TogetherProvider",
+        ),
     }
 
     @classmethod

aisuite/providers/__init__.py

@@ -1,6 +0,0 @@
-"""Provides the individual provider interfaces for each FM provider."""
-
-from .fireworks_interface import FireworksInterface
-from .octo_interface import OctoInterface
-from .replicate_interface import ReplicateInterface
-from .together_interface import TogetherInterface

aisuite/providers/fireworks_provider.py

@@ -0,0 +1,65 @@
+import os
+import httpx
+from aisuite.provider import Provider, LLMError
+from aisuite.framework import ChatCompletionResponse
+
+
+class FireworksProvider(Provider):
+    """
+    Fireworks AI Provider using httpx for direct API calls.
+    """
+
+    BASE_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
+
+    def __init__(self, **config):
+        """
+        Initialize the Fireworks provider with the given configuration.
+        The API key is fetched from the config or environment variables.
+        """
+        self.api_key = config.get("api_key", os.getenv("FIREWORKS_API_KEY"))
+        if not self.api_key:
+            raise ValueError(
+                "Fireworks API key is missing. Please provide it in the config or set the FIREWORKS_API_KEY environment variable."
+            )
+
+        # Optionally set a custom timeout (default to 30s)
+        self.timeout = config.get("timeout", 30)
+
+    def chat_completions_create(self, model, messages, **kwargs):
+        """
+        Makes a request to the Fireworks AI chat completions endpoint using httpx.
+        """
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+        data = {
+            "model": model,
+            "messages": messages,
+            **kwargs,  # Pass any additional arguments to the API
+        }
+
+        try:
+            # Make the request to Fireworks AI endpoint.
+            response = httpx.post(
+                self.BASE_URL, json=data, headers=headers, timeout=self.timeout
+            )
+            response.raise_for_status()
+        except httpx.HTTPStatusError as http_err:
+            raise LLMError(f"Fireworks AI request failed: {http_err}")
+        except Exception as e:
+            raise LLMError(f"An error occurred: {e}")
+
+        # Return the normalized response
+        return self._normalize_response(response.json())
+
+    def _normalize_response(self, response_data):
+        """
+        Normalize the response to a common format (ChatCompletionResponse).
+        """
+        normalized_response = ChatCompletionResponse()
+        normalized_response.choices[0].message.content = response_data["choices"][0][
+            "message"
+        ]["content"]
+        return normalized_response

aisuite/providers/together_provider.py

@@ -0,0 +1,65 @@
+import os
+import httpx
+from aisuite.provider import Provider, LLMError
+from aisuite.framework import ChatCompletionResponse
+
+
+class TogetherProvider(Provider):
+    """
+    Together AI Provider using httpx for direct API calls.
+    """
+
+    BASE_URL = "https://api.together.xyz/v1/chat/completions"
+
+    def __init__(self, **config):
+        """
+        Initialize the Fireworks provider with the given configuration.
+        The API key is fetched from the config or environment variables.
+        """
+        self.api_key = config.get("api_key", os.getenv("TOGETHER_API_KEY"))
+        if not self.api_key:
+            raise ValueError(
+                "Together API key is missing. Please provide it in the config or set the TOGETHER_API_KEY environment variable."
+            )
+
+        # Optionally set a custom timeout (default to 30s)
+        self.timeout = config.get("timeout", 30)
+
+    def chat_completions_create(self, model, messages, **kwargs):
+        """
+        Makes a request to the Fireworks AI chat completions endpoint using httpx.
+        """
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+        data = {
+            "model": model,
+            "messages": messages,
+            **kwargs,  # Pass any additional arguments to the API
+        }
+
+        try:
+            # Make the request to Fireworks AI endpoint.
+            response = httpx.post(
+                self.BASE_URL, json=data, headers=headers, timeout=self.timeout
+            )
+            response.raise_for_status()
+        except httpx.HTTPStatusError as http_err:
+            raise LLMError(f"Together AI request failed: {http_err}")
+        except Exception as e:
+            raise LLMError(f"An error occurred: {e}")
+
+        # Return the normalized response
+        return self._normalize_response(response.json())
+
+    def _normalize_response(self, response_data):
+        """
+        Normalize the response to a common format (ChatCompletionResponse).
+        """
+        normalized_response = ChatCompletionResponse()
+        normalized_response.choices[0].message.content = response_data["choices"][0][
+            "message"
+        ]["content"]
+        return normalized_response