mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
669aed0f8a
* document embedding writer & query * Added test query for doc embeddings * Bump version * Added doc rag prompt * Document RAG service
25 lines
581 B
Python
Executable file
25 lines
581 B
Python
Executable file
#!/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)
|
|
|