vellum-ai 0.6.7__tar.gz → 0.6.9__tar.gz

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/PKG-INFO

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

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/pyproject.toml

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

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/src/vellum/__init__.py

@@ -193,6 +193,7 @@ from .types import (
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
     IntfloatMultilingualE5LargeEnum,
+    IterationStateEnum,
     JsonEnum,
     JsonInputRequest,
     JsonVariableValue,
@@ -826,6 +827,7 @@ __all__ = [
     "InstructorVectorizerConfigRequest",
     "InternalServerError",
     "IntfloatMultilingualE5LargeEnum",
+    "IterationStateEnum",
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/src/vellum/core/client_wrapper.py

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

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/src/vellum/resources/document_indexes/client.py

@@ -256,10 +256,10 @@ class DocumentIndexesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
-        Used to fully update a Document Index given its ID.
+        Used to fully update a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - label: str. A human-readable label for the document index
 
@@ -329,10 +329,10 @@ class DocumentIndexesClient:
 
     def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
-        Used to delete a Document Index given its ID.
+        Used to delete a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
@@ -385,10 +385,10 @@ class DocumentIndexesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
-        Used to partial update a Document Index given its ID.
+        Used to partial update a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - label: typing.Optional[str]. A human-readable label for the document index
 
@@ -458,6 +458,60 @@ class DocumentIndexesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def remove_document(
+        self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Removes a Document from a Document Index without deleting the Document itself.
+
+        Parameters:
+            - document_id: str. Either the Vellum-generated ID or the originally supplied external_id that uniquely identifies the Document you'd like to remove.
+
+            - id: str. Either the Vellum-generated ID or the originally specified name that uniquely identifies the Document Index from which you'd like to remove a Document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.document_indexes.remove_document(
+            document_id="document_id",
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="DELETE",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/",
+                f"v1/document-indexes/{jsonable_encoder(id)}/documents/{jsonable_encoder(document_id)}",
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        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 AsyncDocumentIndexesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -694,10 +748,10 @@ class AsyncDocumentIndexesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
-        Used to fully update a Document Index given its ID.
+        Used to fully update a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - label: str. A human-readable label for the document index
 
@@ -767,10 +821,10 @@ class AsyncDocumentIndexesClient:
 
     async def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
-        Used to delete a Document Index given its ID.
+        Used to delete a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
@@ -823,10 +877,10 @@ class AsyncDocumentIndexesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DocumentIndexRead:
         """
-        Used to partial update a Document Index given its ID.
+        Used to partial update a Document Index given its ID or name.
 
         Parameters:
-            - id: str. A UUID string identifying this document index.
+            - id: str. Either the Document Index's ID or its unique name
 
             - label: typing.Optional[str]. A human-readable label for the document index
 
@@ -895,3 +949,57 @@ class AsyncDocumentIndexesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def remove_document(
+        self, document_id: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Removes a Document from a Document Index without deleting the Document itself.
+
+        Parameters:
+            - document_id: str. Either the Vellum-generated ID or the originally supplied external_id that uniquely identifies the Document you'd like to remove.
+
+            - id: str. Either the Vellum-generated ID or the originally specified name that uniquely identifies the Document Index from which you'd like to remove a Document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+        await client.document_indexes.remove_document(
+            document_id="document_id",
+            id="id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="DELETE",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/",
+                f"v1/document-indexes/{jsonable_encoder(id)}/documents/{jsonable_encoder(document_id)}",
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        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.6.7 → vellum_ai-0.6.9}/src/vellum/types/__init__.py

@@ -230,6 +230,7 @@ from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEve
 from .instructor_vectorizer_config import InstructorVectorizerConfig
 from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
 from .intfloat_multilingual_e_5_large_enum import IntfloatMultilingualE5LargeEnum
+from .iteration_state_enum import IterationStateEnum
 from .json_enum import JsonEnum
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
@@ -884,6 +885,7 @@ __all__ = [
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
     "IntfloatMultilingualE5LargeEnum",
+    "IterationStateEnum",
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",

vellum_ai-0.6.9/src/vellum/types/iteration_state_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+IterationStateEnum = typing.Union[typing.Literal["INITIATED", "FULFILLED"], typing.Any]

{vellum_ai-0.6.7 → vellum_ai-0.6.9}/src/vellum/types/map_node_result_data.py

@@ -5,10 +5,12 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .iteration_state_enum import IterationStateEnum
 
 
 class MapNodeResultData(pydantic_v1.BaseModel):
     execution_ids: typing.List[str]
+    iteration_state: typing.Optional[IterationStateEnum] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}