vellum-ai 0.3.5__py3-none-any.whl → 0.3.7__py3-none-any.whl

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/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/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.5",
+            "X-Fern-SDK-Version": "0.3.7",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

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/resources/folder_entities/__init__.py

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

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.5.dist-info → vellum_ai-0.3.7.dist-info}/METADATA

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

{vellum_ai-0.3.5.dist-info → vellum_ai-0.3.7.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=aLZO9kazzLFke5f7B88oC3peXNupeB23aCL4DymcEKc,25339
-vellum/client.py,sha256=AwAWecRomi1ccqhnU8IdyXHh_FvnfVcpGLNYRL5ordY,61052
+vellum/__init__.py,sha256=suJJOLGlcpd1_lryb4KwPCbyyVJ11Fz-heOV3lB0JQQ,25383
+vellum/client.py,sha256=GXr3l2QqlYu7Fb690wuDD-9rnnbqqvcTDpc48bbjTr4,61329
 vellum/core/__init__.py,sha256=QJS3CJ2TYP2E1Tge0CS6Z7r8LTNzJHQVX1hD3558eP0,519
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=SFL89DB_CXYnSsprt-Nz0zWJPsGTMibyH5_Gun7VpN0,1212
+vellum/core/client_wrapper.py,sha256=YQr66uYsp2oKQr0vo4qu84rJSVTUzB2U8oPYkWi8tMM,1212
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
 vellum/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
@@ -14,7 +14,7 @@ vellum/errors/forbidden_error.py,sha256=dgnatOGair3CvxljCE45_qwN_yefzcw2G0vw88wr
 vellum/errors/internal_server_error.py,sha256=E0rgqJC0-LcetLi1HmSi92KpvNkGSRCIdBeEqT_ln1s,252
 vellum/errors/not_found_error.py,sha256=P65k-Lm2RuefAVSNLER5hH-4P99SGohKy2cOPSrIxNk,246
 vellum/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/resources/__init__.py,sha256=vtYfeQ0Owhn3uLX2j7Y16vrh6-8Y4QkZ1IDu9Dne9Po,631
+vellum/resources/__init__.py,sha256=E7dvnCySQgcXn9AADWlkZw4zNN84-8dH3WZ-BgPclhs,675
 vellum/resources/deployments/__init__.py,sha256=AE0TcFwLrLBljM0ZDX-pPw4Kqt-1f5JDpIok2HS80QI,157
 vellum/resources/deployments/client.py,sha256=rqD2zsLBlRwWE619Nb9m3wSEC9DAkz8RLJX8lVx_Ags,10812
 vellum/resources/deployments/types/__init__.py,sha256=IhwnmoXJ0r_QEhh1b2tBcaAm_x3fWMVuIhYmAapp_ZA,183
@@ -23,6 +23,8 @@ vellum/resources/document_indexes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8
 vellum/resources/document_indexes/client.py,sha256=nAZ27OvbZg4d94CW-J8JBImiyClWvGaRhkQeEgwUZGA,10894
 vellum/resources/documents/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/documents/client.py,sha256=V64IA5_vrjq52i_DUV1LEy4jvcwalww5fYxoiVauabA,17776
+vellum/resources/folder_entities/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+vellum/resources/folder_entities/client.py,sha256=KBC-YKkUQTE31etwG0D9MJF_VW_KiZ0eCi18JYzLE8s,3748
 vellum/resources/model_versions/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/model_versions/client.py,sha256=6OQ4x0yqdd3kjjI88HXbJLx1W7qblme4YueHYr7j9pg,3053
 vellum/resources/registered_prompts/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -287,7 +289,7 @@ vellum/types/workflow_result_event_output_data_number.py,sha256=5IG_4XZhUZjwCfX1
 vellum/types/workflow_result_event_output_data_search_results.py,sha256=FZeTLuHIBJp0AZUqBOzzMN4ntf_Q3hKP4m3vIzVq2cQ,1404
 vellum/types/workflow_result_event_output_data_string.py,sha256=XJ7ZFTS2eqIMwa-zXFPDowu3o3JnRUfxC1MJIk8nPDI,1478
 vellum/types/workflow_stream_event.py,sha256=OQUSzwoM-OCfWxNzeOVVLsjCue_WWqin3tGMtwvp_rc,873
-vellum_ai-0.3.5.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.3.5.dist-info/METADATA,sha256=D0pa-7G30aRi6lkGH-V1FqkqfELLubFunrwWKAnTc30,3486
-vellum_ai-0.3.5.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.3.5.dist-info/RECORD,,
+vellum_ai-0.3.7.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.3.7.dist-info/METADATA,sha256=-ckpfJoQ8AQlGQ3m58d4zbzH8bVV-GnvdYwcOTWySnk,3486
+vellum_ai-0.3.7.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.3.7.dist-info/RECORD,,