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

graphlit_api/create_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,
 )
@@ -40,6 +42,7 @@ class CreateWorkflowCreateWorkflow(BaseModel):
     indexing: Optional["CreateWorkflowCreateWorkflowIndexing"]
     preparation: Optional["CreateWorkflowCreateWorkflowPreparation"]
     extraction: Optional["CreateWorkflowCreateWorkflowExtraction"]
+    classification: Optional["CreateWorkflowCreateWorkflowClassification"]
     enrichment: Optional["CreateWorkflowCreateWorkflowEnrichment"]
     storage: Optional["CreateWorkflowCreateWorkflowStorage"]
     actions: Optional[List[Optional["CreateWorkflowCreateWorkflowActions"]]]
@@ -132,6 +135,7 @@ class CreateWorkflowCreateWorkflowPreparationJobsConnector(BaseModel):
     model_document: Optional[
         "CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocument"
     ] = Field(alias="modelDocument")
+    mistral: Optional["CreateWorkflowCreateWorkflowPreparationJobsConnectorMistral"]
 
 
 class CreateWorkflowCreateWorkflowPreparationJobsConnectorAzureDocument(BaseModel):
@@ -179,6 +183,10 @@ class CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocumentSpecifica
     id: str
 
 
+class CreateWorkflowCreateWorkflowPreparationJobsConnectorMistral(BaseModel):
+    key: Optional[str]
+
+
 class CreateWorkflowCreateWorkflowExtraction(BaseModel):
     jobs: Optional[List[Optional["CreateWorkflowCreateWorkflowExtractionJobs"]]]
 
