Add update_conversation_message

AGiXT · Feb 1, 2024 · a20c034 · a20c034
1 parent 1ef1ad6
commit a20c034
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/agixtsdk/__init__.py b/agixtsdk/__init__.py
@@ -236,6 +236,24 @@ def delete_conversation_message(
         except Exception as e:
             return self.handle_error(e)
 
+    def update_conversation_message(
+        self, agent_name: str, conversation_name: str, message: str, new_message: str
+    ) -> str:
+        try:
+            response = requests.put(
+                headers=self.headers,
+                url=f"{self.base_uri}/api/conversation/message",
+                json={
+                    "message": message,
+                    "new_message": new_message,
+                    "agent_name": agent_name,
+                    "conversation_name": conversation_name,
+                },
+            )
+            return response.json()["message"]
+        except Exception as e:
+            return self.handle_error(e)
+
     def prompt_agent(
         self,
         agent_name: str,

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