agenta 0.30.0__py3-none-any.whl → 0.30.0a2__py3-none-any.whl

agenta/client/backend/testsets/client.py

@@ -10,8 +10,8 @@ from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
-from ..types.test_set_output_response import TestSetOutputResponse
 from ..core.jsonable_encoder import jsonable_encoder
+from ..types.test_set_output_response import TestSetOutputResponse
 from ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -106,11 +106,7 @@ class TestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def import_testset(
-        self,
-        *,
-        endpoint: typing.Optional[str] = OMIT,
-        testset_name: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> TestSetSimpleResponse:
         """
         Import JSON testset data from an endpoint and save it to MongoDB.
@@ -124,10 +120,6 @@ class TestsetsClient:
 
         Parameters
         ----------
-        endpoint : typing.Optional[str]
-
-        testset_name : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -149,12 +141,7 @@ class TestsetsClient:
         _response = self._client_wrapper.httpx_client.request(
             "testsets/endpoint",
             method="POST",
-            json={
-                "endpoint": endpoint,
-                "testset_name": testset_name,
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -180,59 +167,6 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_testsets(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[TestSetOutputResponse]:
-        """
-        Get all testsets.
-
-        Returns:
-
-        - A list of testset objects.
-
-        Raises:
-
-        - `HTTPException` with status code 404 if no testsets are found.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[TestSetOutputResponse]
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.testsets.get_testsets()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "testsets",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[TestSetOutputResponse],
-                    parse_obj_as(
-                        type_=typing.List[TestSetOutputResponse],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _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)
-
     def create_testset(
         self,
         *,
@@ -311,31 +245,31 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete_testsets(
+    def get_single_testset(
         self,
+        testset_id: str,
         *,
-        testset_ids: typing.Sequence[str],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[str]:
+    ) -> typing.Optional[typing.Any]:
         """
-        Delete specific testsets based on their unique IDs.
+        Fetch a specific testset in a MongoDB collection using its \_id.
 
         Args:
-        testset_ids (List[str]): The unique identifiers of the testsets to delete.
+        testset_id (str): The \_id of the testset to fetch.
 
         Returns:
-        A list of the deleted testsets' IDs.
+        The requested testset if found, else an HTTPException.
 
         Parameters
         ----------
-        testset_ids : typing.Sequence[str]
+        testset_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[str]
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -346,28 +280,21 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.delete_testsets(
-            testset_ids=["testset_ids"],
+        client.testsets.get_single_testset(
+            testset_id="testset_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "testsets",
-            method="DELETE",
-            json={
-                "testset_ids": testset_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+            f"testsets/{jsonable_encoder(testset_id)}",
+            method="GET",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[str],
+                    typing.Optional[typing.Any],
                     parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -386,27 +313,27 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def deprecating_create_testset(
+    def update_testset(
         self,
-        app_id: str,
+        testset_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> typing.Optional[typing.Any]:
         """
-        Create a testset with given name, save the testset to MongoDB.
+        Update a testset with given id, update the testset in MongoDB.
 
         Args:
-        name (str): name of the test set.
-        testset (Dict[str, str]): test set data.
+        testset_id (str): id of the test set to be updated.
+        csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set created.
+        str: The id of the test set updated.
 
         Parameters
         ----------
-        app_id : str
+        testset_id : str
 
         name : str
 
@@ -417,7 +344,7 @@ class TestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -428,15 +355,15 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.deprecating_create_testset(
-            app_id="app_id",
+        client.testsets.update_testset(
+            testset_id="testset_id",
             name="name",
             csvdata=[{"key": "value"}],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
-            method="POST",
+            f"testsets/{jsonable_encoder(testset_id)}",
+            method="PUT",
             json={
                 "name": name,
                 "csvdata": csvdata,
@@ -447,9 +374,9 @@ class TestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    TestSetSimpleResponse,
+                    typing.Optional[typing.Any],
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -468,31 +395,28 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_single_testset(
-        self,
-        testset_id: str,
-        *,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    def get_testsets(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[TestSetOutputResponse]:
         """
-        Fetch a specific testset in a MongoDB collection using its \_id.
-
-        Args:
-        testset_id (str): The \_id of the testset to fetch.
+        Get all testsets.
 
         Returns:
-        The requested testset if found, else an HTTPException.
+
+        - A list of testset objects.
+
+        Raises:
+
+        - `HTTPException` with status code 404 if no testsets are found.
 
         Parameters
         ----------
-        testset_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        typing.List[TestSetOutputResponse]
             Successful Response
 
         Examples
@@ -503,21 +427,19 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.get_single_testset(
-            testset_id="testset_id",
-        )
+        client.testsets.get_testsets()
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(testset_id)}",
+            "testsets",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    typing.List[TestSetOutputResponse],
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=typing.List[TestSetOutputResponse],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -536,38 +458,31 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update_testset(
+    def delete_testsets(
         self,
-        testset_id: str,
         *,
-        name: str,
-        csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        testset_ids: typing.Sequence[str],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> typing.List[str]:
         """
-        Update a testset with given id, update the testset in MongoDB.
+        Delete specific testsets based on their unique IDs.
 
         Args:
-        testset_id (str): id of the test set to be updated.
-        csvdata (NewTestset): New data to replace the old testset.
+        testset_ids (List[str]): The unique identifiers of the testsets to delete.
 
         Returns:
-        str: The id of the test set updated.
+        A list of the deleted testsets' IDs.
 
         Parameters
         ----------
-        testset_id : str
-
-        name : str
-
-        csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+        testset_ids : typing.Sequence[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        typing.List[str]
             Successful Response
 
         Examples
@@ -578,18 +493,15 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.update_testset(
-            testset_id="testset_id",
-            name="name",
-            csvdata=[{"key": "value"}],
+        client.testsets.delete_testsets(
+            testset_ids=["testset_ids"],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(testset_id)}",
-            method="PUT",
+            "testsets",
+            method="DELETE",
             json={
-                "name": name,
-                "csvdata": csvdata,
+                "testset_ids": testset_ids,
             },
             request_options=request_options,
             omit=OMIT,
@@ -597,9 +509,9 @@ class TestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    typing.List[str],
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=typing.List[str],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -715,11 +627,7 @@ class AsyncTestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def import_testset(
-        self,
-        *,
-        endpoint: typing.Optional[str] = OMIT,
-        testset_name: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> TestSetSimpleResponse:
         """
         Import JSON testset data from an endpoint and save it to MongoDB.
@@ -733,10 +641,6 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
-        endpoint : typing.Optional[str]
-
-        testset_name : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -766,12 +670,7 @@ class AsyncTestsetsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "testsets/endpoint",
             method="POST",
-            json={
-                "endpoint": endpoint,
-                "testset_name": testset_name,
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -797,67 +696,6 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_testsets(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[TestSetOutputResponse]:
-        """
-        Get all testsets.
-
-        Returns:
-
-        - A list of testset objects.
-
-        Raises:
-
-        - `HTTPException` with status code 404 if no testsets are found.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[TestSetOutputResponse]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.testsets.get_testsets()
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "testsets",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[TestSetOutputResponse],
-                    parse_obj_as(
-                        type_=typing.List[TestSetOutputResponse],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _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)
-
     async def create_testset(
         self,
         *,
@@ -944,31 +782,31 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def delete_testsets(
+    async def get_single_testset(
         self,
+        testset_id: str,
         *,
-        testset_ids: typing.Sequence[str],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[str]:
+    ) -> typing.Optional[typing.Any]:
         """
-        Delete specific testsets based on their unique IDs.
+        Fetch a specific testset in a MongoDB collection using its \_id.
 
         Args:
-        testset_ids (List[str]): The unique identifiers of the testsets to delete.
+        testset_id (str): The \_id of the testset to fetch.
 
         Returns:
-        A list of the deleted testsets' IDs.
+        The requested testset if found, else an HTTPException.
 
         Parameters
         ----------
-        testset_ids : typing.Sequence[str]
+        testset_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[str]
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -984,31 +822,24 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.delete_testsets(
-                testset_ids=["testset_ids"],
+            await client.testsets.get_single_testset(
+                testset_id="testset_id",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "testsets",
-            method="DELETE",
-            json={
-                "testset_ids": testset_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+            f"testsets/{jsonable_encoder(testset_id)}",
+            method="GET",
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[str],
+                    typing.Optional[typing.Any],
                     parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1027,27 +858,27 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def deprecating_create_testset(
+    async def update_testset(
         self,
-        app_id: str,
+        testset_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> typing.Optional[typing.Any]:
         """
-        Create a testset with given name, save the testset to MongoDB.
+        Update a testset with given id, update the testset in MongoDB.
 
         Args:
-        name (str): name of the test set.
-        testset (Dict[str, str]): test set data.
+        testset_id (str): id of the test set to be updated.
+        csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set created.
+        str: The id of the test set updated.
 
         Parameters
         ----------
-        app_id : str
+        testset_id : str
 
         name : str
 
@@ -1058,7 +889,7 @@ class AsyncTestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -1074,8 +905,8 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.deprecating_create_testset(
-                app_id="app_id",
+            await client.testsets.update_testset(
+                testset_id="testset_id",
                 name="name",
                 csvdata=[{"key": "value"}],
             )
@@ -1084,8 +915,8 @@ class AsyncTestsetsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
-            method="POST",
+            f"testsets/{jsonable_encoder(testset_id)}",
+            method="PUT",
             json={
                 "name": name,
                 "csvdata": csvdata,
@@ -1096,9 +927,9 @@ class AsyncTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    TestSetSimpleResponse,
+                    typing.Optional[typing.Any],
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1117,31 +948,28 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_single_testset(
-        self,
-        testset_id: str,
-        *,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    async def get_testsets(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[TestSetOutputResponse]:
         """
-        Fetch a specific testset in a MongoDB collection using its \_id.
-
-        Args:
-        testset_id (str): The \_id of the testset to fetch.
+        Get all testsets.
 
         Returns:
-        The requested testset if found, else an HTTPException.
+
+        - A list of testset objects.
+
+        Raises:
+
+        - `HTTPException` with status code 404 if no testsets are found.
 
         Parameters
         ----------
-        testset_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        typing.List[TestSetOutputResponse]
             Successful Response
 
         Examples
@@ -1157,24 +985,22 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.get_single_testset(
-                testset_id="testset_id",
-            )
+            await client.testsets.get_testsets()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(testset_id)}",
+            "testsets",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    typing.List[TestSetOutputResponse],
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=typing.List[TestSetOutputResponse],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1193,38 +1019,31 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update_testset(
+    async def delete_testsets(
         self,
-        testset_id: str,
         *,
-        name: str,
-        csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        testset_ids: typing.Sequence[str],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> typing.List[str]:
         """
-        Update a testset with given id, update the testset in MongoDB.
+        Delete specific testsets based on their unique IDs.
 
         Args:
-        testset_id (str): id of the test set to be updated.
-        csvdata (NewTestset): New data to replace the old testset.
+        testset_ids (List[str]): The unique identifiers of the testsets to delete.
 
         Returns:
-        str: The id of the test set updated.
+        A list of the deleted testsets' IDs.
 
         Parameters
         ----------
-        testset_id : str
-
-        name : str
-
-        csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+        testset_ids : typing.Sequence[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        typing.List[str]
             Successful Response
 
         Examples
@@ -1240,21 +1059,18 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.update_testset(
-                testset_id="testset_id",
-                name="name",
-                csvdata=[{"key": "value"}],
+            await client.testsets.delete_testsets(
+                testset_ids=["testset_ids"],
             )
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(testset_id)}",
-            method="PUT",
+            "testsets",
+            method="DELETE",
             json={
-                "name": name,
-                "csvdata": csvdata,
+                "testset_ids": testset_ids,
             },
             request_options=request_options,
             omit=OMIT,
@@ -1262,9 +1078,9 @@ class AsyncTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    typing.List[str],
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=typing.List[str],  # type: ignore
                         object_=_response.json(),
                     ),
                 )