apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-03 04:12:37 +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

22
tests/test-doc-embeddings Executable file
View file

@ -0,0 +1,22 @@
#!/usr/bin/env python3
import pulsar
from trustgraph.clients.document_embeddings_client import DocumentEmbeddingsClient
from trustgraph.clients.embeddings_client import EmbeddingsClient
ec = EmbeddingsClient(pulsar_host="pulsar://localhost:6650")
vectors = ec.request("What caused the space shuttle to explode?")
print(vectors)
llm = DocumentEmbeddingsClient(pulsar_host="pulsar://localhost:6650")
limit=10
resp = llm.request(vectors, limit)
print("Response...")
for val in resp:
print(val)