fix: update GraphQL schema

openhexa/graphql/graphql_client/__init__.py

@@ -120,6 +120,7 @@
     CreateOrganizationError,
     CreatePipelineFromTemplateVersionError,
     CreatePipelineTemplateVersionError,
+    CreateSavedQueryError,
     CreateTeamError,
     CreateTemplateVersionPermissionReason,
     CreateWebappError,
@@ -145,6 +146,7 @@
     DeleteOrganizationInvitationError,
     DeleteOrganizationMemberError,
     DeletePipelineVersionError,
+    DeleteSavedQueryError,
     DeleteTeamError,
     DeleteTemplateVersionError,
     DeleteWebappError,
@@ -160,6 +162,7 @@
     EditWebappFileError,
     EnableTwoFactorError,
     ExecuteSQLError,
+    ExecuteSQLOrigin,
     FileEncoding,
     FileSampleStatus,
     FileType,
@@ -212,6 +215,7 @@
     ResendWorkspaceInvitationError,
     ResolveAssistantProposalError,
     RunDAGError,
+    SavedQueryOrderBy,
     SetDAGRunFavoriteError,
     SetMetadataAttributeError,
     SetPasswordError,
@@ -233,6 +237,7 @@
     UpdateOrganizationSubscriptionError,
     UpdatePipelineError,
     UpdatePipelineVersionError,
+    UpdateSavedQueryError,
     UpdateTeamError,
     UpdateTemplateError,
     UpdateTemplateVersionError,
@@ -298,6 +303,8 @@
     CreatePipelineRecipientInput,
     CreatePipelineTemplateVersionInput,
     CreatePipelineVersionInput,
+    CreateSavedQueryInput,
+    CreateSelfHostedOrganizationInput,
     CreateTeamInput,
     CreateWebappInput,
     CreateWorkspaceInput,
@@ -322,6 +329,7 @@
     DeletePipelineRecipientInput,
     DeletePipelineTemplateInput,
     DeletePipelineVersionInput,
+    DeleteSavedQueryInput,
     DeleteTeamInput,
     DeleteTemplateVersionInput,
     DeleteWebappInput,
@@ -397,6 +405,7 @@
     UpdatePipelineProgressInput,
     UpdatePipelineRecipientInput,
     UpdatePipelineVersionInput,
+    UpdateSavedQueryInput,
     UpdateTeamInput,
     UpdateTemplateInput,
     UpdateTemplateVersionInput,
@@ -601,6 +610,9 @@
     "CreatePipelineTemplateVersionError",
     "CreatePipelineTemplateVersionInput",
     "CreatePipelineVersionInput",
+    "CreateSavedQueryError",
+    "CreateSavedQueryInput",
+    "CreateSelfHostedOrganizationInput",
     "CreateTeamError",
     "CreateTeamInput",
     "CreateTemplateVersionPermissionReason",
@@ -683,6 +695,8 @@
     "DeletePipelineVersionDeletePipelineVersion",
     "DeletePipelineVersionError",
     "DeletePipelineVersionInput",
+    "DeleteSavedQueryError",
+    "DeleteSavedQueryInput",
     "DeleteTeamError",
     "DeleteTeamInput",
     "DeleteTemplateVersionError",
@@ -708,6 +722,7 @@
     "EnableTwoFactorError",
     "EnableTwoFactorInput",
     "ExecuteSQLError",
+    "ExecuteSQLOrigin",
     "FileEncoding",
     "FileSampleStatus",
     "FileType",
@@ -852,6 +867,7 @@
     "RunDAGError",
     "RunDAGInput",
     "RunPipelineInput",
+    "SavedQueryOrderBy",
     "SetDAGRunFavoriteError",
     "SetDAGRunFavoriteInput",
     "SetMetadataAttributeError",
@@ -913,6 +929,8 @@
     "UpdatePipelineRecipientInput",
     "UpdatePipelineVersionError",
     "UpdatePipelineVersionInput",
+    "UpdateSavedQueryError",
+    "UpdateSavedQueryInput",
     "UpdateTeamError",
     "UpdateTeamInput",
     "UpdateTemplateError",

openhexa/graphql/graphql_client/enums.py

@@ -110,6 +110,8 @@ class ArchiveWorkspaceError(str, Enum):
 class AssistantToolName(str, Enum):
     create_pipeline = "create_pipeline"
     get_dataset = "get_dataset"
+    get_db_schema = "get_db_schema"
+    get_db_table_schema = "get_db_table_schema"
     get_help_or_doc = "get_help_or_doc"
     get_pipeline = "get_pipeline"
     get_static_webapp_file = "get_static_webapp_file"
