trustgraph/test-api/test-graph-rag-api
cybermaggedon 92b84441eb
Feature/api gateway (#164)
* Bare bones API gateway
* Working for LLM + prompt
* RAG query works
* Triples query
* Added agent API
* Embeddings API
* Put API tests in a subdir
2024-11-20 19:55:40 +00:00

31 lines
512 B
Python
Executable file

#!/usr/bin/env python3
import requests
import json
import sys
url = "http://localhost:8088/api/v1/"
############################################################################
input = {
"query": "Give me 10 facts",
}
resp = requests.post(
f"{url}graph-rag",
json=input,
)
resp = resp.json()
print(resp)
if "error" in resp:
print(f"Error: {resp['error']}")
sys.exit(1)
print(resp["response"])
sys.exit(0)
############################################################################