kodexa 7.4.415054195687__py3-none-any.whl → 7.4.415279448694__py3-none-any.whl

kodexa/dataclasses/__init__.py

@@ -318,9 +318,27 @@ class LLMDataObject(BaseModel):
             tag_uuid = str(uuid.uuid4())
             for node in nodes_to_label:
                 if node:
+                    confidence = -1 if value.value_path == 'DERIVED' else 1
+
+                    textract_confidence_score_feature = node.get_feature(
+                        "textract", "confidence"
+                    )
+                    if textract_confidence_score_feature is not None:
+                        textract_confidence_value = (
+                            textract_confidence_score_feature.value
+                        )
+
+                        if (
+                            textract_confidence_value
+                            and len(textract_confidence_value) > 0
+                        ):
+                            confidence = textract_confidence_value[0]
+                            confidence = confidence / 100
+                    
+                    logger.info(f"Confidence: {confidence}")
+                    
                     if not node.has_tag(tag):
-                        try:
-                            confidence = -1 if value.value_path == 'DERIVED' else 1
+                        try:                            
                             node.tag(
                                 tag_to_apply=tag,
                                 value=value.normalized_text,
@@ -339,7 +357,7 @@ class LLMDataObject(BaseModel):
                         new_tag = Tag(cell_index=self.cell_index,
                                       uuid=tag_uuid,
                                       value=value.normalized_text,
-                                      confidence=-1,
+                                      confidence=confidence,
                                       group_uuid=self.group_uuid,
                                       parent_group_uuid=parent_group_uuid,
                                       owner_uri=f"assistant://{assistant.id}" if assistant else f"model://taxonomy-llm")

kodexa/model/objects.py

@@ -4833,6 +4833,8 @@ class ModelInteraction(BaseModel):
     model_id: Optional[str] = Field(None, alias="modelId")
     input_tokens: Optional[int] = Field(None, alias="inputTokens")
     output_tokens: Optional[int] = Field(None, alias="outputTokens")
+    thinking_tokens: Optional[int] = Field(None, alias="thinkingTokens")
+    cached_tokens: Optional[int] = Field(None, alias="cachedTokens")
     duration: Optional[int] = None
     note: Optional[str] = None
 

{kodexa-7.4.415054195687.dist-info → kodexa-7.4.415279448694.dist-info}/METADATA

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

{kodexa-7.4.415054195687.dist-info → kodexa-7.4.415279448694.dist-info}/RECORD

@@ -3,7 +3,7 @@ kodexa/assistant/__init__.py,sha256=nlXm_YnV_50hgn0TIT2Fkc2fQ-86OjmctY_j8My9nc4,
 kodexa/assistant/assistant.py,sha256=5KFdbqFSLIZJyDRyZdpcfr448fT-CW4JhYu9A6B9DGY,14663
 kodexa/connectors/__init__.py,sha256=WF6G_MUeU32TlKSUKkpNoNX7dq8iBPliFMep4E8BmZc,328
 kodexa/connectors/connectors.py,sha256=FpUZDkSyHld2b9eYRuVOWzaFtuGoaRuPXXicJB7THbc,10413
-kodexa/dataclasses/__init__.py,sha256=CHMNsOamWA3gY5203gn8Ef5q1fgcczMtWKEvNjIOzPs,19486
+kodexa/dataclasses/__init__.py,sha256=TuSMJAG9rMr5FHjBk66yFnlRmCMvrkBVnD7bORRmi38,20287
 kodexa/dataclasses/templates/llm_data_class.j2,sha256=YWjStW136chV_59JM3AYis3i-0jdrqDvLXsISUW9zDU,660
 kodexa/model/__init__.py,sha256=rtLXYJBxB-rnukhslN9rlqoB3--1H3253HyHGbD_Gc8,796
 kodexa/model/base.py,sha256=CaZK8nMhT1LdCpt4aLhebJGcorjq9qRID1FjnXnP14M,521
@@ -13,7 +13,7 @@ kodexa/model/entities/product.py,sha256=StUhTEeLXmc05cj6XnZppQfeJsqCPbX1jdhsysHH
 kodexa/model/entities/product_group.py,sha256=540fRGyUf34h1BzAN1DiWu6rGgvaj3xDFhZ2k-RvSFY,3617
 kodexa/model/entities/product_subscription.py,sha256=UcmWR-qgLfdV7VCtJNwzgkanoS8nBSL6ngVuxQUK1M8,3810
 kodexa/model/model.py,sha256=q3zEm6pPOB-xPCKbOxmTMqLALzmQr2Ppam8knApoSEE,119645
-kodexa/model/objects.py,sha256=ZxoGPKgSxl7XHqTg24vnQUGzArMTEnQc7hGEwwN7ERE,200487
+kodexa/model/objects.py,sha256=aBiz7QL44IXNrUaq3EndVY7vnpiIYBMBzHTiYcx_KHA,200629
 kodexa/model/persistence.py,sha256=jUgQ8xwsAFIoZ_bEynxCDEWhUII42eN0e0Mum0dkQPg,72043
 kodexa/model/utils.py,sha256=YEG3f8YzBil06D0P3_dfx2S9GnHREzyrjnlWwQ7oPlU,3038
 kodexa/pipeline/__init__.py,sha256=sA7f5D6qkdMrpp2xTIeefnrUBI6xxEEWostvxfX_1Cs,236
@@ -45,7 +45,7 @@ kodexa/testing/test_utils.py,sha256=v44p__gE7ia67W7WeHN2HBFCWSCUrCZt7G4xBNCmwf8,
 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.4.415054195687.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-kodexa-7.4.415054195687.dist-info/METADATA,sha256=cGDYEHLSuNus18safzpEQ2nX1zD7eoaOycgDJ48_nRM,2916
-kodexa-7.4.415054195687.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-kodexa-7.4.415054195687.dist-info/RECORD,,
+kodexa-7.4.415279448694.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+kodexa-7.4.415279448694.dist-info/METADATA,sha256=K56e4Efv8K3ummtuVSlvfn7OABENsyUDKsJql-G81mA,2916
+kodexa-7.4.415279448694.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+kodexa-7.4.415279448694.dist-info/RECORD,,