mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
23 lines
504 B
Text
23 lines
504 B
Text
|
#!/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)
|
||
|
|
|