mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-21 19:21:03 +02:00
68 lines
1.3 KiB
Python
Executable file
68 lines
1.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import requests
|
|
import json
|
|
import sys
|
|
|
|
url = "http://localhost:8088/api/v1/"
|
|
|
|
############################################################################
|
|
|
|
input = {
|
|
"system": "Respond in French. Use long word, form of numbers, no digits",
|
|
# "prompt": "Add 2 and 12"
|
|
"prompt": "Add 12 and 14, and then make a poem about llamas which incorporates that number. Then write a joke about llamas"
|
|
}
|
|
|
|
resp = requests.post(
|
|
f"{url}text-completion",
|
|
json=input,
|
|
)
|
|
|
|
resp = resp.json()
|
|
|
|
if "error" in resp:
|
|
print(f"Error: {resp['error']}")
|
|
sys.exit(1)
|
|
|
|
print(resp["response"])
|
|
|
|
############################################################################
|
|
|
|
input = {
|
|
"id": "question",
|
|
"variables": {
|
|
"question": "Write a joke about llamas."
|
|
}
|
|
}
|
|
|
|
resp = requests.post(
|
|
f"{url}prompt",
|
|
json=input,
|
|
)
|
|
|
|
if "error" in resp:
|
|
print(f"Error: {resp['error']}")
|
|
sys.exit(1)
|
|
|
|
print(resp["response"])
|
|
|
|
sys.exit(0)
|
|
############################################################################
|
|
|
|
input = {
|
|
"id": "extract-definitions",
|
|
"variables": {
|
|
"text": "A cat is a large mammal."
|
|
}
|
|
}
|
|
|
|
resp = requests.post(
|
|
f"{url}prompt",
|
|
json=input,
|
|
)
|
|
|
|
print(json.dumps(resp.json()["object"], indent=4))
|
|
|
|
############################################################################
|
|
|