mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-06-15 01:35:13 +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
22
tests/test-doc-embeddings
Executable file
22
tests/test-doc-embeddings
Executable file
|
|
@ -0,0 +1,22 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import pulsar
|
||||
from trustgraph.clients.document_embeddings_client import DocumentEmbeddingsClient
|
||||
from trustgraph.clients.embeddings_client import EmbeddingsClient
|
||||
|
||||
ec = EmbeddingsClient(pulsar_host="pulsar://localhost:6650")
|
||||
|
||||
vectors = ec.request("What caused the space shuttle to explode?")
|
||||
|
||||
print(vectors)
|
||||
|
||||
llm = DocumentEmbeddingsClient(pulsar_host="pulsar://localhost:6650")
|
||||
|
||||
limit=10
|
||||
|
||||
resp = llm.request(vectors, limit)
|
||||
|
||||
print("Response...")
|
||||
for val in resp:
|
||||
print(val)
|
||||
|
||||
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)
|
||||
|
||||
15
tests/test-doc-rag
Executable file
15
tests/test-doc-rag
Executable file
|
|
@ -0,0 +1,15 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import pulsar
|
||||
from trustgraph.clients.document_rag_client import DocumentRagClient
|
||||
|
||||
rag = DocumentRagClient(pulsar_host="pulsar://localhost:6650")
|
||||
|
||||
query="""
|
||||
This knowledge graph describes the Space Shuttle disaster.
|
||||
Present 20 facts which are present in the knowledge graph."""
|
||||
|
||||
resp = rag.request(query)
|
||||
|
||||
print(resp)
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue