persona-sdk-python 0.1.0__tar.gz → 0.1.2__tar.gz

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/.gitignore

@@ -5,3 +5,4 @@ __pycache__/
 dist/
 *.egg-info/
 .eggs/
+.env

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: persona-sdk-python
-Version: 0.1.0
+Version: 0.1.2
 Summary: Python SDK client for Persona AI platform
 Requires-Python: >=3.10
 Requires-Dist: httpx>=0.27.0

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/persona_sdk/__init__.py

@@ -12,8 +12,8 @@ from persona_sdk.api.service_prices import ServicePricesApi
 
 
 class PersonaSdk:
-    def __init__(self, base_url: str, api_key: str, workflows_url: str = None):
-        self.client = PersonaClient(base_url, api_key)
+    def __init__(self, base_url: str, api_key: str, proxy: str = None, workflows_url: str = None):
+        self.client = PersonaClient(base_url, api_key, proxy=proxy)
         self.workflows_client = PersonaClient(workflows_url, api_key) if workflows_url else self.client
         self.agents = AgentsApi(self.client)
         self.workflows = WorkflowsApi(self.workflows_client)

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/persona_sdk/api/projects.py

@@ -25,3 +25,18 @@ class ProjectsApi:
 
     async def delete(self, project_id: str) -> dict:
         return await self._client.delete(f"/projects/{project_id}")
+
+    async def get_subscription(self, project_id: str) -> dict:
+        return await self._client.get(f"/projects/{project_id}/subscription")
+
+    async def change_subscription_plan(self, project_id: str, to_plan_type: str) -> dict:
+        return await self._client.post(
+            f"/projects/{project_id}/subscription/actions/change-plan",
+            json={"toPlanType": to_plan_type},
+        )
+
+    async def add_subscription_credits(self, project_id: str, credits: int) -> dict:
+        return await self._client.post(
+            f"/projects/{project_id}/subscription/actions/add-credits",
+            json={"credits": credits},
+        )

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/persona_sdk/client.py

@@ -12,13 +12,14 @@ class PersonaClient:
             JWT tokens are automatically detected and validated by the server.
     """
 
-    def __init__(self, base_url: str, api_key: str):
+    def __init__(self, base_url: str, api_key: str, proxy: str = None):
         self.base_url = base_url.rstrip("/")
         self.api_key = api_key
         self._client = httpx.AsyncClient(
             base_url=self.base_url,
             headers={"x-persona-apikey": self.api_key},
             timeout=30.0,
+            proxy=proxy,
         )
 
     async def get(self, path: str, params: dict = None) -> dict:

{persona_sdk_python-0.1.0 → persona_sdk_python-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "persona-sdk-python"
-version = "0.1.0"
+version = "0.1.2"
 description = "Python SDK client for Persona AI platform"
 requires-python = ">=3.10"
 dependencies = [