graphlit-client 1.0.20250531002__tar.gz → 1.0.20250622001__tar.gz

{graphlit_client-1.0.20250531002/graphlit_client.egg-info → graphlit_client-1.0.20250622001}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphlit-client
-Version: 1.0.20250531002
+Version: 1.0.20250622001
 Summary: Graphlit API Python Client
 Home-page: https://github.com/graphlit/graphlit-client-python
 Author: Unstruk Data Inc.

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/__init__.py

@@ -20,6 +20,10 @@ from .ask_graphlit import (
 )
 from .async_base_client import AsyncBaseClient
 from .base_model import BaseModel, Upload
+from .branch_conversation import (
+    BranchConversation,
+    BranchConversationBranchConversation,
+)
 from .clear_conversation import ClearConversation, ClearConversationClearConversation
 from .client import Client
 from .close_conversation import CloseConversation, CloseConversationCloseConversation
@@ -259,6 +263,7 @@ from .create_workflow import (
     CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocument,
     CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocumentSpecification,
     CreateWorkflowCreateWorkflowPreparationJobsConnectorPage,
+    CreateWorkflowCreateWorkflowPreparationJobsConnectorReducto,
     CreateWorkflowCreateWorkflowPreparationSummarizations,
     CreateWorkflowCreateWorkflowPreparationSummarizationsSpecification,
     CreateWorkflowCreateWorkflowStorage,
@@ -597,6 +602,10 @@ from .enums import (
     PolicyTimeTypes,
     ProductFacetTypes,
     PromptStrategyTypes,
+    ReductoEnrichmentModes,
+    ReductoExtractionModes,
+    ReductoOcrModes,
+    ReductoOcrSystems,
     RegexSourceTypes,
     RenditionTypes,
     ReplicateModels,
@@ -737,6 +746,7 @@ from .get_content import (
     GetContentContentEmailTo,
     GetContentContentFeed,
     GetContentContentFrames,
+    GetContentContentH3,
     GetContentContentImage,
     GetContentContentIssue,
     GetContentContentLanguage,
@@ -1024,6 +1034,7 @@ from .get_workflow import (
     GetWorkflowWorkflowPreparationJobsConnectorModelDocument,
     GetWorkflowWorkflowPreparationJobsConnectorModelDocumentSpecification,
     GetWorkflowWorkflowPreparationJobsConnectorPage,
+    GetWorkflowWorkflowPreparationJobsConnectorReducto,
     GetWorkflowWorkflowPreparationSummarizations,
     GetWorkflowWorkflowPreparationSummarizationsSpecification,
     GetWorkflowWorkflowStorage,
@@ -1361,6 +1372,7 @@ from .input_types import (
     PromptStrategyUpdateInput,
     RedditFeedPropertiesInput,
     RedditFeedPropertiesUpdateInput,
+    ReductoDocumentPreparationPropertiesInput,
     RegexClassificationRuleInput,
     RegexContentClassificationPropertiesInput,
     ReplicateModelPropertiesInput,
@@ -1428,12 +1440,50 @@ from .input_types import (
 )
 from .is_content_done import IsContentDone, IsContentDoneIsContentDone
 from .is_feed_done import IsFeedDone, IsFeedDoneIsFeedDone
+from .lookup_contents import (
+    LookupContents,
+    LookupContentsLookupContents,
+    LookupContentsLookupContentsResults,
+    LookupContentsLookupContentsResultsAddress,
+    LookupContentsLookupContentsResultsAudio,
+    LookupContentsLookupContentsResultsChildren,
+    LookupContentsLookupContentsResultsCollections,
+    LookupContentsLookupContentsResultsDocument,
+    LookupContentsLookupContentsResultsEmail,
+    LookupContentsLookupContentsResultsEmailBcc,
+    LookupContentsLookupContentsResultsEmailCc,
+    LookupContentsLookupContentsResultsEmailFrom,
+    LookupContentsLookupContentsResultsEmailTo,
+    LookupContentsLookupContentsResultsFeed,
+    LookupContentsLookupContentsResultsFrames,
+    LookupContentsLookupContentsResultsH3,
+    LookupContentsLookupContentsResultsImage,
+    LookupContentsLookupContentsResultsIssue,
+    LookupContentsLookupContentsResultsLanguage,
+    LookupContentsLookupContentsResultsLinks,
+    LookupContentsLookupContentsResultsLocation,
+    LookupContentsLookupContentsResultsObservations,
+    LookupContentsLookupContentsResultsObservationsObservable,
+    LookupContentsLookupContentsResultsObservationsOccurrences,
+    LookupContentsLookupContentsResultsObservationsOccurrencesBoundingBox,
+    LookupContentsLookupContentsResultsObservationsRelated,
+    LookupContentsLookupContentsResultsOwner,
+    LookupContentsLookupContentsResultsPackage,
+    LookupContentsLookupContentsResultsPages,
+    LookupContentsLookupContentsResultsPagesChunks,
+    LookupContentsLookupContentsResultsPagesImages,
+    LookupContentsLookupContentsResultsParent,
+    LookupContentsLookupContentsResultsSegments,
+    LookupContentsLookupContentsResultsVideo,
+    LookupContentsLookupContentsResultsWorkflow,
+)
 from .lookup_credits import LookupCredits, LookupCreditsLookupCredits
 from .lookup_usage import LookupUsage, LookupUsageLookupUsage
 from .map_web import MapWeb, MapWebMapWeb
 from .operations import (
     ADD_CONTENTS_TO_COLLECTIONS_GQL,
     ASK_GRAPHLIT_GQL,
+    BRANCH_CONVERSATION_GQL,
     CLEAR_CONVERSATION_GQL,
     CLOSE_CONVERSATION_GQL,
     COMPLETE_CONVERSATION_GQL,
@@ -1629,6 +1679,7 @@ from .operations import (
     INGEST_URI_GQL,
     IS_CONTENT_DONE_GQL,
     IS_FEED_DONE_GQL,
+    LOOKUP_CONTENTS_GQL,
     LOOKUP_CREDITS_GQL,
     LOOKUP_USAGE_GQL,
     MAP_WEB_GQL,
@@ -1868,7 +1919,6 @@ from .query_collections import (
     QueryCollections,
     QueryCollectionsCollections,
     QueryCollectionsCollectionsResults,
-    QueryCollectionsCollectionsResultsContents,
     QueryCollectionsCollectionsResultsOwner,
 )
 from .query_contents import (
@@ -1877,7 +1927,6 @@ from .query_contents import (
     QueryContentsContentsResults,
     QueryContentsContentsResultsAddress,
     QueryContentsContentsResultsAudio,
-    QueryContentsContentsResultsCollections,
     QueryContentsContentsResultsDocument,
     QueryContentsContentsResultsEmail,
     QueryContentsContentsResultsEmailBcc,
@@ -1891,11 +1940,6 @@ from .query_contents import (
     QueryContentsContentsResultsLanguage,
     QueryContentsContentsResultsLinks,
     QueryContentsContentsResultsLocation,
-    QueryContentsContentsResultsObservations,
-    QueryContentsContentsResultsObservationsObservable,
-    QueryContentsContentsResultsObservationsOccurrences,
-    QueryContentsContentsResultsObservationsOccurrencesBoundingBox,
-    QueryContentsContentsResultsObservationsRelated,
     QueryContentsContentsResultsOwner,
     QueryContentsContentsResultsPackage,
     QueryContentsContentsResultsPages,
@@ -2115,8 +2159,13 @@ from .query_models import (
 from .query_notion_databases import (
     QueryNotionDatabases,
     QueryNotionDatabasesNotionDatabases,
+    QueryNotionDatabasesNotionDatabasesResults,
+)
+from .query_notion_pages import (
+    QueryNotionPages,
+    QueryNotionPagesNotionPages,
+    QueryNotionPagesNotionPagesResults,
 )
-from .query_notion_pages import QueryNotionPages, QueryNotionPagesNotionPages
 from .query_one_drive_folders import (
     QueryOneDriveFolders,
     QueryOneDriveFoldersOneDriveFolders,
@@ -2258,6 +2307,7 @@ from .query_workflows import (
     QueryWorkflowsWorkflowsResultsPreparationJobsConnectorModelDocument,
     QueryWorkflowsWorkflowsResultsPreparationJobsConnectorModelDocumentSpecification,
     QueryWorkflowsWorkflowsResultsPreparationJobsConnectorPage,
+    QueryWorkflowsWorkflowsResultsPreparationJobsConnectorReducto,
     QueryWorkflowsWorkflowsResultsPreparationSummarizations,
     QueryWorkflowsWorkflowsResultsPreparationSummarizationsSpecification,
     QueryWorkflowsWorkflowsResultsStorage,
@@ -2485,6 +2535,7 @@ from .update_workflow import (
     UpdateWorkflowUpdateWorkflowPreparationJobsConnectorModelDocument,
     UpdateWorkflowUpdateWorkflowPreparationJobsConnectorModelDocumentSpecification,
     UpdateWorkflowUpdateWorkflowPreparationJobsConnectorPage,
+    UpdateWorkflowUpdateWorkflowPreparationJobsConnectorReducto,
     UpdateWorkflowUpdateWorkflowPreparationSummarizations,
     UpdateWorkflowUpdateWorkflowPreparationSummarizationsSpecification,
     UpdateWorkflowUpdateWorkflowStorage,
@@ -2547,6 +2598,7 @@ from .upsert_workflow import (
     UpsertWorkflowUpsertWorkflowPreparationJobsConnectorModelDocument,
     UpsertWorkflowUpsertWorkflowPreparationJobsConnectorModelDocumentSpecification,
     UpsertWorkflowUpsertWorkflowPreparationJobsConnectorPage,
+    UpsertWorkflowUpsertWorkflowPreparationJobsConnectorReducto,
     UpsertWorkflowUpsertWorkflowPreparationSummarizations,
     UpsertWorkflowUpsertWorkflowPreparationSummarizationsSpecification,
     UpsertWorkflowUpsertWorkflowStorage,
@@ -2606,6 +2658,7 @@ __all__ = [
     "AzureOpenAIModelPropertiesUpdateInput",
     "AzureOpenAIModels",
     "AzureTextExtractionPropertiesInput",
+    "BRANCH_CONVERSATION_GQL",
     "BaseModel",
     "BedrockModelPropertiesInput",
     "BedrockModelPropertiesUpdateInput",
@@ -2614,6 +2667,8 @@ __all__ = [
     "BoundingBoxInput",
     "BoxFeedPropertiesInput",
     "BoxFeedPropertiesUpdateInput",
+    "BranchConversation",
+    "BranchConversationBranchConversation",
     "CLEAR_CONVERSATION_GQL",
     "CLOSE_CONVERSATION_GQL",
     "COMPLETE_CONVERSATION_GQL",
@@ -2944,6 +2999,7 @@ __all__ = [
     "CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocument",
     "CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocumentSpecification",
     "CreateWorkflowCreateWorkflowPreparationJobsConnectorPage",
+    "CreateWorkflowCreateWorkflowPreparationJobsConnectorReducto",
     "CreateWorkflowCreateWorkflowPreparationSummarizations",
     "CreateWorkflowCreateWorkflowPreparationSummarizationsSpecification",
     "CreateWorkflowCreateWorkflowStorage",
@@ -3422,6 +3478,7 @@ __all__ = [
     "GetContentContentEmailTo",
     "GetContentContentFeed",
     "GetContentContentFrames",
+    "GetContentContentH3",
     "GetContentContentImage",
     "GetContentContentIssue",
     "GetContentContentLanguage",
@@ -3692,6 +3749,7 @@ __all__ = [
     "GetWorkflowWorkflowPreparationJobsConnectorModelDocument",
     "GetWorkflowWorkflowPreparationJobsConnectorModelDocumentSpecification",
     "GetWorkflowWorkflowPreparationJobsConnectorPage",
+    "GetWorkflowWorkflowPreparationJobsConnectorReducto",
     "GetWorkflowWorkflowPreparationSummarizations",
     "GetWorkflowWorkflowPreparationSummarizationsSpecification",
     "GetWorkflowWorkflowStorage",
@@ -3814,6 +3872,7 @@ __all__ = [
     "JinaModelPropertiesInput",
     "JinaModelPropertiesUpdateInput",
     "JinaModels",
+    "LOOKUP_CONTENTS_GQL",
     "LOOKUP_CREDITS_GQL",
     "LOOKUP_USAGE_GQL",
     "LabelFacetInput",
@@ -3827,6 +3886,41 @@ __all__ = [
     "LinearProjectsInput",
     "LinkStrategyInput",
     "LinkTypes",
+    "LookupContents",
+    "LookupContentsLookupContents",
+    "LookupContentsLookupContentsResults",
+    "LookupContentsLookupContentsResultsAddress",
+    "LookupContentsLookupContentsResultsAudio",
+    "LookupContentsLookupContentsResultsChildren",
+    "LookupContentsLookupContentsResultsCollections",
+    "LookupContentsLookupContentsResultsDocument",
+    "LookupContentsLookupContentsResultsEmail",
+    "LookupContentsLookupContentsResultsEmailBcc",
+    "LookupContentsLookupContentsResultsEmailCc",
+    "LookupContentsLookupContentsResultsEmailFrom",
+    "LookupContentsLookupContentsResultsEmailTo",
+    "LookupContentsLookupContentsResultsFeed",
+    "LookupContentsLookupContentsResultsFrames",
+    "LookupContentsLookupContentsResultsH3",
+    "LookupContentsLookupContentsResultsImage",
+    "LookupContentsLookupContentsResultsIssue",
+    "LookupContentsLookupContentsResultsLanguage",
+    "LookupContentsLookupContentsResultsLinks",
+    "LookupContentsLookupContentsResultsLocation",
+    "LookupContentsLookupContentsResultsObservations",
+    "LookupContentsLookupContentsResultsObservationsObservable",
+    "LookupContentsLookupContentsResultsObservationsOccurrences",
+    "LookupContentsLookupContentsResultsObservationsOccurrencesBoundingBox",
+    "LookupContentsLookupContentsResultsObservationsRelated",
+    "LookupContentsLookupContentsResultsOwner",
+    "LookupContentsLookupContentsResultsPackage",
+    "LookupContentsLookupContentsResultsPages",
+    "LookupContentsLookupContentsResultsPagesChunks",
+    "LookupContentsLookupContentsResultsPagesImages",
+    "LookupContentsLookupContentsResultsParent",
+    "LookupContentsLookupContentsResultsSegments",
+    "LookupContentsLookupContentsResultsVideo",
+    "LookupContentsLookupContentsResultsWorkflow",
     "LookupCredits",
     "LookupCreditsLookupCredits",
     "LookupUsage",
@@ -4148,14 +4242,12 @@ __all__ = [
     "QueryCollections",
     "QueryCollectionsCollections",
     "QueryCollectionsCollectionsResults",
-    "QueryCollectionsCollectionsResultsContents",
     "QueryCollectionsCollectionsResultsOwner",
     "QueryContents",
     "QueryContentsContents",
     "QueryContentsContentsResults",
     "QueryContentsContentsResultsAddress",
     "QueryContentsContentsResultsAudio",
-    "QueryContentsContentsResultsCollections",
     "QueryContentsContentsResultsDocument",
     "QueryContentsContentsResultsEmail",
     "QueryContentsContentsResultsEmailBcc",
@@ -4169,11 +4261,6 @@ __all__ = [
     "QueryContentsContentsResultsLanguage",
     "QueryContentsContentsResultsLinks",
     "QueryContentsContentsResultsLocation",
-    "QueryContentsContentsResultsObservations",
-    "QueryContentsContentsResultsObservationsObservable",
-    "QueryContentsContentsResultsObservationsOccurrences",
-    "QueryContentsContentsResultsObservationsOccurrencesBoundingBox",
-    "QueryContentsContentsResultsObservationsRelated",
     "QueryContentsContentsResultsOwner",
     "QueryContentsContentsResultsPackage",
     "QueryContentsContentsResultsPages",
@@ -4354,8 +4441,10 @@ __all__ = [
     "QueryModelsModelsResultsMetadata",
     "QueryNotionDatabases",
     "QueryNotionDatabasesNotionDatabases",
+    "QueryNotionDatabasesNotionDatabasesResults",
     "QueryNotionPages",
     "QueryNotionPagesNotionPages",
+    "QueryNotionPagesNotionPagesResults",
     "QueryOneDriveFolders",
     "QueryOneDriveFoldersOneDriveFolders",
     "QueryOneDriveFoldersOneDriveFoldersResults",
@@ -4481,6 +4570,7 @@ __all__ = [
     "QueryWorkflowsWorkflowsResultsPreparationJobsConnectorModelDocument",
     "QueryWorkflowsWorkflowsResultsPreparationJobsConnectorModelDocumentSpecification",
     "QueryWorkflowsWorkflowsResultsPreparationJobsConnectorPage",
+    "QueryWorkflowsWorkflowsResultsPreparationJobsConnectorReducto",
     "QueryWorkflowsWorkflowsResultsPreparationSummarizations",
     "QueryWorkflowsWorkflowsResultsPreparationSummarizationsSpecification",
     "QueryWorkflowsWorkflowsResultsStorage",
@@ -4495,6 +4585,11 @@ __all__ = [
     "RSSFeedPropertiesUpdateInput",
     "RedditFeedPropertiesInput",
     "RedditFeedPropertiesUpdateInput",
+    "ReductoDocumentPreparationPropertiesInput",
+    "ReductoEnrichmentModes",
+    "ReductoExtractionModes",
+    "ReductoOcrModes",
+    "ReductoOcrSystems",
     "RegexClassificationRuleInput",
     "RegexContentClassificationPropertiesInput",
     "RegexSourceTypes",
@@ -4797,6 +4892,7 @@ __all__ = [
     "UpdateWorkflowUpdateWorkflowPreparationJobsConnectorModelDocument",
     "UpdateWorkflowUpdateWorkflowPreparationJobsConnectorModelDocumentSpecification",
     "UpdateWorkflowUpdateWorkflowPreparationJobsConnectorPage",
+    "UpdateWorkflowUpdateWorkflowPreparationJobsConnectorReducto",
     "UpdateWorkflowUpdateWorkflowPreparationSummarizations",
     "UpdateWorkflowUpdateWorkflowPreparationSummarizationsSpecification",
     "UpdateWorkflowUpdateWorkflowStorage",
@@ -4858,6 +4954,7 @@ __all__ = [
     "UpsertWorkflowUpsertWorkflowPreparationJobsConnectorModelDocument",
     "UpsertWorkflowUpsertWorkflowPreparationJobsConnectorModelDocumentSpecification",
     "UpsertWorkflowUpsertWorkflowPreparationJobsConnectorPage",
+    "UpsertWorkflowUpsertWorkflowPreparationJobsConnectorReducto",
     "UpsertWorkflowUpsertWorkflowPreparationSummarizations",
     "UpsertWorkflowUpsertWorkflowPreparationSummarizationsSpecification",
     "UpsertWorkflowUpsertWorkflowStorage",

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/ask_graphlit.py

@@ -41,6 +41,7 @@ class AskGraphlitAskGraphlitMessage(BaseModel):
     )
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -89,6 +90,7 @@ class AskGraphlitAskGraphlitMessageCitationsContent(BaseModel):
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional["AskGraphlitAskGraphlitMessageCitationsContentVideo"]
     audio: Optional["AskGraphlitAskGraphlitMessageCitationsContentAudio"]
     image: Optional["AskGraphlitAskGraphlitMessageCitationsContentImage"]

graphlit_client-1.0.20250622001/graphlit_api/branch_conversation.py

@@ -0,0 +1,25 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import ConversationTypes, EntityState
+
+
+class BranchConversation(BaseModel):
+    branch_conversation: Optional["BranchConversationBranchConversation"] = Field(
+        alias="branchConversation"
+    )
+
+
+class BranchConversationBranchConversation(BaseModel):
+    id: str
+    name: str
+    state: EntityState
+    type: Optional[ConversationTypes]
+
+
+BranchConversation.model_rebuild()

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/client.py

@@ -7,6 +7,7 @@ from .add_contents_to_collections import AddContentsToCollections
 from .ask_graphlit import AskGraphlit
 from .async_base_client import AsyncBaseClient
 from .base_model import UNSET, UnsetType
+from .branch_conversation import BranchConversation
 from .clear_conversation import ClearConversation
 from .close_conversation import CloseConversation
 from .complete_conversation import CompleteConversation
@@ -314,12 +315,14 @@ from .input_types import (
 )
 from .is_content_done import IsContentDone
 from .is_feed_done import IsFeedDone
+from .lookup_contents import LookupContents
 from .lookup_credits import LookupCredits
 from .lookup_usage import LookupUsage
 from .map_web import MapWeb
 from .operations import (
     ADD_CONTENTS_TO_COLLECTIONS_GQL,
     ASK_GRAPHLIT_GQL,
+    BRANCH_CONVERSATION_GQL,
     CLEAR_CONVERSATION_GQL,
     CLOSE_CONVERSATION_GQL,
     COMPLETE_CONVERSATION_GQL,
@@ -515,6 +518,7 @@ from .operations import (
     INGEST_URI_GQL,
     IS_CONTENT_DONE_GQL,
     IS_FEED_DONE_GQL,
+    LOOKUP_CONTENTS_GQL,
     LOOKUP_CREDITS_GQL,
     LOOKUP_USAGE_GQL,
     MAP_WEB_GQL,
@@ -1430,6 +1434,7 @@ class Client(AsyncBaseClient):
         name: Union[Optional[str], UnsetType] = UNSET,
         description: Union[Optional[str], UnsetType] = UNSET,
         event_date: Union[Optional[Any], UnsetType] = UNSET,
+        id: Union[Optional[str], UnsetType] = UNSET,
         collections: Union[Optional[List[EntityReferenceInput]], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
@@ -1439,6 +1444,7 @@ class Client(AsyncBaseClient):
             "name": name,
             "description": description,
             "eventDate": event_date,
+            "id": id,
             "collections": collections,
             "correlationId": correlation_id,
         }
@@ -1456,6 +1462,7 @@ class Client(AsyncBaseClient):
         text: str,
         name: Union[Optional[str], UnsetType] = UNSET,
         text_type: Union[Optional[TextTypes], UnsetType] = UNSET,
+        id: Union[Optional[str], UnsetType] = UNSET,
         collections: Union[Optional[List[EntityReferenceInput]], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
@@ -1464,6 +1471,7 @@ class Client(AsyncBaseClient):
             "text": text,
             "name": name,
             "textType": text_type,
+            "id": id,
             "collections": collections,
             "correlationId": correlation_id,
         }
@@ -1588,6 +1596,22 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return IsContentDone.model_validate(data)
 
+    async def lookup_contents(
+        self,
+        ids: List[str],
+        correlation_id: Union[Optional[str], UnsetType] = UNSET,
+        **kwargs: Any
+    ) -> LookupContents:
+        variables: Dict[str, object] = {"ids": ids, "correlationId": correlation_id}
+        response = await self.execute(
+            query=LOOKUP_CONTENTS_GQL,
+            operation_name="LookupContents",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return LookupContents.model_validate(data)
+
     async def publish_contents(
         self,
         publish_prompt: str,
@@ -1824,6 +1848,17 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return AskGraphlit.model_validate(data)
 
+    async def branch_conversation(self, id: str, **kwargs: Any) -> BranchConversation:
+        variables: Dict[str, object] = {"id": id}
+        response = await self.execute(
+            query=BRANCH_CONVERSATION_GQL,
+            operation_name="BranchConversation",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return BranchConversation.model_validate(data)
+
     async def clear_conversation(self, id: str, **kwargs: Any) -> ClearConversation:
         variables: Dict[str, object] = {"id": id}
         response = await self.execute(
@@ -1850,12 +1885,18 @@ class Client(AsyncBaseClient):
         self,
         completion: str,
         id: str,
+        completion_time: Union[Optional[Any], UnsetType] = UNSET,
+        ttft: Union[Optional[Any], UnsetType] = UNSET,
+        throughput: Union[Optional[float], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
     ) -> CompleteConversation:
         variables: Dict[str, object] = {
             "completion": completion,
             "id": id,
+            "completionTime": completion_time,
+            "ttft": ttft,
+            "throughput": throughput,
             "correlationId": correlation_id,
         }
         response = await self.execute(
@@ -1979,6 +2020,8 @@ class Client(AsyncBaseClient):
         prompt: str,
         id: Union[Optional[str], UnsetType] = UNSET,
         specification: Union[Optional[EntityReferenceInput], UnsetType] = UNSET,
+        tools: Union[Optional[List[ToolDefinitionInput]], UnsetType] = UNSET,
+        system_prompt: Union[Optional[str], UnsetType] = UNSET,
         include_details: Union[Optional[bool], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
@@ -1987,6 +2030,8 @@ class Client(AsyncBaseClient):
             "prompt": prompt,
             "id": id,
             "specification": specification,
+            "tools": tools,
+            "systemPrompt": system_prompt,
             "includeDetails": include_details,
             "correlationId": correlation_id,
         }
@@ -2046,6 +2091,7 @@ class Client(AsyncBaseClient):
         data: Union[Optional[str], UnsetType] = UNSET,
         id: Union[Optional[str], UnsetType] = UNSET,
         specification: Union[Optional[EntityReferenceInput], UnsetType] = UNSET,
+        system_prompt: Union[Optional[str], UnsetType] = UNSET,
         tools: Union[Optional[List[ToolDefinitionInput]], UnsetType] = UNSET,
         require_tool: Union[Optional[bool], UnsetType] = UNSET,
         include_details: Union[Optional[bool], UnsetType] = UNSET,
@@ -2058,6 +2104,7 @@ class Client(AsyncBaseClient):
             "data": data,
             "id": id,
             "specification": specification,
+            "systemPrompt": system_prompt,
             "tools": tools,
             "requireTool": require_tool,
             "includeDetails": include_details,

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/complete_conversation.py

@@ -52,6 +52,7 @@ class CompleteConversationCompleteConversationMessage(BaseModel):
     ] = Field(alias="toolCalls")
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -100,6 +101,7 @@ class CompleteConversationCompleteConversationMessageCitationsContent(BaseModel)
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional[
         "CompleteConversationCompleteConversationMessageCitationsContentVideo"
     ]
@@ -284,6 +286,7 @@ class CompleteConversationCompleteConversationDetailsMessages(BaseModel):
     ] = Field(alias="toolCalls")
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -336,6 +339,7 @@ class CompleteConversationCompleteConversationDetailsMessagesCitationsContent(
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional[
         "CompleteConversationCompleteConversationDetailsMessagesCitationsContentVideo"
     ]

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/continue_conversation.py

@@ -52,6 +52,7 @@ class ContinueConversationContinueConversationMessage(BaseModel):
     ] = Field(alias="toolCalls")
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -100,6 +101,7 @@ class ContinueConversationContinueConversationMessageCitationsContent(BaseModel)
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional[
         "ContinueConversationContinueConversationMessageCitationsContentVideo"
     ]
@@ -284,6 +286,7 @@ class ContinueConversationContinueConversationDetailsMessages(BaseModel):
     ] = Field(alias="toolCalls")
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -336,6 +339,7 @@ class ContinueConversationContinueConversationDetailsMessagesCitationsContent(
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional[
         "ContinueConversationContinueConversationDetailsMessagesCitationsContentVideo"
     ]

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/create_workflow.py

@@ -22,6 +22,10 @@ from .enums import (
     IntegrationServiceTypes,
     LinkTypes,
     ObservableTypes,
+    ReductoEnrichmentModes,
+    ReductoExtractionModes,
+    ReductoOcrModes,
+    ReductoOcrSystems,
     RegexSourceTypes,
     StoragePolicyTypes,
     SummarizationTypes,
@@ -139,6 +143,7 @@ class CreateWorkflowCreateWorkflowPreparationJobsConnector(BaseModel):
     model_document: Optional[
         "CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocument"
     ] = Field(alias="modelDocument")
+    reducto: Optional["CreateWorkflowCreateWorkflowPreparationJobsConnectorReducto"]
     mistral: Optional["CreateWorkflowCreateWorkflowPreparationJobsConnectorMistral"]
 
 
@@ -191,6 +196,15 @@ class CreateWorkflowCreateWorkflowPreparationJobsConnectorModelDocumentSpecifica
     id: str
 
 
+class CreateWorkflowCreateWorkflowPreparationJobsConnectorReducto(BaseModel):
+    ocr_mode: Optional[ReductoOcrModes] = Field(alias="ocrMode")
+    ocr_system: Optional[ReductoOcrSystems] = Field(alias="ocrSystem")
+    extraction_mode: Optional[ReductoExtractionModes] = Field(alias="extractionMode")
+    enable_enrichment: Optional[bool] = Field(alias="enableEnrichment")
+    enrichment_mode: Optional[ReductoEnrichmentModes] = Field(alias="enrichmentMode")
+    key: Optional[str]
+
+
 class CreateWorkflowCreateWorkflowPreparationJobsConnectorMistral(BaseModel):
     key: Optional[str]
 

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/describe_encoded_image.py

@@ -35,6 +35,7 @@ class DescribeEncodedImageDescribeEncodedImage(BaseModel):
     ] = Field(alias="toolCalls")
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -83,6 +84,7 @@ class DescribeEncodedImageDescribeEncodedImageCitationsContent(BaseModel):
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional["DescribeEncodedImageDescribeEncodedImageCitationsContentVideo"]
     audio: Optional["DescribeEncodedImageDescribeEncodedImageCitationsContentAudio"]
     image: Optional["DescribeEncodedImageDescribeEncodedImageCitationsContentImage"]

{graphlit_client-1.0.20250531002 → graphlit_client-1.0.20250622001}/graphlit_api/describe_image.py

@@ -33,6 +33,7 @@ class DescribeImageDescribeImage(BaseModel):
     )
     tokens: Optional[int]
     throughput: Optional[float]
+    ttft: Optional[Any]
     completion_time: Optional[Any] = Field(alias="completionTime")
     timestamp: Optional[Any]
     model_service: Optional[ModelServiceTypes] = Field(alias="modelService")
@@ -81,6 +82,7 @@ class DescribeImageDescribeImageCitationsContent(BaseModel):
     posts: Optional[List[str]]
     chapters: Optional[List[str]]
     questions: Optional[List[str]]
+    quotes: Optional[List[str]]
     video: Optional["DescribeImageDescribeImageCitationsContentVideo"]
     audio: Optional["DescribeImageDescribeImageCitationsContentAudio"]
     image: Optional["DescribeImageDescribeImageCitationsContentImage"]