trustgraph/test-api
2024-11-20 16:57:46 +00:00

57 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,
)
print(resp.json()["response"])
sys.exit(0)
############################################################################
input = {
"id": "question",
"variables": {
"question": "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}prompt",
json=input,
)
print(resp.json()["text"])
############################################################################
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))
############################################################################