trustgraph-base 1.3.7__tar.gz → 1.3.9__tar.gz

{trustgraph_base-1.3.7 → trustgraph_base-1.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-base
-Version: 1.3.7
+Version: 1.3.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-1.3.7 → trustgraph_base-1.3.9}/trustgraph/base/__init__.py

@@ -31,4 +31,5 @@ from . graph_rag_client import GraphRagClientSpec
 from . tool_service import ToolService
 from . tool_client import ToolClientSpec
 from . agent_client import AgentClientSpec
+from . structured_query_client import StructuredQueryClientSpec
 

trustgraph_base-1.3.9/trustgraph/base/structured_query_client.py

@@ -0,0 +1,33 @@
+from . request_response_spec import RequestResponse, RequestResponseSpec
+from .. schema import StructuredQueryRequest, StructuredQueryResponse
+
+class StructuredQueryClient(RequestResponse):
+    async def structured_query(self, question, timeout=600):
+        resp = await self.request(
+            StructuredQueryRequest(
+                question = question
+            ),
+            timeout=timeout
+        )
+
+        if resp.error:
+            raise RuntimeError(resp.error.message)
+
+        # Return the full response structure for the tool to handle
+        return {
+            "data": resp.data,
+            "errors": resp.errors if resp.errors else [],
+            "error": resp.error
+        }
+
+class StructuredQueryClientSpec(RequestResponseSpec):
+    def __init__(
+            self, request_name, response_name,
+    ):
+        super(StructuredQueryClientSpec, self).__init__(
+            request_name = request_name,
+            request_schema = StructuredQueryRequest,
+            response_name = response_name,
+            response_schema = StructuredQueryResponse,
+            impl = StructuredQueryClient,
+        )

trustgraph_base-1.3.9/trustgraph/base_version.py

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

{trustgraph_base-1.3.7 → trustgraph_base-1.3.9}/trustgraph_base.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-base
-Version: 1.3.7
+Version: 1.3.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-1.3.7 → trustgraph_base-1.3.9}/trustgraph_base.egg-info/SOURCES.txt

@@ -40,6 +40,7 @@ trustgraph/base/pubsub.py
 trustgraph/base/request_response_spec.py
 trustgraph/base/setting_spec.py
 trustgraph/base/spec.py
+trustgraph/base/structured_query_client.py
 trustgraph/base/subscriber.py
 trustgraph/base/subscriber_spec.py
 trustgraph/base/text_completion_client.py

trustgraph_base-1.3.7/trustgraph/base_version.py

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