@@ -244,6 +246,11 @@ class CreatePipelineTemplateVersionError(str, Enum):
     WORKSPACE_NOT_FOUND = "WORKSPACE_NOT_FOUND"
 
 
+class CreateSavedQueryError(str, Enum):
+    PERMISSION_DENIED = "PERMISSION_DENIED"
+    WORKSPACE_NOT_FOUND = "WORKSPACE_NOT_FOUND"
+
+
 class CreateTeamError(str, Enum):
     NAME_DUPLICATE = "NAME_DUPLICATE"
     PERMISSION_DENIED = "PERMISSION_DENIED"
@@ -410,6 +417,11 @@ class DeletePipelineVersionError(str, Enum):
     PIPELINE_VERSION_NOT_FOUND = "PIPELINE_VERSION_NOT_FOUND"
 
 
+class DeleteSavedQueryError(str, Enum):
+    PERMISSION_DENIED = "PERMISSION_DENIED"
+    SAVED_QUERY_NOT_FOUND = "SAVED_QUERY_NOT_FOUND"
+
+
 class DeleteTeamError(str, Enum):
     NOT_FOUND = "NOT_FOUND"
     PERMISSION_DENIED = "PERMISSION_DENIED"
@@ -481,6 +493,11 @@ class ExecuteSQLError(str, Enum):
     QUERY_TIMEOUT = "QUERY_TIMEOUT"
 
 
+class ExecuteSQLOrigin(str, Enum):
+    DATA_STUDIO = "DATA_STUDIO"
+    OTHER = "OTHER"
+
+
 class FileEncoding(str, Enum):
     BASE64 = "BASE64"
     TEXT = "TEXT"
@@ -830,6 +847,13 @@ class RunDAGError(str, Enum):
     INVALID_CONFIG = "INVALID_CONFIG"
 
 
+class SavedQueryOrderBy(str, Enum):
+    NAME_ASC = "NAME_ASC"
+    NAME_DESC = "NAME_DESC"
+    UPDATED_AT_ASC = "UPDATED_AT_ASC"
+    UPDATED_AT_DESC = "UPDATED_AT_DESC"
+
+
 class SetDAGRunFavoriteError(str, Enum):
     INVALID = "INVALID"
     MISSING_LABEL = "MISSING_LABEL"
@@ -952,6 +976,11 @@ class UpdatePipelineVersionError(str, Enum):
     PERMISSION_DENIED = "PERMISSION_DENIED"
 
 
+class UpdateSavedQueryError(str, Enum):
+    PERMISSION_DENIED = "PERMISSION_DENIED"
+    SAVED_QUERY_NOT_FOUND = "SAVED_QUERY_NOT_FOUND"
+
+
 class UpdateTeamError(str, Enum):
     NAME_DUPLICATE = "NAME_DUPLICATE"
     NOT_FOUND = "NOT_FOUND"

openhexa/graphql/graphql_client/input_types.py

@@ -222,6 +222,18 @@ class CreatePipelineVersionInput(BaseModel):
     zipfile: Optional[str] = None
 
 
+class CreateSavedQueryInput(BaseModel):
+    content: str
+    description: Optional[str] = None
+    name: str
+    workspace_slug: str = Field(alias="workspaceSlug")
+
+
+class CreateSelfHostedOrganizationInput(BaseModel):
+    name: str
+    short_name: Optional[str] = Field(alias="shortName", default=None)
+
+
 class CreateTeamInput(BaseModel):
     name: str
 
@@ -244,7 +256,7 @@ class CreateWorkspaceInput(BaseModel):
     description: Optional[str] = None
     load_sample_data: Optional[bool] = Field(alias="loadSampleData", default=None)
     name: str
-    organization_id: Optional[Any] = Field(alias="organizationId", default=None)
+    organization_id: Any = Field(alias="organizationId")
     slug: Optional[str] = None
 
 
@@ -337,6 +349,10 @@ class DeletePipelineVersionInput(BaseModel):
     id: Any
 
 
+class DeleteSavedQueryInput(BaseModel):
+    id: str
+
+
 class DeleteTeamInput(BaseModel):
     id: Any
 
@@ -808,6 +824,13 @@ class UpdatePipelineVersionInput(BaseModel):
     name: Optional[str] = None
 
 
+class UpdateSavedQueryInput(BaseModel):
+    content: Optional[str] = None
+    description: Optional[str] = None
+    id: str
+    name: Optional[str] = None
+
+
 class UpdateTeamInput(BaseModel):
     id: Any
     name: Optional[str] = None