mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-21 19:21:03 +02:00
Tidy test output
This commit is contained in:
parent
f26eacb439
commit
6838f0b94e
1 changed files with 27 additions and 3 deletions
|
|
@ -1,21 +1,45 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import json
|
import json
|
||||||
|
import textwrap
|
||||||
from trustgraph.clients.agent_client import AgentClient
|
from trustgraph.clients.agent_client import AgentClient
|
||||||
|
|
||||||
|
def wrap(text, width=75):
|
||||||
|
|
||||||
|
if text is None: text = "n/a"
|
||||||
|
|
||||||
|
out = textwrap.wrap(
|
||||||
|
text, width=width
|
||||||
|
)
|
||||||
|
return "\n".join(out)
|
||||||
|
|
||||||
|
def output(text, prefix="> ", width=78):
|
||||||
|
|
||||||
|
out = textwrap.indent(
|
||||||
|
text, prefix=prefix
|
||||||
|
)
|
||||||
|
print(out)
|
||||||
|
|
||||||
p = AgentClient(pulsar_host="pulsar://localhost:6650")
|
p = AgentClient(pulsar_host="pulsar://localhost:6650")
|
||||||
|
|
||||||
q = "How many cats does Mark have? Calculate that number raised to 0.4 power. Is that number lower than the numeric part of the mission identifier of the Space Shuttle Challenger on its last mission? If so, give me an apple pie recipe, otherwise return a poem about cheese."
|
q = "How many cats does Mark have? Calculate that number raised to 0.4 power. Is that number lower than the numeric part of the mission identifier of the Space Shuttle Challenger on its last mission? If so, give me an apple pie recipe, otherwise return a poem about cheese."
|
||||||
|
|
||||||
|
output(wrap(q), "\U00002753 ")
|
||||||
|
print()
|
||||||
|
|
||||||
def think(x):
|
def think(x):
|
||||||
print(f"think: {x}")
|
output(wrap(x), "\U0001f914 ")
|
||||||
|
print()
|
||||||
|
|
||||||
def observe(x):
|
def observe(x):
|
||||||
print(f"observe: {x}")
|
output(wrap(x), "\U0001f4a1 ")
|
||||||
|
print()
|
||||||
|
|
||||||
resp = p.request(
|
resp = p.request(
|
||||||
question=q, think=think, observe=observe,
|
question=q, think=think, observe=observe,
|
||||||
)
|
)
|
||||||
|
|
||||||
print(resp)
|
output(resp, "\U0001f4ac ")
|
||||||
|
print()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue