apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-17 19:35:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tweak GraphRAG and DocRAG params, path hops == 2 (#312)

Browse source
This commit is contained in:
cybermaggedon 2025-03-13 00:44:25 +00:00 • committed by GitHub
parent ef845d6c9b
commit 88768a791b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
templates/components/document-rag.jsonnet
View file

@ -18,6 +18,8 @@ local prompts = import "prompts/mixtral.jsonnet";
"document-rag", "document-rag",
"-p", "-p",
url.pulsar, url.pulsar,
"--doc-limit",
std.toString($["document-rag-doc-limit"]),
"--prompt-request-queue", "--prompt-request-queue",
"non-persistent://tg/request/prompt-rag", "non-persistent://tg/request/prompt-rag",
"--prompt-response-queue", "--prompt-response-queue",

9
templates/components/graph-rag.jsonnet
View file

@ -4,9 +4,10 @@ local url = import "values/url.jsonnet";
{ {
"graph-rag-entity-limit":: 20, "graph-rag-entity-limit":: 50,
"graph-rag-triple-limit":: 10, "graph-rag-triple-limit":: 30,
"graph-rag-max-subgraph-size":: 1000, "graph-rag-max-subgraph-size":: 400,
"graph-rag-max-path-length":: 2,
"kg-extract-definitions" +: { "kg-extract-definitions" +: {
@ -119,6 +120,8 @@ local url = import "values/url.jsonnet";
std.toString($["graph-rag-triple-limit"]), std.toString($["graph-rag-triple-limit"]),
"--max-subgraph-size", "--max-subgraph-size",
std.toString($["graph-rag-max-subgraph-size"]), std.toString($["graph-rag-max-subgraph-size"]),
"--max-path-length",
std.toString($["graph-rag-max-path-length"]),
]) ])
.with_limits("0.5", "128M") .with_limits("0.5", "128M")
.with_reservations("0.1", "128M"); .with_reservations("0.1", "128M");