trustgraph-base 2.7.7__tar.gz → 2.7.9__tar.gz

{trustgraph_base-2.7.7 → trustgraph_base-2.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-base
-Version: 2.7.7
+Version: 2.7.9
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_base-2.7.9/trustgraph/base_version.py

@@ -0,0 +1 @@
+__version__ = "2.7.9"

{trustgraph_base-2.7.7 → trustgraph_base-2.7.9}/trustgraph/messaging/translators/knowledge.py

@@ -1,3 +1,4 @@
+import logging
 from typing import Dict, Any, Tuple, Optional
 from ...schema import (
     KnowledgeRequest, KnowledgeResponse, Triples, GraphEmbeddings,
@@ -8,6 +9,8 @@ from ...schema import (
 from .base import MessageTranslator
 from .primitives import ValueTranslator, SubgraphTranslator
 
+logger = logging.getLogger(__name__)
+
 
 class KnowledgeRequestTranslator(MessageTranslator):
     """Translator for KnowledgeRequest schema objects"""
@@ -264,6 +267,15 @@ class KnowledgeResponseTranslator(MessageTranslator):
                 }
             }
 
+        # Error response
+        if obj.error:
+            return {
+                "error": {
+                    "type": obj.error.type,
+                    "message": obj.error.message,
+                }
+            }
+
         # End of stream marker
         if obj.eos is True:
             return {"eos": True}
@@ -274,14 +286,13 @@ class KnowledgeResponseTranslator(MessageTranslator):
     def encode_with_completion(self, obj: KnowledgeResponse) -> Tuple[Dict[str, Any], bool]:
         """Returns (response_dict, is_final)"""
         response = self.encode(obj)
-        
-        # Check if this is a final response
+
         is_final = (
-            obj.ids is not None or  # List response
-            obj.eos is True or      # End of stream
-            (not obj.triples and not obj.graph_embeddings
+            obj.ids is not None or
+            obj.eos is True or
+            (not obj.error and not obj.triples and not obj.graph_embeddings
              and not obj.document_embeddings
-             and not obj.library_metadata and not obj.library_blob)  # Empty response
+             and not obj.library_metadata and not obj.library_blob)
         )
-        
+
         return response, is_final

{trustgraph_base-2.7.7 → trustgraph_base-2.7.9}/trustgraph/schema/knowledge/knowledge.py

@@ -66,7 +66,7 @@ class KnowledgeRequest:
 @dataclass
 class KnowledgeResponse:
     error: Error | None = None
-    ids: list[str] = field(default_factory=list)
+    ids: list[str] | None = None
     eos: bool = False     # Indicates end of knowledge core stream
     triples: Triples | None = None
     graph_embeddings: GraphEmbeddings | None = None

{trustgraph_base-2.7.7 → trustgraph_base-2.7.9}/trustgraph_base.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-base
-Version: 2.7.7
+Version: 2.7.9
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_base-2.7.7/trustgraph/base_version.py

@@ -1 +0,0 @@
-__version__ = "2.7.7"