tg-put-kg-core command (#369)

trustgraph-flow/trustgraph/cores/knowledge.py

@@ -36,9 +36,9 @@ class KnowledgeManager:
             )
         )
 
-    async def fetch_kg_core(self, request, respond):
+    async def get_kg_core(self, request, respond):
 
-        print("Fetch core...", flush=True)
+        print("Get core...", flush=True)
 
         async def publish_triples(t):
             await respond(
@@ -76,7 +76,7 @@ class KnowledgeManager:
             publish_ge,
         )
 
-        print("Fetch complete", flush=True)
+        print("Get complete", flush=True)
 
         await respond(
             KnowledgeResponse(
@@ -102,3 +102,23 @@ class KnowledgeManager:
             )
         )
 
+    async def put_kg_core(self, request, respond):
+
+        if request.triples:
+            await self.table_store.add_triples(request.triples)
+
+        if request.graph_embeddings:
+            await self.table_store.add_graph_embeddings(
+                request.graph_embeddings
+            )
+
+        await respond(
+            KnowledgeResponse(
+                error = None,
+                ids = None,
+                eos = False,
+                triples = None,
+                graph_embeddings = None
+            )
+        )
+

trustgraph-flow/trustgraph/cores/service.py

@@ -125,8 +125,9 @@ class Processor(AsyncProcessor):
 
         impls = {
             "list-kg-cores": self.knowledge.list_kg_cores,
-            "fetch-kg-core": self.knowledge.fetch_kg_core,
+            "get-kg-core": self.knowledge.get_kg_core,
             "delete-kg-core": self.knowledge.delete_kg_core,
+            "put-kg-core": self.knowledge.put_kg_core,
         }
 
         if v.operation not in impls:
@@ -150,12 +151,10 @@ class Processor(AsyncProcessor):
 
         try:
 
+            # We don't send a response back here, the processing
+            # implementation sends whatever it needs to send.
             await self.process_request(v, id)
 
-#            await self.knowledge_response_producer.send(
-#                resp, properties={"id": id}
-#            )
-
             return
 
         except RequestError as e: