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

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

View file

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

View file

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