kodexa 7.0.11272958808__py3-none-any.whl → 7.0.11276748448__py3-none-any.whl

kodexa/platform/kodexa.py

@@ -16,6 +16,7 @@ from typing import Dict
 
 import requests
 from appdirs import AppDirs
+from mistune.plugins.formatting import insert
 
 from kodexa.connectors import get_source
 from kodexa.connectors.connectors import get_caller_dir, FolderConnector
@@ -29,7 +30,7 @@ from kodexa.model.objects import (
     DocumentFamilyEvent,
     ChannelEvent,
     DataObjectEvent,
-    WorkspaceEvent,
+    WorkspaceEvent, DocumentFamily,
 )
 from kodexa.pipeline import PipelineContext, PipelineStatistics
 from kodexa.platform.client import DocumentStoreEndpoint, KodexaClient, process_response
@@ -884,7 +885,7 @@ class EventHelper:
             ContentObject: The posted content object.
         """
         files = {"content": content}
-        data = {"contentObjectJson": json.dumps(content_object.dict(by_alias=True))}
+        data = {"contentObjectJson": json.dumps(content_object.model_dump(by_alias=True))}
         logger.info("Posting back content object to execution object")
         co_response = requests.post(
             f"{KodexaPlatform.get_url()}/api/sessions/{self.event.session_id}/executions/{self.event.execution.id}/objects",
@@ -901,7 +902,7 @@ class EventHelper:
 
         return ContentObject.model_validate(co_response.json())
 
-    def build_pipeline_context(self) -> PipelineContext:
+    def build_pipeline_context(self, event) -> PipelineContext:
         """Builds a pipeline context.
 
         Returns:
@@ -911,16 +912,25 @@ class EventHelper:
             context={}, content_provider=self, execution_id=self.event.execution.id
         )
 
-        if self.event.store_ref and self.event.document_family_id:
-            logger.info("We have storeRef and document family")
-            rds: DocumentStoreEndpoint = KodexaClient().get_object_by_ref(
-                "store", self.event.store_ref
-            )
-            document_family = rds.get_family(self.event.document_family_id)
-
-            context.document_family = document_family
-            context.document_store = rds
-
+        if isinstance(event, dict):
+            event = self.get_base_event(event)
+
+        if isinstance(event, DocumentFamilyEvent):
+            # Can we get the document family
+            dfe:DocumentFamilyEvent = self.event
+            if dfe.document_family:
+                logger.info(f"Setting document family for context: {dfe.document_family}")
+                context.document_family = dfe.document_family
+                logger.info(f"Getting document store for family: {context.document_family.store_ref}")
+                context.document_store = KodexaClient().get_object_by_ref("store", context.document_family.store_ref)
+        if isinstance(event, ContentEvent):
+            ce:ContentEvent = self.event
+            if ce.content_object:
+                logger.info(f"Setting content object for context: {ce.content_object}")
+                context.content_object = ce.content_object
+                logger.info(f"Getting document store for content object: {context.content_object.store_ref}")
+                context.document_store = KodexaClient().get_object_by_ref("store", context.content_object.store_ref)
+        logger.info("Returning context")
         return context
 
     def get_input_document(self, context):

{kodexa-7.0.11272958808.dist-info → kodexa-7.0.11276748448.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kodexa
-Version: 7.0.11272958808
+Version: 7.0.11276748448
 Summary: Python SDK for the Kodexa Platform
 Author: Austin Redenbaugh
 Author-email: austin@kodexa.com

{kodexa-7.0.11272958808.dist-info → kodexa-7.0.11276748448.dist-info}/RECORD

@@ -19,7 +19,7 @@ kodexa/pipeline/pipeline.py,sha256=ZYpJAWcwV4YRK589DUhU0vXGQlkNSj4J2TsGbYqTLjo,2
 kodexa/platform/__init__.py,sha256=1O3oiWMg292NPL_NacKDnK1T3_R6cMorrPRue_9e-O4,216
 kodexa/platform/client.py,sha256=VqEx7mwspekMFkOi64aOiLozhB2t_Uk68JI64Uq8BrM,226549
 kodexa/platform/interaction.py,sha256=6zpcwXKNZstUGNS6m4JsoRXAqCZPJHWI-ZN3co8nnF0,1055
-kodexa/platform/kodexa.py,sha256=RVtArtzn1TVhBf3NGC4xe-ZO-VEVDIC3RjLhR06p-uU,34033
+kodexa/platform/kodexa.py,sha256=02n9GEsshJLcqDQbFTRwFBfVB1S8UlZh_1jwmMGXW0Q,34842
 kodexa/selectors/__init__.py,sha256=xA9-4vpyaAZWPSk3bh2kvDLkdv6XEmm7PjFbpziiTIk,100
 kodexa/selectors/ast.py,sha256=gG-1st841IntgBE5V7p3Cq9azaym2jV5lB_AIywQTCI,13269
 kodexa/selectors/core.py,sha256=kkt02DN20gXeaDGoGubPPeeTV7rCr4sxTyELrI0l1YU,3691
@@ -42,7 +42,7 @@ kodexa/testing/test_utils.py,sha256=DrLCkHxdb6AbZ-X3WmTMbQmnVIm55VEBL8MjtUK9POs,
 kodexa/training/__init__.py,sha256=xs2L62YpRkIRfslQwtQZ5Yxjhm7sLzX2TrVX6EuBnZQ,52
 kodexa/training/train_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kodexa/utils/__init__.py,sha256=Pnim1o9_db5YEnNvDTxpM7HG-qTlL6n8JwFwOafU9wo,5928
-kodexa-7.0.11272958808.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-kodexa-7.0.11272958808.dist-info/METADATA,sha256=Yg41TCBZVB42Z8xOt8N4IYzqOgyMaSP54Ywc21wB-Zc,3529
-kodexa-7.0.11272958808.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-kodexa-7.0.11272958808.dist-info/RECORD,,
+kodexa-7.0.11276748448.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+kodexa-7.0.11276748448.dist-info/METADATA,sha256=HNHDyiBjTflGPS_x4S4uhDWn-CrwgMN0mpbA-GrhxoY,3529
+kodexa-7.0.11276748448.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+kodexa-7.0.11276748448.dist-info/RECORD,,