trustgraph-flow 0.19.5__tar.gz → 0.19.6__tar.gz

{trustgraph-flow-0.19.5 → trustgraph-flow-0.19.6}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.19.5
+Version: 0.19.6
 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
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 License: UNKNOWN
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.19.5.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.19.6.tar.gz
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)

trustgraph-flow-0.19.6/trustgraph/flow_version.py

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

trustgraph-flow-0.19.6/trustgraph/gateway/document_rag.py

@@ -0,0 +1,30 @@
+
+from .. schema import DocumentRagQuery, DocumentRagResponse
+from .. schema import document_rag_request_queue
+from .. schema import document_rag_response_queue
+
+from . endpoint import ServiceEndpoint
+from . requestor import ServiceRequestor
+
+class DocumentRagRequestor(ServiceRequestor):
+    def __init__(self, pulsar_host, timeout, auth):
+
+        super(DocumentRagRequestor, self).__init__(
+            pulsar_host=pulsar_host,
+            request_queue=document_rag_request_queue,
+            response_queue=document_rag_response_queue,
+            request_schema=DocumentRagQuery,
+            response_schema=DocumentRagResponse,
+            timeout=timeout,
+        )
+
+    def to_request(self, body):
+        return DocumentRagQuery(
+            query=body["query"],
+            user=body.get("user", "trustgraph"),
+            collection=body.get("collection", "default"),
+        )
+
+    def from_response(self, message):
+        return { "response": message.response }, True
+

{trustgraph-flow-0.19.5 → trustgraph-flow-0.19.6}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.19.5
+Version: 0.19.6
 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
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 License: UNKNOWN
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.19.5.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.19.6.tar.gz
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)

{trustgraph-flow-0.19.5 → trustgraph-flow-0.19.6}/trustgraph_flow.egg-info/SOURCES.txt

@@ -111,6 +111,7 @@ trustgraph/gateway/agent.py
 trustgraph/gateway/auth.py
 trustgraph/gateway/dbpedia.py
 trustgraph/gateway/document_load.py
+trustgraph/gateway/document_rag.py
 trustgraph/gateway/embeddings.py
 trustgraph/gateway/encyclopedia.py
 trustgraph/gateway/endpoint.py

trustgraph-flow-0.19.5/trustgraph/flow_version.py

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