Bedrock, Cohere, and Mixed Support (#6)

* Added Cohere support
* Tweaks to rag prompt
* Bedrock support
* Version to 0.5.7
This commit is contained in:
Jack Colquitt 2024-08-07 11:27:37 -07:00 committed by GitHub
parent 63a30b25eb
commit f349c8a72e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
17 changed files with 1086 additions and 11 deletions

14
tests/test-graph-rag2 Executable file
View file

@ -0,0 +1,14 @@
#!/usr/bin/env python3
import pulsar
from trustgraph.graph_rag_client import GraphRagClient
rag = GraphRagClient(pulsar_host="pulsar://localhost:6650")
query="""List 20 key points to describe the research that led to the discovery of Leo VI.
"""
resp = rag.request(query)
print(resp)