persona-sdk-python 0.2.12__tar.gz → 0.2.13__tar.gz

{persona_sdk_python-0.2.12 → persona_sdk_python-0.2.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: persona-sdk-python
-Version: 0.2.12
+Version: 0.2.13
 Summary: Async Python SDK for the Persona AI platform
 Project-URL: Homepage, https://persona.applica.guru
 Project-URL: Repository, https://bitbucket.org/applicaguru/persona-sdk-python

{persona_sdk_python-0.2.12 → persona_sdk_python-0.2.13}/persona_sdk/__init__.py

@@ -3,6 +3,7 @@ from typing import Union
 from persona_sdk.api.agents import AgentsApi
 from persona_sdk.api.credentials import CredentialsApi
 from persona_sdk.api.features import FeaturesApi
+from persona_sdk.api.known_models import KnownModelsApi
 from persona_sdk.api.knowledge_bases import KnowledgeBasesApi
 from persona_sdk.api.missions import MissionsApi
 from persona_sdk.api.projects import ProjectsApi
@@ -26,6 +27,7 @@ __all__ = [
     "AgentsApi",
     "CredentialsApi",
     "FeaturesApi",
+    "KnownModelsApi",
     "KnowledgeBasesApi",
     "MissionsApi",
     "ProjectsApi",
@@ -93,6 +95,7 @@ class PersonaSdk:
         self.features = FeaturesApi(self.client)
         self.missions = MissionsApi(self.client)
         self.service_prices = ServicePricesApi(self.client)
+        self.known_models = KnownModelsApi(self.client)
 
     async def close(self):
         await self.client.close()

persona_sdk_python-0.2.13/persona_sdk/api/known_models.py

@@ -0,0 +1,24 @@
+from persona_sdk.client import PersonaClient
+
+
+class KnownModelsApi:
+    def __init__(self, client: PersonaClient):
+        self._client = client
+
+    async def list(self, keyword: str = None, page: int = 1, size: int = 100) -> dict:
+        params = {"page": page, "size": size}
+        if keyword:
+            params["keyword"] = keyword
+        return await self._client.get("/known-models", params=params)
+
+    async def get(self, known_model_id: str) -> dict:
+        return await self._client.get(f"/known-models/{known_model_id}")
+
+    async def create(self, known_model: dict) -> dict:
+        return await self._client.post("/known-models", json=known_model)
+
+    async def update(self, known_model_id: str, known_model: dict) -> dict:
+        return await self._client.put(f"/known-models/{known_model_id}", json=known_model)
+
+    async def delete(self, known_model_id: str) -> dict:
+        return await self._client.delete(f"/known-models/{known_model_id}")

{persona_sdk_python-0.2.12 → persona_sdk_python-0.2.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "persona-sdk-python"
-version = "0.2.12"
+version = "0.2.13"
 description = "Async Python SDK for the Persona AI platform"
 readme = "README.md"
 requires-python = ">=3.10"