Add conversation endpoints

AGiXT · Jul 6, 2023 · fa539bb · fa539bb
1 parent a184772
commit fa539bb
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 11 deletions.
diff --git a/agixtsdk/__init__.py b/agixtsdk/__init__.py
@@ -142,22 +142,23 @@ def get_agentconfig(self, agent_name: str) -> Dict[str, Any]:
         except requests.RequestException:
             return self.handle_error(response)
 
-    def get_chat_history(self, agent_name: str) -> List[Dict[str, Any]]:
+    def get_conversations(self, agent_name: str) -> List[str]:
         try:
             response = requests.get(
-                headers=self.headers, url=f"{self.base_uri}/api/{agent_name}/chat"
+                headers=self.headers,
+                url=f"{self.base_uri}/api/{agent_name}/conversations",
             )
-            return response.json()["chat_history"]
+            return response.json()["conversations"]
         except requests.RequestException:
             return self.handle_error(response)
 
-    def get_history(
+    def get_conversation(
         self, agent_name: str, conversation_name: str, limit: int = 100, page: int = 1
     ) -> List[Dict[str, Any]]:
         try:
             response = requests.get(
                 headers=self.headers,
-                url=f"{self.base_uri}/api/conversation",
+                url=f"{self.base_uri}/api/{agent_name}/conversations",
                 json={
                     "conversation_name": conversation_name,
                     "agent_name": agent_name,
@@ -169,22 +170,46 @@ def get_history(
         except requests.RequestException:
             return self.handle_error(response)
 
-    def delete_agent_history(self, agent_name: str) -> str:
+    def new_conversation(self, agent_name: str, conversation_name: str):
+        try:
+            response = requests.post(
+                headers=self.headers,
+                url=f"{self.base_uri}/api/conversation",
+                json={
+                    "conversation_name": conversation_name,
+                    "agent_name": agent_name,
+                },
+            )
+            return response.json()["conversation_history"]
+        except requests.RequestException:
+            return self.handle_error(response)
+
+    def delete_conversation(self, agent_name: str, conversation_name: str) -> str:
         try:
             response = requests.delete(
                 headers=self.headers,
-                url=f"{self.base_uri}/api/agent/{agent_name}/history",
+                url=f"{self.base_uri}/api/conversation",
+                json={
+                    "conversation_name": conversation_name,
+                    "agent_name": agent_name,
+                },
             )
             return response.json()["message"]
         except requests.RequestException:
             return self.handle_error(response)
 
-    def delete_history_message(self, agent_name: str, message: str) -> str:
+    def delete_conversation_message(
+        self, agent_name: str, conversation_name: str, message: str
+    ) -> str:
         try:
             response = requests.delete(
                 headers=self.headers,
-                url=f"{self.base_uri}/api/agent/{agent_name}/history/message",
-                json={"message": message},
+                url=f"{self.base_uri}/api/conversation/message",
+                json={
+                    "message": message,
+                    "agent_name": agent_name,
+                    "conversation_name": conversation_name,
+                },
             )
             return response.json()["message"]
         except requests.RequestException:

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name="agixtsdk",
-    version="0.0.11",
+    version="0.0.12",
     description="The AGiXT SDK for Python.",
     long_description=long_description,
     long_description_content_type="text/markdown",