kodexa 7.0.11272958808__tar.gz → 7.0.11276748448__tar.gz

{kodexa-7.0.11272958808 → kodexa-7.0.11276748448}/PKG-INFO

@@ -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 → kodexa-7.0.11276748448}/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 → kodexa-7.0.11276748448}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kodexa"
-version = "7.0.011272958808"
+version = "7.0.011276748448"
 description = "Python SDK for the Kodexa Platform"
 authors = ["Austin Redenbaugh <austin@kodexa.com>", "Philip Dodds <philip@kodexa.com>", "Romar Cablao <rcablao@kodexa.com>", "Amadea Paula Dodds <amadeapaula@kodexa.com>"]
 readme = "README.md"