@@ -240,6 +248,63 @@ class CreateWorkflowCreateWorkflowExtractionJobsConnectorModelTextSpecification(
     id: str
 
 
+class CreateWorkflowCreateWorkflowClassification(BaseModel):
+    jobs: Optional[List[Optional["CreateWorkflowCreateWorkflowClassificationJobs"]]]
+
+
+class CreateWorkflowCreateWorkflowClassificationJobs(BaseModel):
+    connector: Optional["CreateWorkflowCreateWorkflowClassificationJobsConnector"]
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnector(BaseModel):
+    type: ContentClassificationServiceTypes
+    content_type: Optional[ContentTypes] = Field(alias="contentType")
+    file_type: Optional[FileTypes] = Field(alias="fileType")
+    model: Optional["CreateWorkflowCreateWorkflowClassificationJobsConnectorModel"]
+    regex: Optional["CreateWorkflowCreateWorkflowClassificationJobsConnectorRegex"]
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnectorModel(BaseModel):
+    specification: Optional[
+        "CreateWorkflowCreateWorkflowClassificationJobsConnectorModelSpecification"
+    ]
+    rules: Optional[
+        List[
+            Optional[
+                "CreateWorkflowCreateWorkflowClassificationJobsConnectorModelRules"
+            ]
+        ]
+    ]
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnectorModelSpecification(
+    BaseModel
+):
+    id: str
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnectorModelRules(BaseModel):
+    then: Optional[str]
+    if_: Optional[str] = Field(alias="if")
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnectorRegex(BaseModel):
+    rules: Optional[
+        List[
+            Optional[
+                "CreateWorkflowCreateWorkflowClassificationJobsConnectorRegexRules"
+            ]
+        ]
+    ]
+
+
+class CreateWorkflowCreateWorkflowClassificationJobsConnectorRegexRules(BaseModel):
+    then: Optional[str]
+    type: Optional[RegexSourceTypes]
+    path: Optional[str]
+    matches: Optional[str]
+
+
 class CreateWorkflowCreateWorkflowEnrichment(BaseModel):
     link: Optional["CreateWorkflowCreateWorkflowEnrichmentLink"]
     jobs: Optional[List[Optional["CreateWorkflowCreateWorkflowEnrichmentJobs"]]]
@@ -297,6 +362,7 @@ class CreateWorkflowCreateWorkflowActionsConnector(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["CreateWorkflowCreateWorkflowActionsConnectorSlack"]
+    email: Optional["CreateWorkflowCreateWorkflowActionsConnectorEmail"]
 
 
 class CreateWorkflowCreateWorkflowActionsConnectorSlack(BaseModel):
@@ -304,6 +370,12 @@ class CreateWorkflowCreateWorkflowActionsConnectorSlack(BaseModel):
     channel: str
 
 
+class CreateWorkflowCreateWorkflowActionsConnectorEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 CreateWorkflow.model_rebuild()
 CreateWorkflowCreateWorkflow.model_rebuild()
 CreateWorkflowCreateWorkflowIngestion.model_rebuild()
@@ -320,6 +392,11 @@ CreateWorkflowCreateWorkflowExtractionJobs.model_rebuild()
 CreateWorkflowCreateWorkflowExtractionJobsConnector.model_rebuild()
 CreateWorkflowCreateWorkflowExtractionJobsConnectorModelImage.model_rebuild()
 CreateWorkflowCreateWorkflowExtractionJobsConnectorModelText.model_rebuild()
+CreateWorkflowCreateWorkflowClassification.model_rebuild()
+CreateWorkflowCreateWorkflowClassificationJobs.model_rebuild()
+CreateWorkflowCreateWorkflowClassificationJobsConnector.model_rebuild()
+CreateWorkflowCreateWorkflowClassificationJobsConnectorModel.model_rebuild()
+CreateWorkflowCreateWorkflowClassificationJobsConnectorRegex.model_rebuild()
 CreateWorkflowCreateWorkflowEnrichment.model_rebuild()
 CreateWorkflowCreateWorkflowEnrichmentJobs.model_rebuild()
 CreateWorkflowCreateWorkflowEnrichmentJobsConnector.model_rebuild()

graphlit_api/enums.py

@@ -28,6 +28,8 @@ class OpenAIModels(str, Enum):
     GPT4O_MINI_128K_20240718 = "GPT4O_MINI_128K_20240718"
     GPT4O_MINI_128K = "GPT4O_MINI_128K"
     GPT4O_CHAT_128K = "GPT4O_CHAT_128K"
+    GPT45_PREVIEW_128K = "GPT45_PREVIEW_128K"
+    GPT45_PREVIEW_128K_20250227 = "GPT45_PREVIEW_128K_20250227"
     O1_MINI_128K = "O1_MINI_128K"
     O1_MINI_128K_20240912 = "O1_MINI_128K_20240912"
     O1_PREVIEW_128K = "O1_PREVIEW_128K"
@@ -82,6 +84,7 @@ class EntityState(str, Enum):
     SANITIZED = "SANITIZED"
     EXTRACTED = "EXTRACTED"
     ENRICHED = "ENRICHED"
+    CLASSIFIED = "CLASSIFIED"
     CHANGED = "CHANGED"
     ARCHIVED = "ARCHIVED"
     APPROVED = "APPROVED"
@@ -259,6 +262,7 @@ class SearchQueryTypes(str, Enum):
 class IntegrationServiceTypes(str, Enum):
     SLACK = "SLACK"
     WEB_HOOK = "WEB_HOOK"
+    EMAIL = "EMAIL"
 
 
 class LinkTypes(str, Enum):
@@ -339,6 +343,7 @@ class SpecificationTypes(str, Enum):
     TEXT_EMBEDDING = "TEXT_EMBEDDING"
     IMAGE_EMBEDDING = "IMAGE_EMBEDDING"
     EXTRACTION = "EXTRACTION"
+    CLASSIFICATION = "CLASSIFICATION"
     PREPARATION = "PREPARATION"
 
 
@@ -496,6 +501,8 @@ class ElevenLabsModels(str, Enum):
     MULTILINGUAL_V1 = "MULTILINGUAL_V1"
     MULTILINGUAL_V2 = "MULTILINGUAL_V2"
     ENGLISH_V1 = "ENGLISH_V1"
+    FLASH_V2 = "FLASH_V2"
+    FLASH_V2_5 = "FLASH_V2_5"
     TURBO_V2 = "TURBO_V2"
     TURBO_V2_5 = "TURBO_V2_5"
 
@@ -768,6 +775,12 @@ class FilePreparationServiceTypes(str, Enum):
     EMAIL = "EMAIL"
     MODEL_DOCUMENT = "MODEL_DOCUMENT"
     ASSEMBLY_AI = "ASSEMBLY_AI"
+    MISTRAL_DOCUMENT = "MISTRAL_DOCUMENT"
+
+
+class RegexSourceTypes(str, Enum):
+    MARKDOWN = "MARKDOWN"
+    METADATA = "METADATA"
 
 
 class FacetValueTypes(str, Enum):
@@ -797,6 +810,8 @@ class CohereModels(str, Enum):
     COMMAND_R_PLUS_202404 = "COMMAND_R_PLUS_202404"
     COMMAND_R_PLUS_202408 = "COMMAND_R_PLUS_202408"
     COMMAND_R7_B_202412 = "COMMAND_R7_B_202412"
+    COMMAND_A = "COMMAND_A"
+    COMMAND_A_202503 = "COMMAND_A_202503"
     CUSTOM = "CUSTOM"
 
 
@@ -998,6 +1013,8 @@ class TextRoles(str, Enum):
     TABLE_CAPTION = "TABLE_CAPTION"
     TABLE_HEADER = "TABLE_HEADER"
     TABLE = "TABLE"
+    IMAGE = "IMAGE"
+    IMAGE_CAPTION = "IMAGE_CAPTION"
     FIGURE = "FIGURE"
     FIGURE_CAPTION = "FIGURE_CAPTION"
     BUTTON = "BUTTON"
@@ -1025,6 +1042,11 @@ class OrderByTypes(str, Enum):
     RELEVANCE = "RELEVANCE"
 
 
+class ContentClassificationServiceTypes(str, Enum):
+    REGEX = "REGEX"
+    MODEL = "MODEL"
+
+
 class AuthenticationServiceTypes(str, Enum):
     AUTH0 = "AUTH0"
     MICROSOFT_GRAPH = "MICROSOFT_GRAPH"

graphlit_api/get_alert.py

@@ -180,6 +180,7 @@ class GetAlertAlertIntegration(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["GetAlertAlertIntegrationSlack"]
+    email: Optional["GetAlertAlertIntegrationEmail"]
 
 
 class GetAlertAlertIntegrationSlack(BaseModel):
@@ -187,6 +188,12 @@ class GetAlertAlertIntegrationSlack(BaseModel):
     channel: str
 
 
+class GetAlertAlertIntegrationEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 class GetAlertAlertPublishing(BaseModel):
     type: ContentPublishingServiceTypes
     eleven_labs: Optional["GetAlertAlertPublishingElevenLabs"] = Field(

graphlit_api/get_content.py

@@ -305,9 +305,20 @@ class GetContentContentPages(BaseModel):
     index: Optional[int]
     text: Optional[str]
     relevance: Optional[float]
+    images: Optional[List[Optional["GetContentContentPagesImages"]]]
     chunks: Optional[List[Optional["GetContentContentPagesChunks"]]]
 
 
+class GetContentContentPagesImages(BaseModel):
+    id: Optional[str]
+    mime_type: Optional[str] = Field(alias="mimeType")
+    data: Optional[str]
+    left: Optional[int]
+    right: Optional[int]
+    top: Optional[int]
+    bottom: Optional[int]
+
+
 class GetContentContentPagesChunks(BaseModel):
     index: Optional[int]
     page_index: Optional[int] = Field(alias="pageIndex")

graphlit_api/get_user.py

@@ -65,6 +65,7 @@ class GetUserUserConnectorsIntegration(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["GetUserUserConnectorsIntegrationSlack"]
+    email: Optional["GetUserUserConnectorsIntegrationEmail"]
 
 
 class GetUserUserConnectorsIntegrationSlack(BaseModel):
@@ -72,6 +73,12 @@ class GetUserUserConnectorsIntegrationSlack(BaseModel):
     channel: str
 
 
+class GetUserUserConnectorsIntegrationEmail(BaseModel):
+    from_: str = Field(alias="from")
+    subject: str
+    to: List[str]
+
+
 GetUser.model_rebuild()
 GetUserUser.model_rebuild()
 GetUserUserConnectors.model_rebuild()

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"]]]
@@ -131,6 +134,7 @@ class GetWorkflowWorkflowPreparationJobsConnector(BaseModel):
     model_document: Optional[
         "GetWorkflowWorkflowPreparationJobsConnectorModelDocument"
     ] = Field(alias="modelDocument")
+    mistral: Optional["GetWorkflowWorkflowPreparationJobsConnectorMistral"]
 
 
 class GetWorkflowWorkflowPreparationJobsConnectorAzureDocument(BaseModel):
@@ -176,6 +180,10 @@ class GetWorkflowWorkflowPreparationJobsConnectorModelDocumentSpecification(Base
     id: str
 
 
+class GetWorkflowWorkflowPreparationJobsConnectorMistral(BaseModel):
+    key: Optional[str]
+
+
 class GetWorkflowWorkflowExtraction(BaseModel):
     jobs: Optional[List[Optional["GetWorkflowWorkflowExtractionJobs"]]]
 
@@ -233,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"]]]
@@ -290,6 +345,7 @@ class GetWorkflowWorkflowActionsConnector(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["GetWorkflowWorkflowActionsConnectorSlack"]
+    email: Optional["GetWorkflowWorkflowActionsConnectorEmail"]
 
 
 class GetWorkflowWorkflowActionsConnectorSlack(BaseModel):
@@ -297,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()
@@ -313,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)
@@ -492,6 +498,10 @@ class MedicalDrugFacetInput(BaseModel):
     facet: Optional[MedicalDrugFacetTypes] = None
 
 
+class MistralDocumentPreparationPropertiesInput(BaseModel):
+    key: Optional[str] = None
+
+
 class DocumentPreparationPropertiesInput(BaseModel):
     include_images: Optional[bool] = Field(alias="includeImages", default=None)
 
@@ -762,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(
@@ -982,6 +996,7 @@ class GoogleDriveFeedPropertiesInput(BaseModel):
 class IntegrationConnectorUpdateInput(BaseModel):
     uri: Optional[str] = None
     slack: Optional["SlackIntegrationPropertiesInput"] = None
+    email: Optional["EmailIntegrationPropertiesInput"] = None
 
 
 class MedicalDeviceFilter(BaseModel):
@@ -1047,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")
@@ -1122,6 +1142,10 @@ class ObservationInput(BaseModel):
     occurrences: List["ObservationOccurrenceInput"]
 
 
+class LinearProjectsInput(BaseModel):
+    key: str
+
+
 class ContentGraphInput(BaseModel):
     types: Optional[List[Optional[ObservableTypes]]] = None
 
@@ -1147,6 +1171,7 @@ class IntegrationConnectorInput(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str] = None
     slack: Optional["SlackIntegrationPropertiesInput"] = None
+    email: Optional["EmailIntegrationPropertiesInput"] = None
 
 
 class SummarizationStrategyInput(BaseModel):
@@ -1258,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
@@ -1456,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
@@ -1656,6 +1692,7 @@ class FilePreparationConnectorInput(BaseModel):
     assembly_ai: Optional["AssemblyAIAudioPreparationPropertiesInput"] = Field(
         alias="assemblyAI", default=None
     )
+    mistral: Optional["MistralDocumentPreparationPropertiesInput"] = None
     model_document: Optional["ModelDocumentPreparationPropertiesInput"] = Field(
         alias="modelDocument", default=None
     )
@@ -2153,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)
@@ -2169,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
@@ -2614,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
@@ -2658,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
 
@@ -2867,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
 
@@ -3450,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()
@@ -3463,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()
@@ -3494,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()
@@ -3522,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")