Add the anthropic interface

.env.sample

@@ -0,0 +1,3 @@
+ANTHROPIC_API_KEY=""
+GROQ_API_KEY=""
+OPENAI_API_KEY=""

.pre-commit-config.yaml

@@ -8,4 +8,4 @@ repos:
         # supported by your project here, or alternatively use
         # pre-commit's default_language_version, see
         # https://pre-commit.com/#top_level-default_language_version
-        language_version: python3.11
+        language_version: python3.12

aimodels/client/completions.py

@@ -31,7 +31,7 @@ def create(self, model=None, temperature=0, messages=None):
         interface, model_name = self.topmost_instance.get_provider_interface(model)
 
         return interface.chat_completion_create(
-            messages,
+            messages=messages,
             model=model_name,
             temperature=temperature,
         )

aimodels/client/multi_fm_client.py

@@ -1,7 +1,6 @@
 """MultiFMClient manages a Chat across multiple provider interfaces."""
 
-from ..providers.openai_interface import OpenAIInterface
-from ..providers.groq_interface import GroqInterface
+from ..providers import AnthropicInterface, OpenAIInterface, GroqInterface
 from .chat import Chat
 
 
@@ -30,6 +29,7 @@ def __init__(self):
         self.all_factories = {
             "openai": OpenAIInterface,
             "groq": GroqInterface,
+            "anthropic": AnthropicInterface,
         }
 
     def get_provider_interface(self, model):

aimodels/framework/__init__.py

@@ -1,3 +1,4 @@
 """Provides the ProviderInterface for defining the interface that all FM providers must implement."""
 
 from .provider_interface import ProviderInterface
+from .chat_completion_response import ChatCompletionResponse

aimodels/framework/chat_completion_response.py

@@ -0,0 +1,8 @@
+from aimodels.framework.choice import Choice
+
+
+class ChatCompletionResponse:
+    """Used to conform to the response model of OpenAI"""
+
+    def __init__(self):
+        self.choices = [Choice()]  # Adjust the range as needed for more choices

aimodels/framework/choice.py

@@ -0,0 +1,6 @@
+from aimodels.framework.message import Message
+
+
+class Choice:
+    def __init__(self):
+        self.message = Message()

aimodels/framework/message.py

@@ -0,0 +1,6 @@
+"""Interface to hold contents of api responses when they do not conform to the OpenAI style response"""
+
+
+class Message:
+    def __init__(self):
+        self.content = None

aimodels/providers/__init__.py

@@ -1,3 +1,5 @@
 """Provides the individual provider interfaces for each FM provider."""
 
 from .openai_interface import OpenAIInterface
+from .groq_interface import GroqInterface
+from .anthropic_interface import AnthropicInterface

aimodels/providers/anthropic_interface.py

@@ -0,0 +1,67 @@
+"""The interface to the Anthropic API."""
+
+import os
+
+from aimodels.framework import ProviderInterface, ChatCompletionResponse
+
+
+class AnthropicInterface(ProviderInterface):
+    """Implements the ProviderInterface for interacting with the Anthropic API."""
+
+    def __init__(self):
+        """Set up the Anthropic API client with the key from the user's environment"""
+        import anthropic
+
+        self.anthropic_client = anthropic.Anthropic(
+            api_key=os.getenv("ANTHROPIC_API_KEY")
+        )
+
+    def chat_completion_create(self, messages=None, model=None, temperature=0):
+        """Request chat completions from the Anthropic API.
+
+        Args:
+        ----
+            model (str): Identifies the specific provider/model to use.
+            messages (list of dict): A list of message objects in chat history.
+            temperature (float): The temperature to use in the completion.
+
+        Returns:
+        -------
+            The ChatCompletionResponse with the completion result.
+
+        """
+        anthropic_messages = []
+        system_message = None
+        for msg in messages:
+            if "role" in msg and "content" in msg:
+                if msg["role"] == "system":
+                    system_message = msg["content"]
+                else:
+                    temp_msg = msg.copy()
+                    temp_msg["content"] = [
+                        {"type": "text", "text": temp_msg["content"]}
+                    ]
+                    anthropic_messages.append(temp_msg)
+
+        if system_message is None:
+            response = self.anthropic_client.messages.create(
+                messages=anthropic_messages,
+                model=model,
+                max_tokens=4096,
+                temperature=temperature,
+            )
+        else:
+            response = self.anthropic_client.messages.create(
+                messages=anthropic_messages,
+                model=model,
+                system=system_message,
+                max_tokens=4096,
+                temperature=temperature,
+            )
+
+        text_response = response.content[0].text
+
+        chat_completion_response = ChatCompletionResponse()
+        chat_completion_response.choices[0].message.content = text_response
+
+        return chat_completion_response

examples/test_anthropic.ipynb

@@ -0,0 +1,92 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "id": "initial_id",
+   "metadata": {
+    "collapsed": true,
+    "ExecuteTime": {
+     "end_time": "2024-07-02T23:20:19.015491Z",
+     "start_time": "2024-07-02T23:20:19.004272Z"
+    }
+   },
+   "source": [
+    "import sys\n",
+    "sys.path.append('../aimodels')\n",
+    "\n",
+    "from dotenv import load_dotenv, find_dotenv\n",
+    "\n",
+    "load_dotenv(find_dotenv())"
+   ],
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "True"
+      ]
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 1
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2024-07-03T02:22:26.282827Z",
+     "start_time": "2024-07-03T02:22:18.193996Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "from aimodels.client import MultiFMClient\n",
+    "\n",
+    "client = MultiFMClient()\n",
+    "model_string = \"anthropic:claude-3-opus-20240229\"\n",
+    "messages=[{\"role\": \"system\", \"content\": \"Respond in Pirate English.\"}, \n",
+    "          {\"role\": \"user\", \"content\": \"Tell me a joke\"} ] \n",
+    "\n",
+    "response = client.chat.completions.create(model=model_string, messages=messages)\n",
+    "print(response.choices[0].message.content)"
+   ],
+   "id": "adebd2f0b578a909",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Arrr, me bucko, 'ere be a jolly jest fer ye!\n",
+      "\n",
+      "What did th' pirate say on 'is 80th birthday? \"Aye matey!\"\n",
+      "\n",
+      "Ye see, it be a play on words, as \"Aye matey\" sounds like \"I'm eighty\". Har har har! 'Tis a clever bit o' pirate humor, if I do say so meself. Now, 'ow about ye fetch me a mug o' grog while I spin ye another yarn?\n"
+     ]
+    }
+   ],
+   "execution_count": 6
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 2
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython2",
+   "version": "2.7.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}