Updated tests

This commit is contained in:
Cyber MacGeddon 2025-07-11 16:43:27 +01:00
parent 4ff85cd6bd
commit d25c2f4d10
40 changed files with 301 additions and 485 deletions

25
tests.manual/test-doc-prompt Executable file
View file

@ -0,0 +1,25 @@
#!/usr/bin/env python3
import pulsar
from trustgraph.clients.prompt_client import PromptClient
p = PromptClient(pulsar_host="pulsar://localhost:6650")
docs = [
"In our house there is a big cat and a small cat.",
"The small cat is black.",
"The big cat is called Fred.",
"The orange stripey cat is big.",
"The black cat pounces on the big cat.",
"The black cat is called Hope."
]
query="What is the name of the cat who pounces on Fred? Provide a full explanation."
resp = p.request_document_prompt(
query=query,
documents=docs,
)
print(resp)