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

graphlit_api/client.py

@@ -224,6 +224,7 @@ from .input_types import (
     FeedFilter,
     FeedInput,
     FeedUpdateInput,
+    IntegrationConnectorInput,
     LabelFilter,
     LabelInput,
     LabelUpdateInput,
@@ -561,6 +562,7 @@ from .operations import (
     REVISE_TEXT_GQL,
     SCREENSHOT_PAGE_GQL,
     SEARCH_WEB_GQL,
+    SEND_NOTIFICATION_GQL,
     SUGGEST_CONVERSATION_GQL,
     SUMMARIZE_CONTENTS_GQL,
     SUMMARIZE_TEXT_GQL,
@@ -594,6 +596,10 @@ from .operations import (
     UPDATE_SPECIFICATION_GQL,
     UPDATE_USER_GQL,
     UPDATE_WORKFLOW_GQL,
+    UPSERT_CATEGORY_GQL,
+    UPSERT_LABEL_GQL,
+    UPSERT_SPECIFICATION_GQL,
+    UPSERT_WORKFLOW_GQL,
 )
 from .prompt import Prompt
 from .prompt_conversation import PromptConversation
@@ -649,6 +655,7 @@ from .revise_image import ReviseImage
 from .revise_text import ReviseText
 from .screenshot_page import ScreenshotPage
 from .search_web import SearchWeb
+from .send_notification import SendNotification
 from .suggest_conversation import SuggestConversation
 from .summarize_contents import SummarizeContents
 from .summarize_text import SummarizeText
@@ -682,6 +689,10 @@ from .update_software import UpdateSoftware
 from .update_specification import UpdateSpecification
 from .update_user import UpdateUser
 from .update_workflow import UpdateWorkflow
+from .upsert_category import UpsertCategory
+from .upsert_label import UpsertLabel
+from .upsert_specification import UpsertSpecification
+from .upsert_workflow import UpsertWorkflow
 
 
 def gql(q: str) -> str:
@@ -968,6 +979,19 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return UpdateCategory.model_validate(data)
 
+    async def upsert_category(
+        self, category: CategoryInput, **kwargs: Any
+    ) -> UpsertCategory:
+        variables: Dict[str, object] = {"category": category}
+        response = await self.execute(
+            query=UPSERT_CATEGORY_GQL,
+            operation_name="UpsertCategory",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return UpsertCategory.model_validate(data)
+
     async def add_contents_to_collections(
         self,
         contents: List[EntityReferenceInput],
@@ -2699,6 +2723,17 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return UpdateLabel.model_validate(data)
 
+    async def upsert_label(self, label: LabelInput, **kwargs: Any) -> UpsertLabel:
+        variables: Dict[str, object] = {"label": label}
+        response = await self.execute(
+            query=UPSERT_LABEL_GQL,
+            operation_name="UpsertLabel",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return UpsertLabel.model_validate(data)
+
     async def count_medical_conditions(
         self,
         filter: Union[Optional[MedicalConditionFilter], UnsetType] = UNSET,
@@ -4125,6 +4160,27 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return UpdateMedicalTherapy.model_validate(data)
 
+    async def send_notification(
+        self,
+        connector: IntegrationConnectorInput,
+        text: str,
+        text_type: Union[Optional[TextTypes], UnsetType] = UNSET,
+        **kwargs: Any
+    ) -> SendNotification:
+        variables: Dict[str, object] = {
+            "connector": connector,
+            "text": text,
+            "textType": text_type,
+        }
+        response = await self.execute(
+            query=SEND_NOTIFICATION_GQL,
+            operation_name="SendNotification",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return SendNotification.model_validate(data)
+
     async def create_observation(
         self, observation: ObservationInput, **kwargs: Any
     ) -> CreateObservation:
@@ -5198,6 +5254,19 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return UpdateSpecification.model_validate(data)
 
+    async def upsert_specification(
+        self, specification: SpecificationInput, **kwargs: Any
+    ) -> UpsertSpecification:
+        variables: Dict[str, object] = {"specification": specification}
+        response = await self.execute(
+            query=UPSERT_SPECIFICATION_GQL,
+            operation_name="UpsertSpecification",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return UpsertSpecification.model_validate(data)
+
     async def count_users(
         self,
         filter: Union[Optional[UserFilter], UnsetType] = UNSET,
@@ -5426,3 +5495,16 @@ class Client(AsyncBaseClient):
         )
         data = self.get_data(response)
         return UpdateWorkflow.model_validate(data)
+
+    async def upsert_workflow(
+        self, workflow: WorkflowInput, **kwargs: Any
+    ) -> UpsertWorkflow:
+        variables: Dict[str, object] = {"workflow": workflow}
+        response = await self.execute(
+            query=UPSERT_WORKFLOW_GQL,
+            operation_name="UpsertWorkflow",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return UpsertWorkflow.model_validate(data)

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"]]]
@@ -245,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"]]]
@@ -302,6 +362,7 @@ class CreateWorkflowCreateWorkflowActionsConnector(BaseModel):
     type: IntegrationServiceTypes
     uri: Optional[str]
     slack: Optional["CreateWorkflowCreateWorkflowActionsConnectorSlack"]
+    email: Optional["CreateWorkflowCreateWorkflowActionsConnectorEmail"]
 
 
 class CreateWorkflowCreateWorkflowActionsConnectorSlack(BaseModel):
@@ -309,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()
@@ -325,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

@@ -84,6 +84,7 @@ class EntityState(str, Enum):
     SANITIZED = "SANITIZED"
     EXTRACTED = "EXTRACTED"
     ENRICHED = "ENRICHED"
+    CLASSIFIED = "CLASSIFIED"
     CHANGED = "CHANGED"
     ARCHIVED = "ARCHIVED"
     APPROVED = "APPROVED"
@@ -261,6 +262,7 @@ class SearchQueryTypes(str, Enum):
 class IntegrationServiceTypes(str, Enum):
     SLACK = "SLACK"
     WEB_HOOK = "WEB_HOOK"
+    EMAIL = "EMAIL"
 
 
 class LinkTypes(str, Enum):
@@ -341,6 +343,7 @@ class SpecificationTypes(str, Enum):
     TEXT_EMBEDDING = "TEXT_EMBEDDING"
     IMAGE_EMBEDDING = "IMAGE_EMBEDDING"
     EXTRACTION = "EXTRACTION"
+    CLASSIFICATION = "CLASSIFICATION"
     PREPARATION = "PREPARATION"
 
 
@@ -498,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"
 
@@ -685,6 +690,7 @@ class DeepseekModels(str, Enum):
 class SearchServiceTypes(str, Enum):
     TAVILY = "TAVILY"
     EXA = "EXA"
+    PODSCAN = "PODSCAN"
 
 
 class MedicalContraindicationFacetTypes(str, Enum):
@@ -773,6 +779,11 @@ class FilePreparationServiceTypes(str, Enum):
     MISTRAL_DOCUMENT = "MISTRAL_DOCUMENT"
 
 
+class RegexSourceTypes(str, Enum):
+    MARKDOWN = "MARKDOWN"
+    METADATA = "METADATA"
+
+
 class FacetValueTypes(str, Enum):
     VALUE = "VALUE"
     RANGE = "RANGE"
@@ -800,6 +811,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"
 
 
@@ -937,7 +950,6 @@ class UserTypes(str, Enum):
 
 
 class EntityExtractionServiceTypes(str, Enum):
-    ROBOFLOW_IMAGE = "ROBOFLOW_IMAGE"
     MODEL_TEXT = "MODEL_TEXT"
     MODEL_IMAGE = "MODEL_IMAGE"
     OPEN_AI_IMAGE = "OPEN_AI_IMAGE"
@@ -1030,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

@@ -30,7 +30,6 @@ class GetContentContent(BaseModel):
     id: str
     name: str
     creation_date: Any = Field(alias="creationDate")
-    relevance: Optional[float]
     owner: "GetContentContentOwner"
     state: EntityState
     original_date: Optional[Any] = Field(alias="originalDate")

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"]]]
@@ -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
@@ -1756,6 +1784,7 @@ class ContentFilter(BaseModel):
     search_type: Optional[SearchTypes] = Field(alias="searchType", default=None)
     query_type: Optional[SearchQueryTypes] = Field(alias="queryType", default=None)
     number_similar: Optional[int] = Field(alias="numberSimilar", default=None)
+    image: Optional[str] = None
     disable_inheritance: Optional[bool] = Field(
         alias="disableInheritance", default=None
     )
@@ -2162,6 +2191,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 +2212,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 +2665,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 +2715,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 +2925,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 +3509,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 +3523,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 +3555,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 +3584,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")