graphlit-client 1.0.20250306001__py3-none-any.whl → 1.0.20250313001__py3-none-any.whl

graphlit_api/get_workflow.py

@@ -10,6 +10,7 @@ from .enums import (
     AssemblyAIModels,
     AzureDocumentIntelligenceModels,
     AzureDocumentIntelligenceVersions,
+    ContentClassificationServiceTypes,
     ContentIndexingServiceTypes,
     ContentTypes,
     DeepgramModels,
@@ -21,6 +22,7 @@ from .enums import (
     IntegrationServiceTypes,
     LinkTypes,
     ObservableTypes,
+    RegexSourceTypes,
     StoragePolicyTypes,
     SummarizationTypes,
 )
@@ -41,6 +43,7 @@ class GetWorkflowWorkflow(BaseModel):
     indexing: Optional["GetWorkflowWorkflowIndexing"]
     preparation: Optional["GetWorkflowWorkflowPreparation"]
     extraction: Optional["GetWorkflowWorkflowExtraction"]
+    classification: Optional["GetWorkflowWorkflowClassification"]
     enrichment: Optional["GetWorkflowWorkflowEnrichment"]
     storage: Optional["GetWorkflowWorkflowStorage"]
     actions: Optional[List[Optional["GetWorkflowWorkflowActions"]]]
@@ -238,6 +241,53 @@ class GetWorkflowWorkflowExtractionJobsConnectorModelTextSpecification(BaseModel
     id: str
 
 
+class GetWorkflowWorkflowClassification(BaseModel):
+    jobs: Optional[List[Optional["GetWorkflowWorkflowClassificationJobs"]]]
+
+
+class GetWorkflowWorkflowClassificationJobs(BaseModel):
+    connector: Optional["GetWorkflowWorkflowClassificationJobsConnector"]
+
+
+class GetWorkflowWorkflowClassificationJobsConnector(BaseModel):
+    type: ContentClassificationServiceTypes
+    content_type: Optional[ContentTypes] = Field(alias="contentType")
+    file_type: Optional[FileTypes] = Field(alias="fileType")
+    model: Optional["GetWorkflowWorkflowClassificationJobsConnectorModel"]
+    regex: Optional["GetWorkflowWorkflowClassificationJobsConnectorRegex"]
+
+
+class GetWorkflowWorkflowClassificationJobsConnectorModel(BaseModel):
+    specification: Optional[
+        "GetWorkflowWorkflowClassificationJobsConnectorModelSpecification"
+    ]
+    rules: Optional[
+        List[Optional["GetWorkflowWorkflowClassificationJobsConnectorModelRules"]]
+    ]
+
+
+class GetWorkflowWorkflowClassificationJobsConnectorModelSpecification(BaseModel):
+    id: str
+
+
+class GetWorkflowWorkflowClassificationJobsConnectorModelRules(BaseModel):
+    then: Optional[str]
+    if_: Optional[str] = Field(alias="if")
+
+
+class GetWorkflowWorkflowClassificationJobsConnectorRegex(BaseModel):
+    rules: Optional[
+        List[Optional["GetWorkflowWorkflowClassificationJobsConnectorRegexRules"]]
+    ]
+
+
+class GetWorkflowWorkflowClassificationJobsConnectorRegexRules(BaseModel):
+    then: Optional[str]
+    type: Optional[RegexSourceTypes]
+    path: Optional[str]
+    matches: Optional[str]
+
+
 class GetWorkflowWorkflowEnrichment(BaseModel):
     link: Optional["GetWorkflowWorkflowEnrichmentLink"]
     jobs: Optional[List[Optional["GetWorkflowWorkflowEnrichmentJobs"]]]
@@ -295,6 +345,7 @@ class GetWorkflowWorkflowActionsConnector(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["GetWorkflowWorkflowActionsConnectorSlack"]
+    email: Optional["GetWorkflowWorkflowActionsConnectorEmail"]
 
 
 class GetWorkflowWorkflowActionsConnectorSlack(BaseModel):
@@ -302,6 +353,12 @@ class GetWorkflowWorkflowActionsConnectorSlack(BaseModel):
     channel: str
 
 
+class GetWorkflowWorkflowActionsConnectorEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 GetWorkflow.model_rebuild()
 GetWorkflowWorkflow.model_rebuild()
 GetWorkflowWorkflowIngestion.model_rebuild()
@@ -318,6 +375,11 @@ GetWorkflowWorkflowExtractionJobs.model_rebuild()
 GetWorkflowWorkflowExtractionJobsConnector.model_rebuild()
 GetWorkflowWorkflowExtractionJobsConnectorModelImage.model_rebuild()
 GetWorkflowWorkflowExtractionJobsConnectorModelText.model_rebuild()
+GetWorkflowWorkflowClassification.model_rebuild()
+GetWorkflowWorkflowClassificationJobs.model_rebuild()
+GetWorkflowWorkflowClassificationJobsConnector.model_rebuild()
+GetWorkflowWorkflowClassificationJobsConnectorModel.model_rebuild()
+GetWorkflowWorkflowClassificationJobsConnectorRegex.model_rebuild()
 GetWorkflowWorkflowEnrichment.model_rebuild()
 GetWorkflowWorkflowEnrichmentJobs.model_rebuild()
 GetWorkflowWorkflowEnrichmentJobsConnector.model_rebuild()

graphlit_api/input_types.py

@@ -19,6 +19,7 @@ from .enums import (
     CohereModels,
     CollectionTypes,
     ConnectorTypes,
+    ContentClassificationServiceTypes,
     ContentFacetTypes,
     ContentIndexingServiceTypes,
     ContentPublishingFormats,
@@ -84,6 +85,7 @@ from .enums import (
     PolicyTimeTypes,
     ProductFacetTypes,
     PromptStrategyTypes,
+    RegexSourceTypes,
     ReplicateModels,
     RepoFacetTypes,
     RerankingModelServiceTypes,
@@ -359,6 +361,10 @@ class MedicalTherapyFacetInput(BaseModel):
     facet: Optional[MedicalTherapyFacetTypes] = None
 
 
+class RegexContentClassificationPropertiesInput(BaseModel):
+    rules: Optional[List[Optional["RegexClassificationRuleInput"]]] = None
+
+
 class ProductFilter(BaseModel):
     search: Optional[str] = None
     order_by: Optional[OrderByTypes] = Field(alias="orderBy", default=None)
@@ -766,6 +772,10 @@ class H3IndexFilter(BaseModel):
     key: Optional[str] = None
 
 
+class ClassificationWorkflowJobInput(BaseModel):
+    connector: Optional["ContentClassificationConnectorInput"] = None
+
+
 class ModelFilter(BaseModel):
     types: Optional[List[Optional[ModelTypes]]] = None
     service_types: Optional[List[Optional[ModelServiceTypes]]] = Field(
@@ -986,6 +996,7 @@ class GoogleDriveFeedPropertiesInput(BaseModel):
 class IntegrationConnectorUpdateInput(BaseModel):
     uri: Optional[str] = None
     slack: Optional["SlackIntegrationPropertiesInput"] = None
+    email: Optional["EmailIntegrationPropertiesInput"] = None
 
 
 class MedicalDeviceFilter(BaseModel):
@@ -1051,6 +1062,11 @@ class AzureFileFeedPropertiesUpdateInput(BaseModel):
     prefix: Optional[str] = None
 
 
+class PromptClassificationRuleInput(BaseModel):
+    if_: Optional[str] = Field(alias="if", default=None)
+    then: Optional[str] = None
+
+
 class ProjectInput(BaseModel):
     name: str
     environment_type: EnvironmentTypes = Field(alias="environmentType")
@@ -1155,6 +1171,7 @@ class IntegrationConnectorInput(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str] = None
     slack: Optional["SlackIntegrationPropertiesInput"] = None
+    email: Optional["EmailIntegrationPropertiesInput"] = None
 
 
 class SummarizationStrategyInput(BaseModel):
@@ -1266,6 +1283,13 @@ class PointInput(BaseModel):
     distance: Optional[float] = None
 
 
+class RegexClassificationRuleInput(BaseModel):
+    type: Optional[RegexSourceTypes] = None
+    path: Optional[str] = None
+    matches: Optional[str] = None
+    then: Optional[str] = None
+
+
 class ContentPublishingConnectorInput(BaseModel):
     type: ContentPublishingServiceTypes
     format: ContentPublishingFormats
@@ -1464,6 +1488,10 @@ class EventFilter(BaseModel):
     events: Optional[List["EntityReferenceFilter"]] = None
 
 
+class ClassificationWorkflowStageInput(BaseModel):
+    jobs: Optional[List[Optional["ClassificationWorkflowJobInput"]]] = None
+
+
 class RepoInput(BaseModel):
     name: str
     uri: Optional[Any] = None
@@ -2162,6 +2190,11 @@ class SiteFeedPropertiesInput(BaseModel):
     read_limit: Optional[int] = Field(alias="readLimit", default=None)
 
 
+class ModelContentClassificationPropertiesInput(BaseModel):
+    specification: Optional["EntityReferenceInput"] = None
+    rules: Optional[List[Optional["PromptClassificationRuleInput"]]] = None
+
+
 class DropboxFeedPropertiesUpdateInput(BaseModel):
     path: Optional[str] = None
     app_key: Optional[str] = Field(alias="appKey", default=None)
@@ -2178,6 +2211,14 @@ class RepoFacetInput(BaseModel):
     facet: Optional[RepoFacetTypes] = None
 
 
+class ContentClassificationConnectorInput(BaseModel):
+    type: Optional[ContentClassificationServiceTypes] = None
+    content_type: Optional[ContentTypes] = Field(alias="contentType", default=None)
+    file_type: Optional[FileTypes] = Field(alias="fileType", default=None)
+    model: Optional["ModelContentClassificationPropertiesInput"] = None
+    regex: Optional["RegexContentClassificationPropertiesInput"] = None
+
+
 class EnrichmentWorkflowStageInput(BaseModel):
     link: Optional["LinkStrategyInput"] = None
     jobs: Optional[List[Optional["EnrichmentWorkflowJobInput"]]] = None
@@ -2623,6 +2664,12 @@ class RevisionStrategyInput(BaseModel):
     count: Optional[int] = None
 
 
+class EmailIntegrationPropertiesInput(BaseModel):
+    subject: str
+    from_: str = Field(alias="from")
+    to: List[str]
+
+
 class ConversationUpdateInput(BaseModel):
     id: str
     name: Optional[str] = None
@@ -2667,6 +2714,7 @@ class WorkflowUpdateInput(BaseModel):
     preparation: Optional["PreparationWorkflowStageInput"] = None
     extraction: Optional["ExtractionWorkflowStageInput"] = None
     enrichment: Optional["EnrichmentWorkflowStageInput"] = None
+    classification: Optional["ClassificationWorkflowStageInput"] = None
     storage: Optional["StorageWorkflowStageInput"] = None
     actions: Optional[List[Optional["WorkflowActionInput"]]] = None
 
@@ -2876,6 +2924,7 @@ class WorkflowInput(BaseModel):
     preparation: Optional["PreparationWorkflowStageInput"] = None
     extraction: Optional["ExtractionWorkflowStageInput"] = None
     enrichment: Optional["EnrichmentWorkflowStageInput"] = None
+    classification: Optional["ClassificationWorkflowStageInput"] = None
     storage: Optional["StorageWorkflowStageInput"] = None
     actions: Optional[List[Optional["WorkflowActionInput"]]] = None
 
@@ -3459,6 +3508,7 @@ StorageWorkflowStageInput.model_rebuild()
 WorkflowFilter.model_rebuild()
 MedicalIndicationInput.model_rebuild()
 UserFilter.model_rebuild()
+RegexContentClassificationPropertiesInput.model_rebuild()
 ProductFilter.model_rebuild()
 EnrichmentWorkflowJobInput.model_rebuild()
 ContentUpdateInput.model_rebuild()
@@ -3472,6 +3522,7 @@ EmbeddingsStrategyInput.model_rebuild()
 ConnectorFilter.model_rebuild()
 ProjectUpdateInput.model_rebuild()
 MedicalStudyFilter.model_rebuild()
+ClassificationWorkflowJobInput.model_rebuild()
 CollectionUpdateInput.model_rebuild()
 SpecificationInput.model_rebuild()
 PackageMetadataInput.model_rebuild()
@@ -3503,6 +3554,7 @@ AudioMetadataInput.model_rebuild()
 SpecificationUpdateInput.model_rebuild()
 ContentPublishingConnectorUpdateInput.model_rebuild()
 EventFilter.model_rebuild()
+ClassificationWorkflowStageInput.model_rebuild()
 ObservationReferenceFilter.model_rebuild()
 MedicalGuidelineFilter.model_rebuild()
 EmailFeedPropertiesInput.model_rebuild()
@@ -3531,6 +3583,8 @@ MedicalDrugUpdateInput.model_rebuild()
 PersonInput.model_rebuild()
 SoftwareFilter.model_rebuild()
 SiteFeedPropertiesInput.model_rebuild()
+ModelContentClassificationPropertiesInput.model_rebuild()
+ContentClassificationConnectorInput.model_rebuild()
 EnrichmentWorkflowStageInput.model_rebuild()
 PersonFilter.model_rebuild()
 FeedFilter.model_rebuild()

graphlit_api/lookup_credits.py

@@ -26,6 +26,7 @@ class LookupCreditsLookupCredits(BaseModel):
     indexing_ratio: Optional[Any] = Field(alias="indexingRatio")
     preparation_ratio: Optional[Any] = Field(alias="preparationRatio")
     extraction_ratio: Optional[Any] = Field(alias="extractionRatio")
+    classification_ratio: Optional[Any] = Field(alias="classificationRatio")
     enrichment_ratio: Optional[Any] = Field(alias="enrichmentRatio")
     publishing_ratio: Optional[Any] = Field(alias="publishingRatio")
     search_ratio: Optional[Any] = Field(alias="searchRatio")

graphlit_api/operations.py

@@ -252,6 +252,7 @@ __all__ = [
     "REVISE_TEXT_GQL",
     "SCREENSHOT_PAGE_GQL",
     "SEARCH_WEB_GQL",
+    "SEND_NOTIFICATION_GQL",
     "SUGGEST_CONVERSATION_GQL",
     "SUMMARIZE_CONTENTS_GQL",
     "SUMMARIZE_TEXT_GQL",
@@ -455,6 +456,11 @@ query GetAlert($id: ID!, $correlationId: String) {
         token
         channel
       }
+      email {
+        from
+        subject
+        to
+      }
     }
     publishing {
       type
@@ -575,6 +581,11 @@ query QueryAlerts($filter: AlertFilter, $correlationId: String) {
           token
           channel
         }
+        email {
+          from
+          subject
+          to
+        }
       }
       publishing {
         type
@@ -7470,6 +7481,14 @@ mutation UpdateMedicalTherapy($medicalTherapy: MedicalTherapyUpdateInput!) {
 }
 """
 
+SEND_NOTIFICATION_GQL = """
+mutation SendNotification($connector: IntegrationConnectorInput!, $text: String!, $textType: TextTypes) {
+  sendNotification(connector: $connector, text: $text, textType: $textType) {
+    result
+  }
+}
+"""
+
 CREATE_OBSERVATION_GQL = """
 mutation CreateObservation($observation: ObservationInput!) {
   createObservation(observation: $observation) {
@@ -8015,6 +8034,7 @@ query LookupCredits($correlationId: String!) {
     indexingRatio
     preparationRatio
     extractionRatio
+    classificationRatio
     enrichmentRatio
     publishingRatio
     searchRatio
@@ -8074,6 +8094,7 @@ query QueryCredits($startDate: DateTime!, $duration: TimeSpan!) {
     indexingRatio
     preparationRatio
     extractionRatio
+    classificationRatio
     enrichmentRatio
     publishingRatio
     searchRatio
@@ -9023,6 +9044,11 @@ query GetUser {
           token
           channel
         }
+        email {
+          from
+          subject
+          to
+        }
       }
     }
   }
@@ -9067,6 +9093,11 @@ query QueryUsers($filter: UserFilter, $correlationId: String) {
             token
             channel
           }
+          email {
+            from
+            subject
+            to
+          }
         }
       }
     }
@@ -9210,6 +9241,32 @@ mutation CreateWorkflow($workflow: WorkflowInput!) {
         }
       }
     }
+    classification {
+      jobs {
+        connector {
+          type
+          contentType
+          fileType
+          model {
+            specification {
+              id
+            }
+            rules {
+              then
+              if
+            }
+          }
+          regex {
+            rules {
+              then
+              type
+              path
+              matches
+            }
+          }
+        }
+      }
+    }
     enrichment {
       link {
         enableCrawling
@@ -9251,6 +9308,11 @@ mutation CreateWorkflow($workflow: WorkflowInput!) {
           token
           channel
         }
+        email {
+          from
+          subject
+          to
+        }
       }
     }
   }
@@ -9409,6 +9471,32 @@ query GetWorkflow($id: ID!, $correlationId: String) {
         }
       }
     }
