graphlit-client 1.0.20250410001__tar.gz → 1.0.20250412001__tar.gz

{graphlit_client-1.0.20250410001/graphlit_client.egg-info → graphlit_client-1.0.20250412001}/PKG-INFO

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

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001}/graphlit_api/__init__.py

@@ -1026,6 +1026,16 @@ from .ingest_encoded_file import (
     IngestEncodedFileIngestEncodedFileObservationsOccurrencesBoundingBox,
     IngestEncodedFileIngestEncodedFileObservationsRelated,
 )
+from .ingest_memory import (
+    IngestMemory,
+    IngestMemoryIngestMemory,
+    IngestMemoryIngestMemoryCollections,
+    IngestMemoryIngestMemoryObservations,
+    IngestMemoryIngestMemoryObservationsObservable,
+    IngestMemoryIngestMemoryObservationsOccurrences,
+    IngestMemoryIngestMemoryObservationsOccurrencesBoundingBox,
+    IngestMemoryIngestMemoryObservationsRelated,
+)
 from .ingest_text import (
     IngestText,
     IngestTextIngestText,
@@ -1574,6 +1584,7 @@ from .operations import (
     GET_WORKFLOW_GQL,
     INGEST_BATCH_GQL,
     INGEST_ENCODED_FILE_GQL,
+    INGEST_MEMORY_GQL,
     INGEST_TEXT_BATCH_GQL,
     INGEST_TEXT_GQL,
     INGEST_URI_GQL,
@@ -3637,6 +3648,7 @@ __all__ = [
     "H3ResolutionTypes",
     "INGEST_BATCH_GQL",
     "INGEST_ENCODED_FILE_GQL",
+    "INGEST_MEMORY_GQL",
     "INGEST_TEXT_BATCH_GQL",
     "INGEST_TEXT_GQL",
     "INGEST_URI_GQL",
@@ -3662,6 +3674,14 @@ __all__ = [
     "IngestEncodedFileIngestEncodedFileObservationsOccurrences",
     "IngestEncodedFileIngestEncodedFileObservationsOccurrencesBoundingBox",
     "IngestEncodedFileIngestEncodedFileObservationsRelated",
+    "IngestMemory",
+    "IngestMemoryIngestMemory",
+    "IngestMemoryIngestMemoryCollections",
+    "IngestMemoryIngestMemoryObservations",
+    "IngestMemoryIngestMemoryObservationsObservable",
+    "IngestMemoryIngestMemoryObservationsOccurrences",
+    "IngestMemoryIngestMemoryObservationsOccurrencesBoundingBox",
+    "IngestMemoryIngestMemoryObservationsRelated",
     "IngestText",
     "IngestTextBatch",
     "IngestTextBatchIngestTextBatch",

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001}/graphlit_api/client.py

@@ -195,6 +195,7 @@ from .get_user import GetUser
 from .get_workflow import GetWorkflow
 from .ingest_batch import IngestBatch
 from .ingest_encoded_file import IngestEncodedFile
+from .ingest_memory import IngestMemory
 from .ingest_text import IngestText
 from .ingest_text_batch import IngestTextBatch
 from .ingest_uri import IngestUri
@@ -504,6 +505,7 @@ from .operations import (
     GET_WORKFLOW_GQL,
     INGEST_BATCH_GQL,
     INGEST_ENCODED_FILE_GQL,
+    INGEST_MEMORY_GQL,
     INGEST_TEXT_BATCH_GQL,
     INGEST_TEXT_GQL,
     INGEST_URI_GQL,
@@ -1412,10 +1414,35 @@ class Client(AsyncBaseClient):
         _data = self.get_data(response)
         return IngestEncodedFile.model_validate(_data)
 
+    async def ingest_memory(
+        self,
+        text: str,
+        name: Union[Optional[str], UnsetType] = UNSET,
+        text_type: Union[Optional[TextTypes], UnsetType] = UNSET,
+        collections: Union[Optional[List[EntityReferenceInput]], UnsetType] = UNSET,
+        correlation_id: Union[Optional[str], UnsetType] = UNSET,
+        **kwargs: Any
+    ) -> IngestMemory:
+        variables: Dict[str, object] = {
+            "text": text,
+            "name": name,
+            "textType": text_type,
+            "collections": collections,
+            "correlationId": correlation_id,
+        }
+        response = await self.execute(
+            query=INGEST_MEMORY_GQL,
+            operation_name="IngestMemory",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return IngestMemory.model_validate(data)
+
     async def ingest_text(
         self,
-        name: str,
         text: str,
+        name: Union[Optional[str], UnsetType] = UNSET,
         text_type: Union[Optional[TextTypes], UnsetType] = UNSET,
         uri: Union[Optional[Any], UnsetType] = UNSET,
         id: Union[Optional[str], UnsetType] = UNSET,
@@ -1429,8 +1456,8 @@ class Client(AsyncBaseClient):
         **kwargs: Any
     ) -> IngestText:
         variables: Dict[str, object] = {
-            "name": name,
             "text": text,
+            "name": name,
             "textType": text_type,
             "uri": uri,
             "id": id,

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001}/graphlit_api/enums.py

@@ -179,6 +179,7 @@ class ContentTypes(str, Enum):
     EMAIL = "EMAIL"
     EVENT = "EVENT"
     ISSUE = "ISSUE"
+    MEMORY = "MEMORY"
 
 
 class CerebrasModels(str, Enum):

graphlit_client-1.0.20250412001/graphlit_api/ingest_memory.py

@@ -0,0 +1,83 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import Any, List, Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import (
+    ContentTypes,
+    EntityState,
+    FileTypes,
+    ObservableTypes,
+    OccurrenceTypes,
+)
+
+
+class IngestMemory(BaseModel):
+    ingest_memory: Optional["IngestMemoryIngestMemory"] = Field(alias="ingestMemory")
+
+
+class IngestMemoryIngestMemory(BaseModel):
+    id: str
+    name: str
+    state: EntityState
+    type: Optional[ContentTypes]
+    file_type: Optional[FileTypes] = Field(alias="fileType")
+    mime_type: Optional[str] = Field(alias="mimeType")
+    uri: Optional[Any]
+    collections: Optional[List[Optional["IngestMemoryIngestMemoryCollections"]]]
+    observations: Optional[List[Optional["IngestMemoryIngestMemoryObservations"]]]
+
+
+class IngestMemoryIngestMemoryCollections(BaseModel):
+    id: str
+    name: str
+
+
+class IngestMemoryIngestMemoryObservations(BaseModel):
+    id: str
+    type: ObservableTypes
+    observable: "IngestMemoryIngestMemoryObservationsObservable"
+    related: Optional["IngestMemoryIngestMemoryObservationsRelated"]
+    related_type: Optional[ObservableTypes] = Field(alias="relatedType")
+    relation: Optional[str]
+    occurrences: Optional[
+        List[Optional["IngestMemoryIngestMemoryObservationsOccurrences"]]
+    ]
+    state: EntityState
+
+
+class IngestMemoryIngestMemoryObservationsObservable(BaseModel):
+    id: str
+    name: Optional[str]
+
+
+class IngestMemoryIngestMemoryObservationsRelated(BaseModel):
+    id: str
+    name: Optional[str]
+
+
+class IngestMemoryIngestMemoryObservationsOccurrences(BaseModel):
+    type: Optional[OccurrenceTypes]
+    confidence: Optional[float]
+    start_time: Optional[Any] = Field(alias="startTime")
+    end_time: Optional[Any] = Field(alias="endTime")
+    page_index: Optional[int] = Field(alias="pageIndex")
+    bounding_box: Optional[
+        "IngestMemoryIngestMemoryObservationsOccurrencesBoundingBox"
+    ] = Field(alias="boundingBox")
+
+
+class IngestMemoryIngestMemoryObservationsOccurrencesBoundingBox(BaseModel):
+    left: Optional[float]
+    top: Optional[float]
+    width: Optional[float]
+    height: Optional[float]
+
+
+IngestMemory.model_rebuild()
+IngestMemoryIngestMemory.model_rebuild()
+IngestMemoryIngestMemoryObservations.model_rebuild()
+IngestMemoryIngestMemoryObservationsOccurrences.model_rebuild()

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001}/graphlit_api/operations.py

@@ -191,6 +191,7 @@ __all__ = [
     "GET_WORKFLOW_GQL",
     "INGEST_BATCH_GQL",
     "INGEST_ENCODED_FILE_GQL",
+    "INGEST_MEMORY_GQL",
     "INGEST_TEXT_BATCH_GQL",
     "INGEST_TEXT_GQL",
     "INGEST_URI_GQL",
@@ -1591,8 +1592,60 @@ mutation IngestEncodedFile($name: String!, $data: String!, $mimeType: String!, $
 }
 """
 
+INGEST_MEMORY_GQL = """
+mutation IngestMemory($text: String!, $name: String, $textType: TextTypes, $collections: [EntityReferenceInput!], $correlationId: String) {
+  ingestMemory(
+    name: $name
+    text: $text
+    textType: $textType
+    collections: $collections
+    correlationId: $correlationId
+  ) {
+    id
+    name
+    state
+    type
+    fileType
+    mimeType
+    uri
+    collections {
+      id
+      name
+    }
+    observations {
+      id
+      type
+      observable {
+        id
+        name
+      }
+      related {
+        id
+        name
+      }
+      relatedType
+      relation
+      occurrences {
+        type
+        confidence
+        startTime
+        endTime
+        pageIndex
+        boundingBox {
+          left
+          top
+          width
+          height
+        }
+      }
+      state
+    }
+  }
+}
+"""
+
 INGEST_TEXT_GQL = """
-mutation IngestText($name: String!, $text: String!, $textType: TextTypes, $uri: URL, $id: ID, $isSynchronous: Boolean, $workflow: EntityReferenceInput, $collections: [EntityReferenceInput!], $observations: [ObservationReferenceInput!], $correlationId: String) {
+mutation IngestText($text: String!, $name: String, $textType: TextTypes, $uri: URL, $id: ID, $isSynchronous: Boolean, $workflow: EntityReferenceInput, $collections: [EntityReferenceInput!], $observations: [ObservationReferenceInput!], $correlationId: String) {
   ingestText(
     name: $name
     text: $text

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001/graphlit_client.egg-info}/PKG-INFO

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

{graphlit_client-1.0.20250410001 → graphlit_client-1.0.20250412001}/graphlit_client.egg-info/SOURCES.txt

@@ -199,6 +199,7 @@ graphlit_api/get_user.py
 graphlit_api/get_workflow.py
 graphlit_api/ingest_batch.py
 graphlit_api/ingest_encoded_file.py
+graphlit_api/ingest_memory.py
 graphlit_api/ingest_text.py
 graphlit_api/ingest_text_batch.py
 graphlit_api/ingest_uri.py