trustgraph/tests/test-embeddings
2024-07-10 23:20:06 +01:00

15 lines
254 B
Python
Executable file

#!/usr/bin/env python3
import pulsar
from trustgraph.embeddings_client import EmbeddingsClient
embed = EmbeddingsClient(pulsar_host="pulsar://localhost:6650")
prompt="Write a funny limerick about a llama"
resp = embed.request(prompt)
print(resp)