+    classification {
+      jobs {
+        connector {
+          type
+          contentType
+          fileType
+          model {
+            specification {
+              id
+            }
+            rules {
+              then
+              if
+            }
+          }
+          regex {
+            rules {
+              then
+              type
+              path
+              matches
+            }
+          }
+        }
+      }
+    }
     enrichment {
       link {
         enableCrawling
@@ -9450,6 +9538,11 @@ query GetWorkflow($id: ID!, $correlationId: String) {
           token
           channel
         }
+        email {
+          from
+          subject
+          to
+        }
       }
     }
   }
@@ -9578,6 +9671,32 @@ query QueryWorkflows($filter: WorkflowFilter, $correlationId: String) {
           }
         }
       }
+      classification {
+        jobs {
+          connector {
+            type
+            contentType
+            fileType
+            model {
+              specification {
+                id
+              }
+              rules {
+                then
+                if
+              }
+            }
+            regex {
+              rules {
+                then
+                type
+                path
+                matches
+              }
+            }
+          }
+        }
+      }
       enrichment {
         link {
           enableCrawling
@@ -9619,6 +9738,11 @@ query QueryWorkflows($filter: WorkflowFilter, $correlationId: String) {
             token
             channel
           }
+          email {
+            from
+            subject
+            to
+          }
         }
       }
     }
