trustgraph-base 1.1.6__tar.gz → 1.1.7__tar.gz

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 1.1.6
+Version: 1.1.7
 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/v1.1.6.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.1.7.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/trustgraph/api/config.py

@@ -49,6 +49,19 @@ class Config:
 
         self.request(input)
 
+    def delete(self, keys):
+
+        # The input consists of system and prompt strings
+        input = {
+            "operation": "delete",
+            "keys": [
+                { "type": v.type, "key": v.key }
+                for v in keys
+            ]
+        }
+
+        self.request(input)
+
     def list(self, type):
 
         # The input consists of system and prompt strings
@@ -67,7 +80,7 @@ class Config:
             "type": type,
         }
 
-        object = self.request(input)["directory"]
+        object = self.request(input)
 
         try:
             return [

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/trustgraph/api/flow.py

@@ -4,6 +4,7 @@ import base64
 
 from .. knowledge import hash, Uri, Literal
 from . types import Triple
+from . exceptions import ProtocolException
 
 def to_value(x):
     if x["e"]: return Uri(x["v"])
@@ -197,7 +198,6 @@ class FlowInstance:
 
     def prompt(self, id, variables):
 
-        # The input consists of system and prompt strings
         input = {
             "id": id,
             "variables": variables
@@ -221,12 +221,37 @@ class FlowInstance:
 
         raise ProtocolException("Response not formatted correctly")
 
+    def mcp_tool(self, name, parameters={}):
+
+        # The input consists of name and parameters
+        input = {
+            "name": name,
+            "parameters": parameters,
+        }
+
+        object = self.request(
+            "service/mcp-tool",
+            input
+        )
+
+        if "text" in object:
+            return object["text"]
+
+        if "object" in object:
+            try:
+                return object["object"]
+            except Exception as e:
+                raise ProtocolException(
+                    "Returned object not well-formed JSON"
+                )
+
+        raise ProtocolException("Response not formatted correctly")
+
     def triples_query(
             self, s=None, p=None, o=None,
             user=None, collection=None, limit=10000
     ):
 
-        # The input consists of system and prompt strings
         input = {
             "limit": limit
         }

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/trustgraph/base/__init__.py

@@ -29,4 +29,5 @@ from . document_embeddings_client import DocumentEmbeddingsClientSpec
 from . agent_service import AgentService
 from . graph_rag_client import GraphRagClientSpec
 from . tool_service import ToolService
+from . tool_client import ToolClientSpec
 

trustgraph-base-1.1.7/trustgraph/base/tool_client.py

@@ -0,0 +1,40 @@
+
+import json
+
+from . request_response_spec import RequestResponse, RequestResponseSpec
+from .. schema import ToolRequest, ToolResponse
+
+class ToolClient(RequestResponse):
+
+    async def invoke(self, name, parameters={}, timeout=600):
+
+        if parameters is None:
+            parameters = {}
+
+        resp = await self.request(
+            ToolRequest(
+                name = name,
+                parameters = json.dumps(parameters),
+            ),
+            timeout=timeout
+        )
+
+        if resp.error:
+            raise RuntimeError(resp.error.message)
+
+        if resp.text: return resp.text
+
+        return json.loads(resp.object)
+
+class ToolClientSpec(RequestResponseSpec):
+    def __init__(
+            self, request_name, response_name,
+    ):
+        super(ToolClientSpec, self).__init__(
+            request_name = request_name,
+            request_schema = ToolRequest,
+            response_name = response_name,
+            response_schema = ToolResponse,
+            impl = ToolClient,
+        )
+

trustgraph-base-1.1.7/trustgraph/base_version.py

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

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/trustgraph_base.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 1.1.6
+Version: 1.1.7
 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/v1.1.6.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.1.7.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

{trustgraph-base-1.1.6 → trustgraph-base-1.1.7}/trustgraph_base.egg-info/SOURCES.txt

@@ -42,6 +42,7 @@ trustgraph/base/spec.py
 trustgraph/base/subscriber.py
 trustgraph/base/subscriber_spec.py
 trustgraph/base/text_completion_client.py
+trustgraph/base/tool_client.py
 trustgraph/base/tool_service.py
 trustgraph/base/triples_client.py
 trustgraph/base/triples_query_service.py

trustgraph-base-1.1.6/trustgraph/base_version.py

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