structifyai 1.179.0__py3-none-any.whl → 1.182.0__py3-none-any.whl

structify/resources/sessions.py

@@ -17,6 +17,7 @@ from ..types import (
     session_finalize_dag_params,
     session_mark_errored_params,
     session_create_session_params,
+    session_edit_node_output_params,
     session_request_confirmation_params,
     session_update_node_progress_params,
     session_upload_dashboard_layout_params,
@@ -49,12 +50,14 @@ from ..types.edge_spec_param import EdgeSpecParam
 from ..types.node_spec_param import NodeSpecParam
 from ..types.workflow_session import WorkflowSession
 from ..types.get_node_response import GetNodeResponse
+from ..types.parquet_edit_param import ParquetEditParam
 from ..types.finalize_dag_response import FinalizeDagResponse
 from ..types.workflow_session_node import WorkflowSessionNode
 from ..types.get_node_logs_response import GetNodeLogsResponse
 from ..types.session_kill_jobs_response import SessionKillJobsResponse
 from ..types.session_get_events_response import SessionGetEventsResponse
 from ..types.workflow_node_execution_status import WorkflowNodeExecutionStatus
+from ..types.session_edit_node_output_response import SessionEditNodeOutputResponse
 from ..types.session_get_node_progress_response import SessionGetNodeProgressResponse
 
 __all__ = ["SessionsResource", "AsyncSessionsResource"]
@@ -150,6 +153,39 @@ class SessionsResource(SyncAPIResource):
             cast_to=WorkflowSession,
         )
 