@@ -9742,6 +9866,32 @@ mutation UpdateWorkflow($workflow: WorkflowUpdateInput!) {
         }
       }
     }
+    classification {
+      jobs {
+        connector {
+          type
+          contentType
+          fileType
+          model {
+            specification {
+              id
+            }
+            rules {
+              then
+              if
+            }
+          }
+          regex {
+            rules {
+              then
+              type
+              path
+              matches
+            }
+          }
+        }
+      }
+    }
     enrichment {
       link {
         enableCrawling
@@ -9783,6 +9933,11 @@ mutation UpdateWorkflow($workflow: WorkflowUpdateInput!) {
           token
           channel
         }
+        email {
+          from
+          subject
+          to
+        }
       }
     }
   }

graphlit_api/query_alerts.py

@@ -186,6 +186,7 @@ class QueryAlertsAlertsResultsIntegration(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["QueryAlertsAlertsResultsIntegrationSlack"]
+    email: Optional["QueryAlertsAlertsResultsIntegrationEmail"]
 
 
 class QueryAlertsAlertsResultsIntegrationSlack(BaseModel):
@@ -193,6 +194,12 @@ class QueryAlertsAlertsResultsIntegrationSlack(BaseModel):
     channel: str
 
 
+class QueryAlertsAlertsResultsIntegrationEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 class QueryAlertsAlertsResultsPublishing(BaseModel):
     type: ContentPublishingServiceTypes
     eleven_labs: Optional["QueryAlertsAlertsResultsPublishingElevenLabs"] = Field(

graphlit_api/query_credits.py

@@ -24,6 +24,7 @@ class QueryCreditsCredits(BaseModel):
     indexing_ratio: Optional[Any] = Field(alias="indexingRatio")
     preparation_ratio: Optional[Any] = Field(alias="preparationRatio")
     extraction_ratio: Optional[Any] = Field(alias="extractionRatio")
+    classification_ratio: Optional[Any] = Field(alias="classificationRatio")
     enrichment_ratio: Optional[Any] = Field(alias="enrichmentRatio")
     publishing_ratio: Optional[Any] = Field(alias="publishingRatio")
     search_ratio: Optional[Any] = Field(alias="searchRatio")

graphlit_api/query_users.py

@@ -69,6 +69,7 @@ class QueryUsersUsersResultsConnectorsIntegration(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["QueryUsersUsersResultsConnectorsIntegrationSlack"]
+    email: Optional["QueryUsersUsersResultsConnectorsIntegrationEmail"]
 
 
 class QueryUsersUsersResultsConnectorsIntegrationSlack(BaseModel):
@@ -76,6 +77,12 @@ class QueryUsersUsersResultsConnectorsIntegrationSlack(BaseModel):
     channel: str
 
 
+class QueryUsersUsersResultsConnectorsIntegrationEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 QueryUsers.model_rebuild()
 QueryUsersUsers.model_rebuild()
 QueryUsersUsersResults.model_rebuild()