Patch-1

ai.py

@@ -8,6 +8,8 @@
 from EdgeGPT.EdgeGPT import Chatbot as Bing
 from EdgeGPT.EdgeGPT import ConversationStyle
 from hugchat import hugchat
+from claude import claude_client
+from claude import claude_wrapper
 
 MODELS = ["bard", "bing", "gpt", "hugchat", "llama"]
 
@@ -121,6 +123,18 @@ async def send(self, text):
         else:
             return response
 
+# to use the claude api library https://github.com/AshwinPathi/claude-api-py
+class ClaudeAPI:
+    def __init__(self, session_key):
+        self.client = claude_client.ClaudeClient(session_key)
+        organizations = self.client.get_organizations()
+        self.claude_obj = claude_wrapper.ClaudeWrapper(self.client, organization_uuid=organizations[0]['uuid'])
+        new_conversation_data = self.claude_obj.start_new_conversation("New Conversation", 'Hi how are you?')
+        self.conversation_uuid = new_conversation_data
+
+    async def send(self, text):
+        response = self.claude_obj.send_message(text, conversation_uuid=self.conversation_uuid)
+        return response
 
 class HugchatAPI:
     def __init__(self, cookie_path):