add persona functions

agixtsdk/__init__.py

@@ -973,6 +973,31 @@ def rename_prompt(
         except Exception as e:
             return self.handle_error(e)
 
+    def get_persona(self, agent_name: str) -> Dict[str, Any]:
+        try:
+            response = requests.get(
+                headers=self.headers,
+                url=f"{self.base_uri}/api/agent/{agent_name}/persona",
+            )
+            if self.verbose:
+                parse_response(response)
+            return response.json()["persona"]
+        except Exception as e:
+            return self.handle_error(e)
+
+    def update_persona(self, agent_name: str, persona: str) -> str:
+        try:
+            response = requests.put(
+                f"{self.base_uri}/api/agent/{agent_name}/persona",
+                headers=self.headers,
+                json={"persona": persona},
+            )
+            if self.verbose:
+                parse_response(response)
+            return response.json()["message"]
+        except Exception as e:
+            return self.handle_error(e)
+
     def get_extension_settings(self) -> Dict[str, Any]:
         try:
             response = requests.get(

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name="agixtsdk",
-    version="0.0.71",
+    version="0.0.72",
     description="The AGiXT SDK for Python.",
     long_description=long_description,
     long_description_content_type="text/markdown",