graphlit-client 1.0.20250410001__py3-none-any.whl → 1.0.20250414001__py3-none-any.whl

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_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_api/enums.py

@@ -28,6 +28,12 @@ class OpenAIModels(str, Enum):
     GPT4O_MINI_128K_20240718 = "GPT4O_MINI_128K_20240718"
     GPT4O_MINI_128K = "GPT4O_MINI_128K"
     GPT4O_CHAT_128K = "GPT4O_CHAT_128K"
+    GPT41_NANO_1024K = "GPT41_NANO_1024K"
+    GPT41_NANO_1024K_20250414 = "GPT41_NANO_1024K_20250414"
+    GPT41_MINI_1024K = "GPT41_MINI_1024K"
+    GPT41_MINI_1024K_20250414 = "GPT41_MINI_1024K_20250414"
+    GPT41_1024K = "GPT41_1024K"
+    GPT41_1024K_20250414 = "GPT41_1024K_20250414"
     GPT45_PREVIEW_128K = "GPT45_PREVIEW_128K"
     GPT45_PREVIEW_128K_20250227 = "GPT45_PREVIEW_128K_20250227"
     O1_MINI_128K = "O1_MINI_128K"
@@ -179,6 +185,7 @@ class ContentTypes(str, Enum):
     EMAIL = "EMAIL"
     EVENT = "EVENT"
     ISSUE = "ISSUE"
+    MEMORY = "MEMORY"
 
 
 class CerebrasModels(str, Enum):

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_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.dist-info → graphlit_client-1.0.20250414001.dist-info}/METADATA

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

{graphlit_client-1.0.20250410001.dist-info → graphlit_client-1.0.20250414001.dist-info}/RECORD

@@ -1,12 +1,12 @@
 graphlit/__init__.py,sha256=4AyigTlFQWP40lnaaQ1H1iRT_B1hIXW9bgPanbwmTvs,32
 graphlit/graphlit.py,sha256=56G5Am72og5QHIjEh9cy4vkfos7YVpGvzRuIPhPbCEU,1949
-graphlit_api/__init__.py,sha256=-1ZQyXGLQw3q1jxNQpdV4HuCR8HVGvt6b5uTeEbAX_I,187154
+graphlit_api/__init__.py,sha256=bKiIjFGuDAcGHvPGsM9jJ4rNXtguE-gaaPM49CHxycY,187947
 graphlit_api/add_contents_to_collections.py,sha256=K7tNpLn8-lRVaVT39iKr-VtCKRWVONyL_h6cC0L606Y,888
 graphlit_api/ask_graphlit.py,sha256=F-mgXMfUqGYCOP4WFQXmZDp5RgK7Cej-nA1RkmRvp7o,6301
 graphlit_api/async_base_client.py,sha256=v0KUVwe2_RIQa8Mn7l_yD5McUe7B03vhclJ9SP4XGgw,12578
 graphlit_api/base_model.py,sha256=o2d-DixASFCGztr3rTiGX0AwgFu7Awr7EgD70FI8a-I,620
 graphlit_api/clear_conversation.py,sha256=5GOmc2wfupV-7EHWyi3v6LA0pSVLtFNCzxPJm42Dp6Y,531
-graphlit_api/client.py,sha256=8TyiLe0_zluZ_92ab5veKltZeodamiTgQAGBc6Ma58w,199849
+graphlit_api/client.py,sha256=qEFAYuk-3ugdImWf5qkU22sW9Cz4pPjNbYQ9SJMf1-k,200821
 graphlit_api/close_conversation.py,sha256=HcIUUiNf7hnuLZ7Fy6IcgfuHMSyWyJ7uOEy1EEETy_4,531
 graphlit_api/complete_conversation.py,sha256=9YmZZ6Gs7MkApNP1eu8tXG4wqOd3FWgPMF6BiJOGo3o,16468
 graphlit_api/continue_conversation.py,sha256=wy_ARPwt_mmmwianxaHCNvK53p_Sbeqkwnoo9d2nTco,16468
@@ -157,7 +157,7 @@ graphlit_api/disable_user.py,sha256=qfFXMq691gPVdFjF-mYzjcx6DIvZnsp5Nny5pl0qI9Q,
 graphlit_api/enable_alert.py,sha256=KctjIbnyZDoDynbuD6IR2fHuNHKAtM07VVgKzTL479g,398
 graphlit_api/enable_feed.py,sha256=BhtLlEGV9p8ICodywdN9-sASWSX3w_4c3zYtf-bdaGE,390
 graphlit_api/enable_user.py,sha256=_0rQnpwVYhBrRcnpfqrKTOiANgOMe-ibUAHmS-XC0qo,390
-graphlit_api/enums.py,sha256=5qzUsuh3w28FXBdzNE9TU-bwaM7FcSivYZta2JCpK4o,26784
+graphlit_api/enums.py,sha256=6uSO4hrV4So_qIKVSACZ1xvcO0qa5vdgLHnIdqYA0sE,27092
 graphlit_api/exceptions.py,sha256=33pJ_dDonmIx_D-43jvhuQYJAXz5HoaPf1DoxXmNsAw,2432
 graphlit_api/extract_contents.py,sha256=qsSy8UOJuqsUnCcuduaNPyaK2mZk_1noECT0Jgx3zRA,961
 graphlit_api/extract_text.py,sha256=jfkL6ep5wzXeoelZ5nRwNRT0t7oDCUt1xrigqkMj-7M,887
