graphlit-client 1.0.20250627002__tar.gz → 1.0.20250627004__tar.gz

{graphlit_client-1.0.20250627002/graphlit_client.egg-info → graphlit_client-1.0.20250627004}/PKG-INFO

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

{graphlit_client-1.0.20250627002 → graphlit_client-1.0.20250627004}/graphlit_api/__init__.py

@@ -1931,6 +1931,7 @@ from .operations import (
     QUERY_WORKFLOWS_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     RETRIEVE_SOURCES_GQL,
+    RETRIEVE_VIEW_GQL,
     REVISE_CONTENT_GQL,
     REVISE_ENCODED_IMAGE_GQL,
     REVISE_IMAGE_GQL,
@@ -1977,6 +1978,7 @@ from .operations import (
     UPSERT_CATEGORY_GQL,
     UPSERT_LABEL_GQL,
     UPSERT_SPECIFICATION_GQL,
+    UPSERT_VIEW_GQL,
     UPSERT_WORKFLOW_GQL,
     VIEW_EXISTS_GQL,
     WORKFLOW_EXISTS_GQL,
@@ -2666,6 +2668,12 @@ from .retrieve_sources import (
     RetrieveSourcesRetrieveSourcesResults,
     RetrieveSourcesRetrieveSourcesResultsContent,
 )
+from .retrieve_view import (
+    RetrieveView,
+    RetrieveViewRetrieveView,
+    RetrieveViewRetrieveViewResults,
+    RetrieveViewRetrieveViewResultsContent,
+)
 from .revise_content import (
     ReviseContent,
     ReviseContentReviseContent,
@@ -2944,6 +2952,60 @@ from .upsert_specification import (
     UpsertSpecification,
     UpsertSpecificationUpsertSpecification,
 )
+from .upsert_view import (
+    UpsertView,
+    UpsertViewUpsertView,
+    UpsertViewUpsertViewAugmentedFilter,
+    UpsertViewUpsertViewAugmentedFilterAnd,
+    UpsertViewUpsertViewAugmentedFilterAndCollections,
+    UpsertViewUpsertViewAugmentedFilterAndFeeds,
+    UpsertViewUpsertViewAugmentedFilterAndObservations,
+    UpsertViewUpsertViewAugmentedFilterAndObservationsObservable,
+    UpsertViewUpsertViewAugmentedFilterAndUsers,
+    UpsertViewUpsertViewAugmentedFilterAndWorkflows,
+    UpsertViewUpsertViewAugmentedFilterCollections,
+    UpsertViewUpsertViewAugmentedFilterContents,
+    UpsertViewUpsertViewAugmentedFilterCreationDateRange,
+    UpsertViewUpsertViewAugmentedFilterDateRange,
+    UpsertViewUpsertViewAugmentedFilterFeeds,
+    UpsertViewUpsertViewAugmentedFilterObservations,
+    UpsertViewUpsertViewAugmentedFilterObservationsObservable,
+    UpsertViewUpsertViewAugmentedFilterOr,
+    UpsertViewUpsertViewAugmentedFilterOrCollections,
+    UpsertViewUpsertViewAugmentedFilterOrFeeds,
+    UpsertViewUpsertViewAugmentedFilterOrObservations,
+    UpsertViewUpsertViewAugmentedFilterOrObservationsObservable,
+    UpsertViewUpsertViewAugmentedFilterOrUsers,
+    UpsertViewUpsertViewAugmentedFilterOrWorkflows,
+    UpsertViewUpsertViewAugmentedFilterSimilarContents,
+    UpsertViewUpsertViewAugmentedFilterUsers,
+    UpsertViewUpsertViewAugmentedFilterWorkflows,
+    UpsertViewUpsertViewFilter,
+    UpsertViewUpsertViewFilterAnd,
+    UpsertViewUpsertViewFilterAndCollections,
+    UpsertViewUpsertViewFilterAndFeeds,
+    UpsertViewUpsertViewFilterAndObservations,
+    UpsertViewUpsertViewFilterAndObservationsObservable,
+    UpsertViewUpsertViewFilterAndUsers,
+    UpsertViewUpsertViewFilterAndWorkflows,
+    UpsertViewUpsertViewFilterCollections,
+    UpsertViewUpsertViewFilterContents,
+    UpsertViewUpsertViewFilterCreationDateRange,
+    UpsertViewUpsertViewFilterDateRange,
+    UpsertViewUpsertViewFilterFeeds,
+    UpsertViewUpsertViewFilterObservations,
+    UpsertViewUpsertViewFilterObservationsObservable,
+    UpsertViewUpsertViewFilterOr,
+    UpsertViewUpsertViewFilterOrCollections,
+    UpsertViewUpsertViewFilterOrFeeds,
+    UpsertViewUpsertViewFilterOrObservations,
+    UpsertViewUpsertViewFilterOrObservationsObservable,
+    UpsertViewUpsertViewFilterOrUsers,
+    UpsertViewUpsertViewFilterOrWorkflows,
+    UpsertViewUpsertViewFilterSimilarContents,
+    UpsertViewUpsertViewFilterUsers,
+    UpsertViewUpsertViewFilterWorkflows,
+)
 from .upsert_workflow import (
     UpsertWorkflow,
     UpsertWorkflowUpsertWorkflow,
@@ -5295,6 +5357,7 @@ __all__ = [
     "QueryWorkflowsWorkflowsResultsStoragePolicy",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
     "RETRIEVE_SOURCES_GQL",
+    "RETRIEVE_VIEW_GQL",
     "REVISE_CONTENT_GQL",
     "REVISE_ENCODED_IMAGE_GQL",
     "REVISE_IMAGE_GQL",
@@ -5334,6 +5397,10 @@ __all__ = [
     "RetrieveSourcesRetrieveSources",
     "RetrieveSourcesRetrieveSourcesResults",
     "RetrieveSourcesRetrieveSourcesResultsContent",
+    "RetrieveView",
+    "RetrieveViewRetrieveView",
+    "RetrieveViewRetrieveViewResults",
+    "RetrieveViewRetrieveViewResultsContent",
     "ReviseContent",
     "ReviseContentReviseContent",
     "ReviseContentReviseContentConversation",
@@ -5496,6 +5563,7 @@ __all__ = [
     "UPSERT_CATEGORY_GQL",
     "UPSERT_LABEL_GQL",
     "UPSERT_SPECIFICATION_GQL",
+    "UPSERT_VIEW_GQL",
     "UPSERT_WORKFLOW_GQL",
     "UnitTypes",
     "UpdateAlert",
@@ -5678,6 +5746,58 @@ __all__ = [
     "UpsertLabelUpsertLabel",
     "UpsertSpecification",
     "UpsertSpecificationUpsertSpecification",
+    "UpsertView",
+    "UpsertViewUpsertView",
+    "UpsertViewUpsertViewAugmentedFilter",
+    "UpsertViewUpsertViewAugmentedFilterAnd",
+    "UpsertViewUpsertViewAugmentedFilterAndCollections",
+    "UpsertViewUpsertViewAugmentedFilterAndFeeds",
+    "UpsertViewUpsertViewAugmentedFilterAndObservations",
+    "UpsertViewUpsertViewAugmentedFilterAndObservationsObservable",
+    "UpsertViewUpsertViewAugmentedFilterAndUsers",
+    "UpsertViewUpsertViewAugmentedFilterAndWorkflows",
+    "UpsertViewUpsertViewAugmentedFilterCollections",
+    "UpsertViewUpsertViewAugmentedFilterContents",
+    "UpsertViewUpsertViewAugmentedFilterCreationDateRange",
+    "UpsertViewUpsertViewAugmentedFilterDateRange",
+    "UpsertViewUpsertViewAugmentedFilterFeeds",
+    "UpsertViewUpsertViewAugmentedFilterObservations",
+    "UpsertViewUpsertViewAugmentedFilterObservationsObservable",
+    "UpsertViewUpsertViewAugmentedFilterOr",
+    "UpsertViewUpsertViewAugmentedFilterOrCollections",
+    "UpsertViewUpsertViewAugmentedFilterOrFeeds",
+    "UpsertViewUpsertViewAugmentedFilterOrObservations",
+    "UpsertViewUpsertViewAugmentedFilterOrObservationsObservable",
+    "UpsertViewUpsertViewAugmentedFilterOrUsers",
+    "UpsertViewUpsertViewAugmentedFilterOrWorkflows",
+    "UpsertViewUpsertViewAugmentedFilterSimilarContents",
+    "UpsertViewUpsertViewAugmentedFilterUsers",
+    "UpsertViewUpsertViewAugmentedFilterWorkflows",
+    "UpsertViewUpsertViewFilter",
+    "UpsertViewUpsertViewFilterAnd",
+    "UpsertViewUpsertViewFilterAndCollections",
+    "UpsertViewUpsertViewFilterAndFeeds",
+    "UpsertViewUpsertViewFilterAndObservations",
+    "UpsertViewUpsertViewFilterAndObservationsObservable",
+    "UpsertViewUpsertViewFilterAndUsers",
+    "UpsertViewUpsertViewFilterAndWorkflows",
+    "UpsertViewUpsertViewFilterCollections",
+    "UpsertViewUpsertViewFilterContents",
+    "UpsertViewUpsertViewFilterCreationDateRange",
+    "UpsertViewUpsertViewFilterDateRange",
+    "UpsertViewUpsertViewFilterFeeds",
+    "UpsertViewUpsertViewFilterObservations",
+    "UpsertViewUpsertViewFilterObservationsObservable",
+    "UpsertViewUpsertViewFilterOr",
+    "UpsertViewUpsertViewFilterOrCollections",
+    "UpsertViewUpsertViewFilterOrFeeds",
+    "UpsertViewUpsertViewFilterOrObservations",
+    "UpsertViewUpsertViewFilterOrObservationsObservable",
+    "UpsertViewUpsertViewFilterOrUsers",
+    "UpsertViewUpsertViewFilterOrWorkflows",
+    "UpsertViewUpsertViewFilterSimilarContents",
+    "UpsertViewUpsertViewFilterUsers",
+    "UpsertViewUpsertViewFilterWorkflows",
     "UpsertWorkflow",
     "UpsertWorkflowUpsertWorkflow",
     "UpsertWorkflowUpsertWorkflowActions",

{graphlit_client-1.0.20250627002 → graphlit_client-1.0.20250627004}/graphlit_api/client.py

@@ -612,6 +612,7 @@ from .operations import (
     QUERY_WORKFLOWS_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     RETRIEVE_SOURCES_GQL,
+    RETRIEVE_VIEW_GQL,
     REVISE_CONTENT_GQL,
     REVISE_ENCODED_IMAGE_GQL,
     REVISE_IMAGE_GQL,
@@ -658,6 +659,7 @@ from .operations import (
     UPSERT_CATEGORY_GQL,
     UPSERT_LABEL_GQL,
     UPSERT_SPECIFICATION_GQL,
+    UPSERT_VIEW_GQL,
     UPSERT_WORKFLOW_GQL,
     VIEW_EXISTS_GQL,
     WORKFLOW_EXISTS_GQL,
@@ -721,6 +723,7 @@ from .query_views import QueryViews
 from .query_workflows import QueryWorkflows
 from .remove_contents_from_collection import RemoveContentsFromCollection
 from .retrieve_sources import RetrieveSources
+from .retrieve_view import RetrieveView
 from .revise_content import ReviseContent
 from .revise_encoded_image import ReviseEncodedImage
 from .revise_image import ReviseImage
@@ -767,6 +770,7 @@ from .update_workflow import UpdateWorkflow
 from .upsert_category import UpsertCategory
 from .upsert_label import UpsertLabel
 from .upsert_specification import UpsertSpecification
+from .upsert_view import UpsertView
 from .upsert_workflow import UpsertWorkflow
 from .view_exists import ViewExists
 from .workflow_exists import WorkflowExists
@@ -2355,6 +2359,31 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return RetrieveSources.model_validate(data)
 
+    async def retrieve_view(
+        self,
+        prompt: str,
+        id: str,
+        retrieval_strategy: Union[Optional[RetrievalStrategyInput], UnsetType] = UNSET,
+        reranking_strategy: Union[Optional[RerankingStrategyInput], UnsetType] = UNSET,
+        correlation_id: Union[Optional[str], UnsetType] = UNSET,
+        **kwargs: Any
+    ) -> RetrieveView:
+        variables: Dict[str, object] = {
+            "prompt": prompt,
+            "id": id,
+            "retrievalStrategy": retrieval_strategy,
+            "rerankingStrategy": reranking_strategy,
+            "correlationId": correlation_id,
+        }
+        response = await self.execute(
+            query=RETRIEVE_VIEW_GQL,
+            operation_name="RetrieveView",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return RetrieveView.model_validate(data)
+
     async def revise_content(
         self,
         prompt: str,
@@ -5973,6 +6002,17 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return UpdateView.model_validate(data)
 
+    async def upsert_view(self, view: ViewInput, **kwargs: Any) -> UpsertView:
+        variables: Dict[str, object] = {"view": view}
+        response = await self.execute(
+            query=UPSERT_VIEW_GQL,
+            operation_name="UpsertView",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return UpsertView.model_validate(data)
+
     async def view_exists(
         self,
         filter: Union[Optional[ViewFilter], UnsetType] = UNSET,

{graphlit_client-1.0.20250627002 → graphlit_client-1.0.20250627004}/graphlit_api/operations.py

@@ -273,6 +273,7 @@ __all__ = [
     "QUERY_WORKFLOWS_GQL",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
     "RETRIEVE_SOURCES_GQL",
+    "RETRIEVE_VIEW_GQL",
     "REVISE_CONTENT_GQL",
     "REVISE_ENCODED_IMAGE_GQL",
     "REVISE_IMAGE_GQL",
@@ -319,6 +320,7 @@ __all__ = [
     "UPSERT_CATEGORY_GQL",
     "UPSERT_LABEL_GQL",
     "UPSERT_SPECIFICATION_GQL",
+    "UPSERT_VIEW_GQL",
     "UPSERT_WORKFLOW_GQL",
     "VIEW_EXISTS_GQL",
     "WORKFLOW_EXISTS_GQL",
@@ -5941,6 +5943,32 @@ mutation RetrieveSources($prompt: String!, $filter: ContentFilter, $augmentedFil
 }
 """
 
+RETRIEVE_VIEW_GQL = """
+mutation RetrieveView($prompt: String!, $id: ID!, $retrievalStrategy: RetrievalStrategyInput, $rerankingStrategy: RerankingStrategyInput, $correlationId: String) {
+  retrieveView(
+    prompt: $prompt
+    id: $id
+    retrievalStrategy: $retrievalStrategy
+    rerankingStrategy: $rerankingStrategy
+    correlationId: $correlationId
+  ) {
+    results {
+      type
+      content {
+        id
+      }
+      text
+      metadata
+      relevance
+      startTime
+      endTime
+      pageNumber
+      frameNumber
+    }
+  }
+}
+"""
+
 REVISE_CONTENT_GQL = """
 mutation ReviseContent($prompt: String!, $content: EntityReferenceInput!, $id: ID, $specification: EntityReferenceInput, $correlationId: String) {
   reviseContent(
@@ -11145,6 +11173,183 @@ mutation UpdateView($view: ViewUpdateInput!) {
 }
 """
 
+UPSERT_VIEW_GQL = """
+mutation UpsertView($view: ViewInput!) {
+  upsertView(view: $view) {
+    id
+    name
+    state
+    type
+    filter {
+      dateRange {
+        from
+        to
+      }
+      inLast
+      creationDateRange {
+        from
+        to
+      }
+      createdInLast
+      types
+      fileTypes
+      formats
+      fileExtensions
+      similarContents {
+        id
+      }
+      contents {
+        id
+      }
+      feeds {
+        id
+      }
+      workflows {
+        id
+      }
+      collections {
+        id
+      }
+      users {
+        id
+      }
+      observations {
+        type
+        observable {
+          id
+        }
+        states
+      }
+      or {
+        feeds {
+          id
+        }
+        workflows {
+          id
+        }
+        collections {
+          id
+        }
+        users {
+          id
+        }
+        observations {
+          type
+          observable {
+            id
+          }
+          states
+        }
+      }
+      and {
+        feeds {
+          id
+        }
+        workflows {
+          id
+        }
+        collections {
+          id
+        }
+        users {
+          id
+        }
+        observations {
+          type
+          observable {
+            id
+          }
+          states
+        }
+      }
+    }
+    augmentedFilter {
+      dateRange {
+        from
+        to
+      }
+      inLast
+      creationDateRange {
+        from
+        to
+      }
+      createdInLast
+      types
+      fileTypes
+      formats
+      fileExtensions
+      similarContents {
+        id
+      }
+      contents {
+        id
+      }
+      feeds {
+        id
+      }
+      workflows {
+        id
+      }
+      collections {
+        id
+      }
+      users {
+        id
+      }
+      observations {
+        type
+        observable {
+          id
+        }
+        states
+      }
+      or {
+        feeds {
+          id
+        }
+        workflows {
+          id
+        }
+        collections {
+          id
+        }
+        users {
+          id
+        }
+        observations {
+          type
+          observable {
+            id
+          }
+          states
+        }
+      }
+      and {
+        feeds {
+          id
+        }
+        workflows {
+          id
+        }
+        collections {
+          id
+        }
+        users {
+          id
+        }
+        observations {
+          type
+          observable {
+            id
+          }
+          states
+        }
+      }
+    }
+  }
+}
+"""
+
 VIEW_EXISTS_GQL = """
 query ViewExists($filter: ViewFilter, $correlationId: String) {
   viewExists(filter: $filter, correlationId: $correlationId) {

graphlit_client-1.0.20250627004/graphlit_api/retrieve_view.py

@@ -0,0 +1,38 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import Any, List, Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import ContentSourceTypes
+
+
+class RetrieveView(BaseModel):
+    retrieve_view: Optional["RetrieveViewRetrieveView"] = Field(alias="retrieveView")
+
+
+class RetrieveViewRetrieveView(BaseModel):
+    results: Optional[List[Optional["RetrieveViewRetrieveViewResults"]]]
+
+
+class RetrieveViewRetrieveViewResults(BaseModel):
+    type: Optional[ContentSourceTypes]
+    content: "RetrieveViewRetrieveViewResultsContent"
+    text: str
+    metadata: Optional[str]
+    relevance: float
+    start_time: Optional[Any] = Field(alias="startTime")
+    end_time: Optional[Any] = Field(alias="endTime")
+    page_number: Optional[int] = Field(alias="pageNumber")
+    frame_number: Optional[int] = Field(alias="frameNumber")
+
+
+class RetrieveViewRetrieveViewResultsContent(BaseModel):
+    id: str
+
+
+RetrieveView.model_rebuild()
+RetrieveViewRetrieveView.model_rebuild()
+RetrieveViewRetrieveViewResults.model_rebuild()