agenta 0.59.0__py3-none-any.whl → 0.59.2__py3-none-any.whl

agenta/client/backend/client.py

@@ -98,13 +98,17 @@ class AgentaApi:
         self._client_wrapper = SyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.Client(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.Client(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.Client(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.Client(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self._raw_client = RawAgentaApi(client_wrapper=self._client_wrapper)
@@ -172,7 +176,7 @@ class AgentaApi:
 
     def update_organization(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
@@ -182,7 +186,7 @@ class AgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : typing.Optional[str]
 
@@ -206,11 +210,11 @@ class AgentaApi:
             api_key="YOUR_API_KEY",
         )
         client.update_organization(
-            org_id="org_id",
+            organization_id="organization_id",
         )
         """
         _response = self._raw_client.update_organization(
-            org_id,
+            organization_id,
             name=name,
             description=description,
             updated_at=updated_at,
@@ -220,7 +224,7 @@ class AgentaApi:
 
     def create_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: str,
         description: typing.Optional[str] = OMIT,
@@ -230,7 +234,7 @@ class AgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : str
 
@@ -254,12 +258,12 @@ class AgentaApi:
             api_key="YOUR_API_KEY",
         )
         client.create_workspace(
-            org_id="org_id",
+            organization_id="organization_id",
             name="name",
         )
         """
         _response = self._raw_client.create_workspace(
-            org_id,
+            organization_id,
             name=name,
             description=description,
             type=type,
@@ -269,7 +273,7 @@ class AgentaApi:
 
     def update_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         name: typing.Optional[str] = OMIT,
@@ -280,7 +284,7 @@ class AgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -306,12 +310,12 @@ class AgentaApi:
             api_key="YOUR_API_KEY",
         )
         client.update_workspace(
-            org_id="org_id",
+            organization_id="organization_id",
             workspace_id="workspace_id",
         )
         """
         _response = self._raw_client.update_workspace(
-            org_id,
+            organization_id,
             workspace_id,
             name=name,
             description=description,
@@ -427,7 +431,7 @@ class AgentaApi:
         workspace_id: str,
         *,
         email: str,
-        org_id: str,
+        organization_id: str,
         role: str,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Optional[typing.Any]:
@@ -437,7 +441,7 @@ class AgentaApi:
         Args:
             workspace_id (str): The ID of the workspace.
             email (str): The email of the user to remove the role from.
-            org_id (str): The ID of the organization.
+            organization_id (str): The ID of the organization.
             role (str): The role to remove from the user.
             request (Request): The FastAPI request object.
 
@@ -454,7 +458,7 @@ class AgentaApi:
 
         email : str
 
-        org_id : str
+        organization_id : str
 
         role : str
 
@@ -476,14 +480,14 @@ class AgentaApi:
         client.unassign_role_from_user(
             workspace_id="workspace_id",
             email="email",
-            org_id="org_id",
+            organization_id="organization_id",
             role="role",
         )
         """
         _response = self._raw_client.unassign_role_from_user(
             workspace_id,
             email=email,
-            org_id=org_id,
+            organization_id=organization_id,
             role=role,
             request_options=request_options,
         )
@@ -657,13 +661,17 @@ class AsyncAgentaApi:
         self._client_wrapper = AsyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.AsyncClient(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.AsyncClient(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.AsyncClient(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self._raw_client = AsyncRawAgentaApi(client_wrapper=self._client_wrapper)
@@ -743,7 +751,7 @@ class AsyncAgentaApi:
 
     async def update_organization(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
@@ -753,7 +761,7 @@ class AsyncAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : typing.Optional[str]
 
@@ -782,14 +790,14 @@ class AsyncAgentaApi:
 
         async def main() -> None:
             await client.update_organization(
-                org_id="org_id",
+                organization_id="organization_id",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._raw_client.update_organization(
-            org_id,
+            organization_id,
             name=name,
             description=description,
             updated_at=updated_at,
@@ -799,7 +807,7 @@ class AsyncAgentaApi:
 
     async def create_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: str,
         description: typing.Optional[str] = OMIT,
@@ -809,7 +817,7 @@ class AsyncAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : str
 
@@ -838,7 +846,7 @@ class AsyncAgentaApi:
 
         async def main() -> None:
             await client.create_workspace(
-                org_id="org_id",
+                organization_id="organization_id",
                 name="name",
             )
 
@@ -846,7 +854,7 @@ class AsyncAgentaApi:
         asyncio.run(main())
         """
         _response = await self._raw_client.create_workspace(
-            org_id,
+            organization_id,
             name=name,
             description=description,
             type=type,
@@ -856,7 +864,7 @@ class AsyncAgentaApi:
 
     async def update_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         name: typing.Optional[str] = OMIT,
@@ -867,7 +875,7 @@ class AsyncAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -898,7 +906,7 @@ class AsyncAgentaApi:
 
         async def main() -> None:
             await client.update_workspace(
-                org_id="org_id",
+                organization_id="organization_id",
                 workspace_id="workspace_id",
             )
 
@@ -906,7 +914,7 @@ class AsyncAgentaApi:
         asyncio.run(main())
         """
         _response = await self._raw_client.update_workspace(
-            org_id,
+            organization_id,
             workspace_id,
             name=name,
             description=description,
@@ -1038,7 +1046,7 @@ class AsyncAgentaApi:
         workspace_id: str,
         *,
         email: str,
-        org_id: str,
+        organization_id: str,
         role: str,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Optional[typing.Any]:
@@ -1048,7 +1056,7 @@ class AsyncAgentaApi:
         Args:
             workspace_id (str): The ID of the workspace.
             email (str): The email of the user to remove the role from.
-            org_id (str): The ID of the organization.
+            organization_id (str): The ID of the organization.
             role (str): The role to remove from the user.
             request (Request): The FastAPI request object.
 
@@ -1065,7 +1073,7 @@ class AsyncAgentaApi:
 
         email : str
 
-        org_id : str
+        organization_id : str
 
         role : str
 
@@ -1092,7 +1100,7 @@ class AsyncAgentaApi:
             await client.unassign_role_from_user(
                 workspace_id="workspace_id",
                 email="email",
-                org_id="org_id",
+                organization_id="organization_id",
                 role="role",
             )
 
@@ -1102,7 +1110,7 @@ class AsyncAgentaApi:
         _response = await self._raw_client.unassign_role_from_user(
             workspace_id,
             email=email,
-            org_id=org_id,
+            organization_id=organization_id,
             role=role,
             request_options=request_options,
         )

agenta/client/backend/organization/client.py

@@ -29,14 +29,17 @@ class OrganizationClient:
         return self._raw_client
 
     def fetch_organization_details(
-        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        organization_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> OrganizationDetails:
         """
         Return the details of the organization.
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -54,11 +57,11 @@ class OrganizationClient:
             api_key="YOUR_API_KEY",
         )
         client.organization.fetch_organization_details(
-            org_id="org_id",
+            organization_id="organization_id",
         )
         """
         _response = self._raw_client.fetch_organization_details(
-            org_id, request_options=request_options
+            organization_id, request_options=request_options
         )
         return _response.data
 
@@ -98,7 +101,7 @@ class OrganizationClient:
 
     def invite_user_to_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         request: typing.Sequence[InviteRequest],
@@ -108,7 +111,7 @@ class OrganizationClient:
         Assigns a role to a user in an organization.
 
         Args:
-            org_id (str): The ID of the organization.
+            organization_id (str): The ID of the organization.
             payload (InviteRequest): The payload containing the organization id, user email, and role to assign.
             workspace_id (str): The ID of the workspace.
 
@@ -121,7 +124,7 @@ class OrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -143,7 +146,7 @@ class OrganizationClient:
             api_key="YOUR_API_KEY",
         )
         client.organization.invite_user_to_workspace(
-            org_id="org_id",
+            organization_id="organization_id",
             workspace_id="workspace_id",
             request=[
                 InviteRequest(
@@ -153,13 +156,16 @@ class OrganizationClient:
         )
         """
         _response = self._raw_client.invite_user_to_workspace(
-            org_id, workspace_id, request=request, request_options=request_options
+            organization_id,
+            workspace_id,
+            request=request,
+            request_options=request_options,
         )
         return _response.data
 
     def resend_invitation(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         email: str,
@@ -178,7 +184,7 @@ class OrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -200,19 +206,19 @@ class OrganizationClient:
             api_key="YOUR_API_KEY",
         )
         client.organization.resend_invitation(
-            org_id="org_id",
+            organization_id="organization_id",
             workspace_id="workspace_id",
             email="email",
         )
         """
         _response = self._raw_client.resend_invitation(
-            org_id, workspace_id, email=email, request_options=request_options
+            organization_id, workspace_id, email=email, request_options=request_options
         )
         return _response.data
 
     def accept_invitation(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         project_id: str,
@@ -233,7 +239,7 @@ class OrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -259,7 +265,7 @@ class OrganizationClient:
             api_key="YOUR_API_KEY",
         )
         client.organization.accept_invitation(
-            org_id="org_id",
+            organization_id="organization_id",
             workspace_id="workspace_id",
             project_id="project_id",
             token="token",
@@ -267,7 +273,7 @@ class OrganizationClient:
         )
         """
         _response = self._raw_client.accept_invitation(
-            org_id,
+            organization_id,
             workspace_id,
             project_id=project_id,
             token=token,
@@ -293,14 +299,17 @@ class AsyncOrganizationClient:
         return self._raw_client
 
     async def fetch_organization_details(
-        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        organization_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> OrganizationDetails:
         """
         Return the details of the organization.
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -323,14 +332,14 @@ class AsyncOrganizationClient:
 
         async def main() -> None:
             await client.organization.fetch_organization_details(
-                org_id="org_id",
+                organization_id="organization_id",
             )
 
 
         asyncio.run(main())
         """
         _response = await self._raw_client.fetch_organization_details(
-            org_id, request_options=request_options
+            organization_id, request_options=request_options
         )
         return _response.data
 
@@ -380,7 +389,7 @@ class AsyncOrganizationClient:
 
     async def invite_user_to_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         request: typing.Sequence[InviteRequest],
@@ -390,7 +399,7 @@ class AsyncOrganizationClient:
         Assigns a role to a user in an organization.
 
         Args:
-            org_id (str): The ID of the organization.
+            organization_id (str): The ID of the organization.
             payload (InviteRequest): The payload containing the organization id, user email, and role to assign.
             workspace_id (str): The ID of the workspace.
 
@@ -403,7 +412,7 @@ class AsyncOrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -430,7 +439,7 @@ class AsyncOrganizationClient:
 
         async def main() -> None:
             await client.organization.invite_user_to_workspace(
-                org_id="org_id",
+                organization_id="organization_id",
                 workspace_id="workspace_id",
                 request=[
                     InviteRequest(
@@ -443,13 +452,16 @@ class AsyncOrganizationClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.invite_user_to_workspace(
-            org_id, workspace_id, request=request, request_options=request_options
+            organization_id,
+            workspace_id,
+            request=request,
+            request_options=request_options,
         )
         return _response.data
 
     async def resend_invitation(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         email: str,
@@ -468,7 +480,7 @@ class AsyncOrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -495,7 +507,7 @@ class AsyncOrganizationClient:
 
         async def main() -> None:
             await client.organization.resend_invitation(
-                org_id="org_id",
+                organization_id="organization_id",
                 workspace_id="workspace_id",
                 email="email",
             )
@@ -504,13 +516,13 @@ class AsyncOrganizationClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.resend_invitation(
-            org_id, workspace_id, email=email, request_options=request_options
+            organization_id, workspace_id, email=email, request_options=request_options
         )
         return _response.data
 
     async def accept_invitation(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         project_id: str,
@@ -531,7 +543,7 @@ class AsyncOrganizationClient:
 
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -562,7 +574,7 @@ class AsyncOrganizationClient:
 
         async def main() -> None:
             await client.organization.accept_invitation(
-                org_id="org_id",
+                organization_id="organization_id",
                 workspace_id="workspace_id",
                 project_id="project_id",
                 token="token",
@@ -573,7 +585,7 @@ class AsyncOrganizationClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.accept_invitation(
-            org_id,
+            organization_id,
             workspace_id,
             project_id=project_id,
             token=token,