trustgraph-base 0.23.17__tar.gz → 0.23.19__tar.gz

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 0.23.17
+Version: 0.23.19
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.23.17.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.23.19.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/trustgraph/api/knowledge.py

@@ -39,3 +39,29 @@ class Knowledge:
 
         self.request(request = input)
 
+    def load_kg_core(self, id, user="trustgraph", flow="0000",
+                     collection="default"):
+
+        # The input consists of system and prompt strings
+        input = {
+            "operation": "load-kg-core",
+            "user": user,
+            "id": id,
+            "flow": flow,
+            "collection": collection,
+        }
+
+        self.request(request = input)
+
+    def unload_kg_core(self, id, user="trustgraph", flow="0000"):
+
+        # The input consists of system and prompt strings
+        input = {
+            "operation": "unload-kg-core",
+            "user": user,
+            "id": id,
+            "flow": flow,
+        }
+
+        self.request(request = input)
+

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/trustgraph/base/consumer.py

@@ -45,7 +45,9 @@ class Consumer:
 
         if hasattr(self, "consumer"):
             if self.consumer:
+                self.consumer.unsubscribe()
                 self.consumer.close()
+                self.consumer = None
 
     async def stop(self):
 
@@ -108,11 +110,16 @@ class Consumer:
             except Exception as e:
 
                 print("consumer loop exception:", e, flush=True)
+                self.consumer.unsubscribe()
                 self.consumer.close()
                 self.consumer = None
                 await asyncio.sleep(self.reconnect_time)
                 continue
 
+        if self.consumer:
+            self.consumer.unsubscribe()
+            self.consumer.close()
+
     async def consume(self):
 
         while self.running:

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/trustgraph/base/subscriber.py

@@ -21,10 +21,21 @@ class Subscriber:
         self.metrics = metrics
         self.task = None
 
+        self.consumer = None
+
     def __del__(self):
+
         self.running = False
 
     async def start(self):
+
+        self.consumer = self.client.subscribe(
+            topic = self.topic,
+            subscription_name = self.subscription,
+            consumer_name = self.consumer_name,
+            schema = JsonSchema(self.schema),
+        )
+
         self.task = asyncio.create_task(self.run())
 
     async def stop(self):
@@ -41,8 +52,6 @@ class Subscriber:
 
     async def run(self):
 
-        consumer = None
-
         while self.running:
 
             if self.metrics:
@@ -50,15 +59,6 @@ class Subscriber:
 
             try:
 
-                # FIXME: Create consumer in start  method so we know
-                # it is definitely running when start completes
-                consumer = self.client.subscribe(
-                    topic = self.topic,
-                    subscription_name = self.subscription,
-                    consumer_name = self.consumer_name,
-                    schema = JsonSchema(self.schema),
-                )
-
                 if self.metrics:
                     self.metrics.state("running")
 
@@ -68,7 +68,7 @@ class Subscriber:
 
                     try:
                         msg = await asyncio.to_thread(
-                            consumer.receive,
+                            self.consumer.receive,
                             timeout_millis=250
                         )
                     except _pulsar.Timeout:
@@ -82,7 +82,7 @@ class Subscriber:
                         self.metrics.received()
 
                     # Acknowledge successful reception of the message
-                    consumer.acknowledge(msg)
+                    self.consumer.acknowledge(msg)
 
                     try:
                         id = msg.properties()["id"]
@@ -124,9 +124,10 @@ class Subscriber:
 
             finally:
 
-                if consumer:
-                    consumer.close()
-                    consumer = None
+                if self.consumer:
+                    self.consumer.unsubscribe()
+                    self.consumer.close()
+                    self.consumer = None
                 
          
             if self.metrics:

trustgraph-base-0.23.19/trustgraph/base_version.py

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

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/trustgraph/schema/knowledge.py

@@ -25,14 +25,22 @@ from . graph import Triples, GraphEmbeddings
 class KnowledgeRequest(Record):
 
     # get-kg-core, delete-kg-core, list-kg-cores, put-kg-core
+    # load-kg-core, unload-kg-core
     operation = String()
 
     # list-kg-cores, delete-kg-core, put-kg-core
     user = String()
 
-    # get-kg-core, list-kg-cores, delete-kg-core, put-kg-core
+    # get-kg-core, list-kg-cores, delete-kg-core, put-kg-core,
+    # load-kg-core, unload-kg-core
     id = String()
 
+    # load-kg-core
+    flow = String()
+
+    # load-kg-core
+    collection = String()
+
     # put-kg-core
     triples = Triples()
     graph_embeddings = GraphEmbeddings()

{trustgraph-base-0.23.17 → trustgraph-base-0.23.19}/trustgraph_base.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 0.23.17
+Version: 0.23.19
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.23.17.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.23.19.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

trustgraph-base-0.23.17/trustgraph/base_version.py

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