+    def edit_node_output(
+        self,
+        node_id: str,
+        *,
+        edits: Iterable[ParquetEditParam],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> SessionEditNodeOutputResponse:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not node_id:
+            raise ValueError(f"Expected a non-empty value for `node_id` but received {node_id!r}")
+        return self._post(
+            f"/sessions/nodes/{node_id}/edit_output",
+            body=maybe_transform({"edits": edits}, session_edit_node_output_params.SessionEditNodeOutputParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionEditNodeOutputResponse,
+        )
+
     def finalize_dag(
         self,
         session_id: str,
@@ -823,6 +859,41 @@ class AsyncSessionsResource(AsyncAPIResource):
             cast_to=WorkflowSession,
         )
 
+    async def edit_node_output(
+        self,
+        node_id: str,
+        *,
+        edits: Iterable[ParquetEditParam],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> SessionEditNodeOutputResponse:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not node_id:
+            raise ValueError(f"Expected a non-empty value for `node_id` but received {node_id!r}")
+        return await self._post(
+            f"/sessions/nodes/{node_id}/edit_output",
+            body=await async_maybe_transform(
+                {"edits": edits}, session_edit_node_output_params.SessionEditNodeOutputParams
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionEditNodeOutputResponse,
+        )
+
     async def finalize_dag(
         self,
         session_id: str,
@@ -1416,6 +1487,9 @@ class SessionsResourceWithRawResponse:
         self.create_session = to_raw_response_wrapper(
             sessions.create_session,
         )
+        self.edit_node_output = to_raw_response_wrapper(
+            sessions.edit_node_output,
+        )
         self.finalize_dag = to_raw_response_wrapper(
             sessions.finalize_dag,
         )
@@ -1474,6 +1548,9 @@ class AsyncSessionsResourceWithRawResponse:
         self.create_session = async_to_raw_response_wrapper(
             sessions.create_session,
         )
+        self.edit_node_output = async_to_raw_response_wrapper(
+            sessions.edit_node_output,
+        )
         self.finalize_dag = async_to_raw_response_wrapper(
             sessions.finalize_dag,
         )
@@ -1532,6 +1609,9 @@ class SessionsResourceWithStreamingResponse:
         self.create_session = to_streamed_response_wrapper(
             sessions.create_session,
         )
+        self.edit_node_output = to_streamed_response_wrapper(
+            sessions.edit_node_output,
+        )
         self.finalize_dag = to_streamed_response_wrapper(
             sessions.finalize_dag,
         )
@@ -1590,6 +1670,9 @@ class AsyncSessionsResourceWithStreamingResponse:
         self.create_session = async_to_streamed_response_wrapper(
             sessions.create_session,
         )
+        self.edit_node_output = async_to_streamed_response_wrapper(
+            sessions.edit_node_output,
+        )
         self.finalize_dag = async_to_streamed_response_wrapper(
             sessions.finalize_dag,
         )

structify/resources/slack.py

@@ -80,12 +80,12 @@ class SlackResource(SyncAPIResource):
         self,
         *,
         event: slack_events_params.Variant1Event,
+        event_id: str,
         team_id: str,
         type: Literal["event_callback"],
         api_app_id: Optional[str] | Omit = omit,
         authed_users: Optional[SequenceNotStr[str]] | Omit = omit,
         event_context: Optional[str] | Omit = omit,
-        event_id: Optional[str] | Omit = omit,
         event_time: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -111,7 +111,7 @@ class SlackResource(SyncAPIResource):
         """
         ...
 
-    @required_args(["challenge", "type"], ["event", "team_id", "type"])
+    @required_args(["challenge", "type"], ["event", "event_id", "team_id", "type"])
     def events(
         self,
         *,
@@ -119,11 +119,11 @@ class SlackResource(SyncAPIResource):
         type: Literal["url_verification"] | Literal["event_callback"],
         token: Optional[str] | Omit = omit,
         event: slack_events_params.Variant1Event | Omit = omit,
+        event_id: str | Omit = omit,
         team_id: str | Omit = omit,
         api_app_id: Optional[str] | Omit = omit,
         authed_users: Optional[SequenceNotStr[str]] | Omit = omit,
         event_context: Optional[str] | Omit = omit,
-        event_id: Optional[str] | Omit = omit,
         event_time: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -142,11 +142,11 @@ class SlackResource(SyncAPIResource):
                         "type": type,
                         "token": token,
                         "event": event,
+                        "event_id": event_id,
                         "team_id": team_id,
                         "api_app_id": api_app_id,
                         "authed_users": authed_users,
                         "event_context": event_context,
-                        "event_id": event_id,
                         "event_time": event_time,
                     },
                     slack_events_params.SlackEventsParams,
@@ -215,12 +215,12 @@ class AsyncSlackResource(AsyncAPIResource):
         self,
         *,
         event: slack_events_params.Variant1Event,
+        event_id: str,
         team_id: str,
         type: Literal["event_callback"],
         api_app_id: Optional[str] | Omit = omit,
         authed_users: Optional[SequenceNotStr[str]] | Omit = omit,
         event_context: Optional[str] | Omit = omit,
-        event_id: Optional[str] | Omit = omit,
         event_time: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -246,7 +246,7 @@ class AsyncSlackResource(AsyncAPIResource):
         """
         ...
 
-    @required_args(["challenge", "type"], ["event", "team_id", "type"])
+    @required_args(["challenge", "type"], ["event", "event_id", "team_id", "type"])
     async def events(
         self,
         *,
@@ -254,11 +254,11 @@ class AsyncSlackResource(AsyncAPIResource):
         type: Literal["url_verification"] | Literal["event_callback"],
         token: Optional[str] | Omit = omit,
         event: slack_events_params.Variant1Event | Omit = omit,
+        event_id: str | Omit = omit,
         team_id: str | Omit = omit,
         api_app_id: Optional[str] | Omit = omit,
         authed_users: Optional[SequenceNotStr[str]] | Omit = omit,
         event_context: Optional[str] | Omit = omit,
-        event_id: Optional[str] | Omit = omit,
         event_time: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -277,11 +277,11 @@ class AsyncSlackResource(AsyncAPIResource):
                         "type": type,
                         "token": token,
                         "event": event,
+                        "event_id": event_id,
                         "team_id": team_id,
                         "api_app_id": api_app_id,
                         "authed_users": authed_users,
                         "event_context": event_context,
-                        "event_id": event_id,
                         "event_time": event_time,
                     },
                     slack_events_params.SlackEventsParams,

structify/resources/teams.py

@@ -15,7 +15,6 @@ from ..types import (
     team_add_member_params,
     team_credits_usage_params,
     team_create_project_params,
-    team_create_link_code_params,
     team_accept_invitation_params,
     team_update_member_role_params,
 )
@@ -44,7 +43,6 @@ from ..types.list_members_response import ListMembersResponse
 from ..types.credits_usage_response import CreditsUsageResponse
 from ..types.list_projects_response import ListProjectsResponse
 from ..types.remove_member_response import RemoveMemberResponse
-from ..types.teams_link_code_response import TeamsLinkCodeResponse
 from ..types.accept_invitation_response import AcceptInvitationResponse
 from ..types.invitation_details_response import InvitationDetailsResponse
 from ..types.update_member_role_response import UpdateMemberRoleResponse
@@ -120,6 +118,9 @@ class TeamsResource(SyncAPIResource):
         slack_team_icon: Optional[str] | Omit = omit,
         slack_team_id: Optional[str] | Omit = omit,
         slack_team_name: Optional[str] | Omit = omit,
+        teams_app_id: Optional[str] | Omit = omit,
+        teams_app_password: Optional[str] | Omit = omit,
+        teams_tenant_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -150,6 +151,9 @@ class TeamsResource(SyncAPIResource):
                     "slack_team_icon": slack_team_icon,
                     "slack_team_id": slack_team_id,
                     "slack_team_name": slack_team_name,
+                    "teams_app_id": teams_app_id,
+                    "teams_app_password": teams_app_password,
+                    "teams_tenant_id": teams_tenant_id,
                 },
                 team_update_params.TeamUpdateParams,
             ),
@@ -278,36 +282,6 @@ class TeamsResource(SyncAPIResource):
             cast_to=AddMemberResponse,
         )
 
-    def create_link_code(
-        self,
-        *,
-        team_id: str,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamsLinkCodeResponse:
-        """
-        Args:
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return self._post(
-            "/teams/link-code",
-            body=maybe_transform({"team_id": team_id}, team_create_link_code_params.TeamCreateLinkCodeParams),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=TeamsLinkCodeResponse,
-        )
-
     def create_project(
         self,
         team_id: str,
@@ -697,6 +671,9 @@ class AsyncTeamsResource(AsyncAPIResource):
         slack_team_icon: Optional[str] | Omit = omit,
         slack_team_id: Optional[str] | Omit = omit,
         slack_team_name: Optional[str] | Omit = omit,
+        teams_app_id: Optional[str] | Omit = omit,
+        teams_app_password: Optional[str] | Omit = omit,
+        teams_tenant_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -727,6 +704,9 @@ class AsyncTeamsResource(AsyncAPIResource):
                     "slack_team_icon": slack_team_icon,
                     "slack_team_id": slack_team_id,
                     "slack_team_name": slack_team_name,
+                    "teams_app_id": teams_app_id,
+                    "teams_app_password": teams_app_password,
+                    "teams_tenant_id": teams_tenant_id,
                 },
                 team_update_params.TeamUpdateParams,
             ),
@@ -857,38 +837,6 @@ class AsyncTeamsResource(AsyncAPIResource):
             cast_to=AddMemberResponse,
         )
 
-    async def create_link_code(
-        self,
-        *,
-        team_id: str,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamsLinkCodeResponse:
-        """
-        Args:
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return await self._post(
-            "/teams/link-code",
-            body=await async_maybe_transform(
-                {"team_id": team_id}, team_create_link_code_params.TeamCreateLinkCodeParams
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=TeamsLinkCodeResponse,
-        )
-
     async def create_project(
         self,
         team_id: str,
@@ -1232,9 +1180,6 @@ class TeamsResourceWithRawResponse:
         self.add_member = to_raw_response_wrapper(
             teams.add_member,
         )
-        self.create_link_code = to_raw_response_wrapper(
-            teams.create_link_code,
-        )
         self.create_project = to_raw_response_wrapper(
             teams.create_project,
         )
@@ -1286,9 +1231,6 @@ class AsyncTeamsResourceWithRawResponse:
         self.add_member = async_to_raw_response_wrapper(
             teams.add_member,
         )
-        self.create_link_code = async_to_raw_response_wrapper(
-            teams.create_link_code,
-        )
         self.create_project = async_to_raw_response_wrapper(
             teams.create_project,
         )
@@ -1340,9 +1282,6 @@ class TeamsResourceWithStreamingResponse:
         self.add_member = to_streamed_response_wrapper(
             teams.add_member,
         )
-        self.create_link_code = to_streamed_response_wrapper(
-            teams.create_link_code,
-        )
         self.create_project = to_streamed_response_wrapper(
             teams.create_project,
         )
@@ -1394,9 +1333,6 @@ class AsyncTeamsResourceWithStreamingResponse:
         self.add_member = async_to_streamed_response_wrapper(
             teams.add_member,
         )
-        self.create_link_code = async_to_streamed_response_wrapper(
-            teams.create_link_code,
-        )
         self.create_project = async_to_streamed_response_wrapper(
             teams.create_project,
         )

structify/resources/wiki.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Optional
+from typing import Optional
 
 import httpx
 
@@ -18,7 +18,7 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
-from ..types.team_wiki_page import TeamWikiPage
+from ..types.wiki_page import WikiPage
 from ..types.wiki_list_response import WikiListResponse
 from ..types.wiki_page_with_references import WikiPageWithReferences
 
@@ -49,7 +49,7 @@ class WikiResource(SyncAPIResource):
         self,
         team_id: str,
         *,
-        content: Dict[str, object],
+        markdown: str,
         slug: str,
         title: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -58,7 +58,7 @@ class WikiResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamWikiPage:
+    ) -> WikiPage:
         """
         Args:
           extra_headers: Send extra headers
@@ -75,7 +75,7 @@ class WikiResource(SyncAPIResource):
             f"/team/{team_id}/wiki",
             body=maybe_transform(
                 {
-                    "content": content,
+                    "markdown": markdown,
                     "slug": slug,
                     "title": title,
                 },
@@ -84,7 +84,7 @@ class WikiResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TeamWikiPage,
+            cast_to=WikiPage,
         )
 
     def update(
@@ -92,7 +92,8 @@ class WikiResource(SyncAPIResource):
         slug: str,
         *,
         team_id: str,
-        content: Dict[str, object],
+        markdown: str,
+        base_version: Optional[int] | Omit = omit,
         title: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -100,7 +101,7 @@ class WikiResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamWikiPage:
+    ) -> WikiPage:
         """
         Args:
           extra_headers: Send extra headers
@@ -119,7 +120,8 @@ class WikiResource(SyncAPIResource):
             f"/team/{team_id}/wiki/{slug}",
             body=maybe_transform(
                 {
-                    "content": content,
+                    "markdown": markdown,
+                    "base_version": base_version,
                     "title": title,
                 },
                 wiki_update_params.WikiUpdateParams,
@@ -127,7 +129,7 @@ class WikiResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TeamWikiPage,
+            cast_to=WikiPage,
         )
 
     def list(
@@ -255,7 +257,7 @@ class AsyncWikiResource(AsyncAPIResource):
         self,
         team_id: str,
         *,
-        content: Dict[str, object],
+        markdown: str,
         slug: str,
         title: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -264,7 +266,7 @@ class AsyncWikiResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamWikiPage:
+    ) -> WikiPage:
         """
         Args:
           extra_headers: Send extra headers
@@ -281,7 +283,7 @@ class AsyncWikiResource(AsyncAPIResource):
             f"/team/{team_id}/wiki",
             body=await async_maybe_transform(
                 {
-                    "content": content,
+                    "markdown": markdown,
                     "slug": slug,
                     "title": title,
                 },
@@ -290,7 +292,7 @@ class AsyncWikiResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TeamWikiPage,
+            cast_to=WikiPage,
         )
 
     async def update(
@@ -298,7 +300,8 @@ class AsyncWikiResource(AsyncAPIResource):
         slug: str,
         *,
         team_id: str,
-        content: Dict[str, object],
+        markdown: str,
+        base_version: Optional[int] | Omit = omit,
         title: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -306,7 +309,7 @@ class AsyncWikiResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> TeamWikiPage:
+    ) -> WikiPage:
         """
         Args:
           extra_headers: Send extra headers
@@ -325,7 +328,8 @@ class AsyncWikiResource(AsyncAPIResource):
             f"/team/{team_id}/wiki/{slug}",
             body=await async_maybe_transform(
                 {
-                    "content": content,
+                    "markdown": markdown,
+                    "base_version": base_version,
                     "title": title,
                 },
                 wiki_update_params.WikiUpdateParams,
@@ -333,7 +337,7 @@ class AsyncWikiResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TeamWikiPage,
+            cast_to=WikiPage,
         )
 
     async def list(

structify/resources/workflow.py

@@ -2,10 +2,12 @@
 
 from __future__ import annotations
 
+from typing import Optional
+
 import httpx
 
 from ..types import workflow_run_params, workflow_stop_params
-from .._types import Body, Query, Headers, NoneType, NotGiven, not_given
+from .._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -45,6 +47,7 @@ class WorkflowResource(SyncAPIResource):
         *,
         chat_session_id: str,
         use_node_cache: bool,
+        edited_node_name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -69,6 +72,7 @@ class WorkflowResource(SyncAPIResource):
                 {
                     "chat_session_id": chat_session_id,
                     "use_node_cache": use_node_cache,
+                    "edited_node_name": edited_node_name,
                 },
                 workflow_run_params.WorkflowRunParams,
             ),
@@ -135,6 +139,7 @@ class AsyncWorkflowResource(AsyncAPIResource):
         *,
         chat_session_id: str,
         use_node_cache: bool,
+        edited_node_name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -159,6 +164,7 @@ class AsyncWorkflowResource(AsyncAPIResource):
                 {
                     "chat_session_id": chat_session_id,
                     "use_node_cache": use_node_cache,
+                    "edited_node_name": edited_node_name,
                 },
                 workflow_run_params.WorkflowRunParams,
             ),