Add get_chain_args endpoint

AGiXT · Aug 7, 2023 · ced2304 · ced2304
1 parent b14eaba
commit ced2304
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/agixtsdk/__init__.py b/agixtsdk/__init__.py
@@ -328,6 +328,16 @@ def get_chain_responses(self, chain_name: str) -> Dict[str, Any]:
         except Exception as e:
             return self.handle_error(e)
 
+    def get_chain_args(self, chain_name: str) -> List[str]:
+        try:
+            response = requests.get(
+                headers=self.headers,
+                url=f"{self.base_uri}/api/chain/{chain_name}/args",
+            )
+            return response.json()["chain_args"]
+        except Exception as e:
+            return self.handle_error(e)
+
     def run_chain(
         self,
         chain_name: str,

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