mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-27 09:26:22 +02:00
669aed0f8a
* document embedding writer & query * Added test query for doc embeddings * Bump version * Added doc rag prompt * Document RAG service
15 lines
344 B
Python
Executable file
15 lines
344 B
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import pulsar
|
|
from trustgraph.clients.document_rag_client import DocumentRagClient
|
|
|
|
rag = DocumentRagClient(pulsar_host="pulsar://localhost:6650")
|
|
|
|
query="""
|
|
This knowledge graph describes the Space Shuttle disaster.
|
|
Present 20 facts which are present in the knowledge graph."""
|
|
|
|
resp = rag.request(query)
|
|
|
|
print(resp)
|
|
|