apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-03 20:32:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added doc embedding support (#41)

Browse source 
* document embedding writer & query
* Added test query for doc embeddings
* Bump version
* Added doc rag prompt
* Document RAG service
This commit is contained in:
cybermaggedon 2024-08-26 23:45:23 +01:00 • committed by GitHub
parent 0159e938a2
commit 669aed0f8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
52 changed files with 1257 additions and 252 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/test-doc-rag Executable file
View file

@ -0,0 +1,15 @@
#!/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)