apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-21 19:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Document RAG

Browse source
This commit is contained in:
Cyber MacGeddon 2025-05-01 09:36:50 +01:00
parent 450f664b1b
commit 2b8ac5b9bc
2 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

13
trustgraph-flow/trustgraph/gateway/document_rag.py
View file

@ -1,20 +1,23 @@
from .. schema import DocumentRagQuery, DocumentRagResponse from .. schema import DocumentRagQuery, DocumentRagResponse
from .. schema import document_rag_request_queue
from .. schema import document_rag_response_queue
from . endpoint import ServiceEndpoint from . endpoint import ServiceEndpoint
from . requestor import ServiceRequestor from . requestor import ServiceRequestor
class DocumentRagRequestor(ServiceRequestor): class DocumentRagRequestor(ServiceRequestor):
def __init__(self, pulsar_client, timeout, auth): def __init__(
self, pulsar_client, request_queue, response_queue, timeout, auth,
consumer, subscriber,
):
super(DocumentRagRequestor, self).__init__( super(DocumentRagRequestor, self).__init__(
pulsar_client=pulsar_client, pulsar_client=pulsar_client,
request_queue=document_rag_request_queue, request_queue=request_queue,
response_queue=document_rag_response_queue, response_queue=response_queue,
request_schema=DocumentRagQuery, request_schema=DocumentRagQuery,
response_schema=DocumentRagResponse, response_schema=DocumentRagResponse,
subscription = subscriber,
consumer_name = consumer,
timeout=timeout, timeout=timeout,
) )

21
trustgraph-flow/trustgraph/gateway/service.py
View file

@ -33,7 +33,7 @@ from .. schema import ConfigPush, config_push_queue
from . text_completion import TextCompletionRequestor from . text_completion import TextCompletionRequestor
from . prompt import PromptRequestor from . prompt import PromptRequestor
from . graph_rag import GraphRagRequestor from . graph_rag import GraphRagRequestor
#from . document_rag import DocumentRagRequestor from . document_rag import DocumentRagRequestor
from . triples_query import TriplesQueryRequestor from . triples_query import TriplesQueryRequestor
from . graph_embeddings_query import GraphEmbeddingsQueryRequestor from . graph_embeddings_query import GraphEmbeddingsQueryRequestor
from . embeddings import EmbeddingsRequestor from . embeddings import EmbeddingsRequestor
@ -41,7 +41,7 @@ from . embeddings import EmbeddingsRequestor
from . agent import AgentRequestor from . agent import AgentRequestor
#from . dbpedia import DbpediaRequestor #from . dbpedia import DbpediaRequestor
#from . internet_search import InternetSearchRequestor #from . internet_search import InternetSearchRequestor
#from . librarian import LibrarianRequestor from . librarian import LibrarianRequestor
from . config import ConfigRequestor from . config import ConfigRequestor
from . flow import FlowRequestor from . flow import FlowRequestor
#from . triples_stream import TriplesStreamEndpoint #from . triples_stream import TriplesStreamEndpoint
@ -141,10 +141,10 @@ class Api:
# pulsar_client=self.pulsar_client, timeout=self.timeout, # pulsar_client=self.pulsar_client, timeout=self.timeout,
# auth = self.auth, # auth = self.auth,
# ), # ),
# "librarian": LibrarianRequestor( (None, "librarian"): LibrarianRequestor(
# pulsar_client=self.pulsar_client, timeout=self.timeout, pulsar_client=self.pulsar_client, timeout=self.timeout,
# auth = self.auth, auth = self.auth,
# ), ),
(None, "config"): ConfigRequestor( (None, "config"): ConfigRequestor(
pulsar_client=self.pulsar_client, timeout=self.timeout, pulsar_client=self.pulsar_client, timeout=self.timeout,
auth = self.auth, auth = self.auth,
@ -207,10 +207,10 @@ class Api:
# endpoint_path = "/api/v1/agent", auth=self.auth, # endpoint_path = "/api/v1/agent", auth=self.auth,
# requestor = self.services["agent"], # requestor = self.services["agent"],
# ), # ),
# ServiceEndpoint( ServiceEndpoint(
# endpoint_path = "/api/v1/librarian", auth=self.auth, endpoint_path = "/api/v1/librarian", auth=self.auth,
# requestor = self.services["librarian"], requestor = self.services[(None, "librarian")],
# ), ),
ServiceEndpoint( ServiceEndpoint(
endpoint_path = "/api/v1/config", auth=self.auth, endpoint_path = "/api/v1/config", auth=self.auth,
requestor = self.services[(None, "config")], requestor = self.services[(None, "config")],
@ -320,6 +320,7 @@ class Api:
"text-completion": TextCompletionRequestor, "text-completion": TextCompletionRequestor,
"prompt": PromptRequestor, "prompt": PromptRequestor,
"graph-rag": GraphRagRequestor, "graph-rag": GraphRagRequestor,
"document-rag": DocumentRagRequestor,
"embeddings": EmbeddingsRequestor, "embeddings": EmbeddingsRequestor,
"graph-embeddings": GraphEmbeddingsQueryRequestor, "graph-embeddings": GraphEmbeddingsQueryRequestor,
"triples-query": TriplesQueryRequestor, "triples-query": TriplesQueryRequestor,