trustgraph/tests/test-llm

17 lines
284 B
Text
Raw Normal View History

2024-07-10 23:20:06 +01:00
#!/usr/bin/env python3
import pulsar
from trustgraph.clients.llm_client import LlmClient
2024-07-10 23:20:06 +01:00
llm = LlmClient(pulsar_host="pulsar://localhost:6650")
system = "You are a lovely assistant."
2024-07-10 23:20:06 +01:00
prompt="Write a funny limerick about a llama"
resp = llm.request(system, prompt)
2024-07-10 23:20:06 +01:00
print(resp)
2024-07-15 17:17:04 +01:00
2024-07-10 23:20:06 +01:00