@@ -195,6 +195,7 @@ graphlit_api/get_user.py,sha256=pdfR920veI3hTBWqbIp8PP_1LCYi8KWTJ2Has8t8URM,2569
 graphlit_api/get_workflow.py,sha256=BWKzz03q51AmFnE0Cvxkm0WqIS2Es42x67r6KZxHuxo,14478
 graphlit_api/ingest_batch.py,sha256=pmO_rAZdG8dPid40h8lnTfKSa5r0EAOmFF7PIg3a_r4,2366
 graphlit_api/ingest_encoded_file.py,sha256=mKoEc5qziw8i-MDT8CrGCfmaupWnIVQkow7cRW_Y3Fw,2607
+graphlit_api/ingest_memory.py,sha256=YF7sn_jvpk_iACg8encyp_gd0wrK0Om4blYzPDI-B8c,2374
 graphlit_api/ingest_text.py,sha256=D4lpV9LTC_u586_ILVrB2rVpHG0-8HivqeOA1GpQuFs,2286
 graphlit_api/ingest_text_batch.py,sha256=gSD1bH3mAPwJzy5TeMJ6UguEgI9yrPUXyz1soExSttM,2521
 graphlit_api/ingest_uri.py,sha256=f71kMRyMoAhef6sJU85ZgGz4aPq_5CDLaDvCeQnLY5A,2248
@@ -204,7 +205,7 @@ graphlit_api/is_feed_done.py,sha256=-FQS2vtDMnNW75K_9jR6IUutvpwLmtoS5yY8wD17CaM,
 graphlit_api/lookup_credits.py,sha256=xisQ4ExnjwWGbPZVU24bLF7pj4Dwp-d9q5GsHVBMoAM,1395
 graphlit_api/lookup_usage.py,sha256=D_cbO0KmXDqRYqQIhNwWXNFGjwNLEy2_5aVa-SYgRzw,1711
 graphlit_api/map_web.py,sha256=2rp4jFD1vDqcQ98mCVTeC0RzPqQxmmcRvHNRl8HJfFA,346
-graphlit_api/operations.py,sha256=2JgFjOWO1trZyVXV8z7GBmXqjZZHUZPr3fnUBgK0bnA,184033
+graphlit_api/operations.py,sha256=BoNi9xxIIzNM-ZRB2J5LNENSkXCZgRZ8S5DStl5dJ4s,184882
 graphlit_api/prompt.py,sha256=4A0P-8hk98sNFNfRovzDBXXCHAGXOs2zThsXY07Q5KI,6015
 graphlit_api/prompt_conversation.py,sha256=zdVzHhZdF2CFL87_8EUcugWYzSFSWc0nCv4JFjhN7lI,16082
 graphlit_api/prompt_specifications.py,sha256=z8JQpfCUfgPeiuolLEGY83Pr32VaI2Ng1Wq0pG-gv2w,6902
@@ -301,8 +302,8 @@ graphlit_api/upsert_label.py,sha256=_bVWrISvyt4G4IcjAKqu8c5P6FDgaODdIGtSToJfNOY,
 graphlit_api/upsert_specification.py,sha256=23eLTL8OLAYE-j_nhjT5NgaCrSUs9Q40rGW_VhDrDoM,643
 graphlit_api/upsert_workflow.py,sha256=QL5Gi8jqF4wG9_qReoW8E5X7VsSIsLzgxIY1aAK0ebo,15692
 graphlit_api/workflow_exists.py,sha256=1XVcqCW_KZ3BwUFx08lwqQdf1ZpJ6Vmi8jBqcrMqYRI,397
-graphlit_client-1.0.20250410001.dist-info/licenses/LICENSE,sha256=ivF8XnUYrNZFQ1wZFMrxWshDb1h7TdSK6Qk8_3WPkhM,1095
-graphlit_client-1.0.20250410001.dist-info/METADATA,sha256=0llo-GvUs9pJXqPwJzSk2GBm7wS1u8ZmBoGyTcyZx_I,3408
-graphlit_client-1.0.20250410001.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graphlit_client-1.0.20250410001.dist-info/top_level.txt,sha256=HUVfNzJrxWuHS-4M5I7XjLa8-mxYQwfx01A4YKJZSYM,22
-graphlit_client-1.0.20250410001.dist-info/RECORD,,
+graphlit_client-1.0.20250414001.dist-info/licenses/LICENSE,sha256=ivF8XnUYrNZFQ1wZFMrxWshDb1h7TdSK6Qk8_3WPkhM,1095
+graphlit_client-1.0.20250414001.dist-info/METADATA,sha256=37pMsc-kbYVFbO1M2dACUwgQsb8Oc79vinvGKW50ZLk,3408
+graphlit_client-1.0.20250414001.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graphlit_client-1.0.20250414001.dist-info/top_level.txt,sha256=HUVfNzJrxWuHS-4M5I7XjLa8-mxYQwfx01A4YKJZSYM,22
+graphlit_client-1.0.20250414001.dist-info/RECORD,,