kodexa 7.4.415054195687__tar.gz → 7.4.415279448694__tar.gz

{kodexa-7.4.415054195687 → kodexa-7.4.415279448694}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kodexa"
-version = "7.4.415054195687"
+version = "7.4.415279448694"
 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"