- Added graph_embeddings_client.py

- Added tests/test-graph-embeddings to demonstrate client works
- Fixed schema
This commit is contained in:
Cyber MacGeddon 2024-08-12 17:35:48 +01:00
parent e2896f6433
commit 6de72d7567
4 changed files with 134 additions and 8 deletions

23
tests/test-graph-embeddings Executable file
View file

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