mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-12 08:42:37 +02:00
Added doc embedding support (#41)
* document embedding writer & query * Added test query for doc embeddings * Bump version * Added doc rag prompt * Document RAG service
This commit is contained in:
parent
0159e938a2
commit
669aed0f8a
52 changed files with 1257 additions and 252 deletions
25
tests/test-doc-prompt
Executable file
25
tests/test-doc-prompt
Executable file
|
|
@ -0,0 +1,25 @@
|
|||
#!/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)
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue