From 4037872e7e4fb643415a04e1afb5aec93547541b Mon Sep 17 00:00:00 2001
From: Russell Ballestrini <russell.ballestrini@gmail.com>
Date: Fri, 15 Mar 2024 10:12:16 -0400
Subject: [PATCH] anthropic.claude-3-haiku-20240307-v1:0 	modified:  
 README.rst 	modified:   app.py

---
 README.rst | 1 +
 app.py     | 9 ++++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/README.rst b/README.rst
index c93d80d..8b8ac14 100644
--- a/README.rst
+++ b/README.rst
@@ -94,6 +94,7 @@ To interact with the various language models, you can use the following commands
 
 - For GPT-3, send a message with ``gpt-3`` and include your prompt.
 - For GPT-4, send a message with ``gpt-4`` and include your prompt.
+- For Claude-haiku, send a message with ``claude-haiku`` and include your prompt.
 - For Claude-sonnet, send a message with ``claude-sonnet`` and include your prompt.
 - For Mistral-tiny, send a message with ``mistral-tiny`` and include your prompt.
 - For Mistral-small, send a message with ``mistral-small`` and include your prompt.
diff --git a/app.py b/app.py
index b9b9d3c..0710bcd 100644
--- a/app.py
+++ b/app.py
@@ -27,7 +27,7 @@ socketio = SocketIO(app, async_mode="eventlet")
 cancellation_requests = {}
 
 system_users = [
-    "anthropic.claude-v2",
+    "anthropic.claude-3-haiku-20240307-v1:0",
     "anthropic.claude-3-sonnet-20240229-v1:0",
     "gpt-3.5-turbo",
     "gpt-4",
@@ -262,6 +262,13 @@ def handle_message(data):
             room=room.name,
         )
 
+        if "claude-haiku" in data["message"]:
+            eventlet.spawn(
+                chat_claude,
+                data["username"],
+                room.name,
+                model_name="anthropic.claude-3-haiku-20240307-v1:0",
+            )
         if "claude-sonnet" in data["message"]:
             eventlet.spawn(chat_claude, data["username"], room.name)
         if "gpt-3" in data["message"]: