trustgraph/tests.manual/test-embeddings

#!/usr/bin/env python3

import pulsar
from trustgraph.clients.embeddings_client import EmbeddingsClient

embed = EmbeddingsClient(
    pulsar_host="pulsar://pulsar:6650",
    input_queue="non-persistent://tg/request/embeddings:default",
    output_queue="non-persistent://tg/response/embeddings:default",
    subscriber="test1",
)

prompt="Write a funny limerick about a llama"

resp = embed.request(prompt)

print(resp)
Trustgraph initial code drop 2024-07-10 23:20:06 +01:00			`#!/usr/bin/env python3`

			`import pulsar`
- Change flawed _client timeout logic which was causing major lags - Moved clients to trustgraph.clients to tidy the parent directory - Version bump 2024-08-20 17:54:11 +01:00			`from trustgraph.clients.embeddings_client import EmbeddingsClient`
Trustgraph initial code drop 2024-07-10 23:20:06 +01:00
Feature/configure flows (#345) - Keeps processing in different flows separate so that data can go to different stores / collections etc. - Potentially supports different processing flows - Tidies the processing API with common base-classes for e.g. LLMs, and automatic configuration of 'clients' to use the right queue names in a flow 2025-04-22 20:21:38 +01:00			`embed = EmbeddingsClient(`
			`pulsar_host="pulsar://pulsar:6650",`
			`input_queue="non-persistent://tg/request/embeddings:default",`
			`output_queue="non-persistent://tg/response/embeddings:default",`
			`subscriber="test1",`
			`)`
Trustgraph initial code drop 2024-07-10 23:20:06 +01:00
			`prompt="Write a funny limerick about a llama"`

			`resp = embed.request(prompt)`

			`print(resp)`