vellum-ai 0.3.6__tar.gz → 0.3.7__tar.gz

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.3.6
+Version: 0.3.7
 Summary: 
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.3.6"
+version = "0.3.7"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/src/vellum/__init__.py

@@ -367,6 +367,7 @@ from .resources import (
     deployments,
     document_indexes,
     documents,
+    folder_entities,
     model_versions,
     registered_prompts,
     sandboxes,
@@ -745,6 +746,7 @@ __all__ = [
     "deployments",
     "document_indexes",
     "documents",
+    "folder_entities",
     "model_versions",
     "registered_prompts",
     "sandboxes",

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/src/vellum/client.py

@@ -18,6 +18,7 @@ from .errors.not_found_error import NotFoundError
 from .resources.deployments.client import AsyncDeploymentsClient, DeploymentsClient
 from .resources.document_indexes.client import AsyncDocumentIndexesClient, DocumentIndexesClient
 from .resources.documents.client import AsyncDocumentsClient, DocumentsClient
+from .resources.folder_entities.client import AsyncFolderEntitiesClient, FolderEntitiesClient
 from .resources.model_versions.client import AsyncModelVersionsClient, ModelVersionsClient
 from .resources.registered_prompts.client import AsyncRegisteredPromptsClient, RegisteredPromptsClient
 from .resources.sandboxes.client import AsyncSandboxesClient, SandboxesClient
@@ -67,6 +68,7 @@ class Vellum:
         self.deployments = DeploymentsClient(client_wrapper=self._client_wrapper)
         self.document_indexes = DocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = DocumentsClient(client_wrapper=self._client_wrapper)
+        self.folder_entities = FolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.model_versions = ModelVersionsClient(client_wrapper=self._client_wrapper)
         self.registered_prompts = RegisteredPromptsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = SandboxesClient(client_wrapper=self._client_wrapper)
@@ -649,6 +651,7 @@ class AsyncVellum:
         self.deployments = AsyncDeploymentsClient(client_wrapper=self._client_wrapper)
         self.document_indexes = AsyncDocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = AsyncDocumentsClient(client_wrapper=self._client_wrapper)
+        self.folder_entities = AsyncFolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.model_versions = AsyncModelVersionsClient(client_wrapper=self._client_wrapper)
         self.registered_prompts = AsyncRegisteredPromptsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = AsyncSandboxesClient(client_wrapper=self._client_wrapper)

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/src/vellum/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.3.6",
+            "X-Fern-SDK-Version": "0.3.7",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.3.6 → vellum_ai-0.3.7}/src/vellum/resources/__init__.py

@@ -4,6 +4,7 @@ from . import (
     deployments,
     document_indexes,
     documents,
+    folder_entities,
     model_versions,
     registered_prompts,
     sandboxes,
@@ -19,6 +20,7 @@ __all__ = [
     "deployments",
     "document_indexes",
     "documents",
+    "folder_entities",
     "model_versions",
     "registered_prompts",
     "sandboxes",

vellum_ai-0.3.7/src/vellum/resources/folder_entities/client.py

@@ -0,0 +1,98 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+import urllib.parse
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.jsonable_encoder import jsonable_encoder
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class FolderEntitiesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def add_entity_to_folder(self, folder_id: str, *, entity_id: str) -> None:
+        """
+        Add an entity to a specific folder or root directory.
+
+        Adding an entity to a folder will remove it from any other folders it might have been a member of.
+
+        Parameters:
+            - folder_id: str. The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root directory (e.g. "PROMPT_SANDBOX").
+
+            - entity_id: str. The ID of the entity you would like to move.
+        ---
+        from vellum.client import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.folder_entities.add_entity_to_folder(
+            folder_id="folder_id",
+            entity_id="entity_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/folders/{folder_id}/add-entity"
+            ),
+            json=jsonable_encoder({"entity_id": entity_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncFolderEntitiesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def add_entity_to_folder(self, folder_id: str, *, entity_id: str) -> None:
+        """
+        Add an entity to a specific folder or root directory.
+
+        Adding an entity to a folder will remove it from any other folders it might have been a member of.
+
+        Parameters:
+            - folder_id: str. The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root directory (e.g. "PROMPT_SANDBOX").
+
+            - entity_id: str. The ID of the entity you would like to move.
+        ---
+        from vellum.client import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+        await client.folder_entities.add_entity_to_folder(
+            folder_id="folder_id",
+            entity_id="entity_id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/folders/{folder_id}/add-entity"
+            ),
+            json=jsonable_encoder({"entity_id": entity_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

vellum_ai-0.3.7/src/vellum/resources/test_suites/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+