[GraphRAG][v2] Implementation of Agentic Retrieval and Performance enhancement of KG construction & Semantic Beam Search #350

ParameswaranSajeenthiran · 2026-01-20T11:15:45Z

Summary

This PR introduces the new Agentic Retrieval and improvements in Knowledge Graph (KG) construction and semantic search, focusing on performance, scalability, and reliability.

1. Agentic Retrieval

Introduced session-based agent orchestration initiated from the frontend to manage agentic retrieval workflows per user query
Implemented agent-driven query understanding, enabling interpretation of user intent and constraints from raw query text
Added semantic beam search plan generation, allowing the agent to construct and rank multi-step retrieval plans
Enabled iterative execution of beam search plans, where intermediate results inform subsequent retrieval decisions
Integrated agent-controlled response generation, synthesizing grounded responses from semantic beam search outputs

2. Knowledge Graph (KG) Construction

Implemented KG construction from PDF with support in both UI and server
Parallelized graph file persistence and embedding generation to reduce ingestion latency
Migrated online edge embedding (API during query) to offline embedding generation and indexing
Implemented tuple validation with retry mechanism for invalid tuples to improve data reliability
General code-level optimizations for ingestion and persistence
Added and extended integration tests for KG construction and persistence pipeline

2. Semantic Beam Search & Graph / Query Optimizations

Introduced Semantic Beam Search for improved multi-hop reasoning and retrieval quality
Added Fennel-based graph partitioning for better scalability
Migrated lightweight queries from processes to threads to enable shared access to large-scale indexes
General code-level optimizations for query execution
Integration tests added for search correctness and query performance

Impact

Faster and more reliable KG ingestion
Reduced dependency on runtime embedding APIs
Scalable query execution and search
Higher-quality semantic search results

Notes for Reviewers

Parallelism changes affect persistence and embedding pipelines — concurrency handling should be reviewed
Offline edge embedding migration introduces new indexing logic

Fix NodeManager

Feature/agentic retrieval

.github/workflows/build.yml

src/frontend/JasmineGraphFrontEnd.cpp

src/rag/util/LLMUtils.cpp

Remove redundant graphRAG github action job

sonarqubecloud · 2026-01-20T12:22:51Z

Quality Gate failed

Failed conditions
9 Security Hotspots
19.9% Duplication on New Code (required ≤ 3%)
D Security Rating on New Code (required ≥ A)
E Reliability Rating on New Code (required ≥ A)

See analysis details on SonarQube Cloud

Catch issues before they fail your Quality Gate with our IDE extension SonarQube for IDE

src/frontend/JasmineGraphFrontEnd.cpp

+        CURL* curl = curl_easy_init();
+        if (curl) {
+            std::string response;
+            curl_easy_setopt(curl, CURLOPT_SSLVERSION, CURL_SSLVERSION_DEFAULT);


src/rag/util/LLMUtils.cpp

+    }
+
+    // TLS + connection settings
+    curl_easy_setopt(curl, CURLOPT_SSLVERSION, CURL_SSLVERSION_DEFAULT);


codecov · 2026-01-20T12:29:40Z

Codecov Report

❌ Patch coverage is 0.65917% with 2562 lines in your changes missing coverage. Please review.
✅ Project coverage is 0.92%. Comparing base (c52c17b) to head (a3ce8dd).

Files with missing lines	Patch %	Lines
src/knowledgegraph/construction/Pipeline.cpp	0.00%	691 Missing ⚠️
src/util/Utils.cpp	4.40%	321 Missing and 26 partials ⚠️
src/frontend/JasmineGraphFrontEnd.cpp	0.00%	318 Missing ⚠️
...ssor/nlp/semanticbeamsearch/SemanticBeamSearch.cpp	0.00%	274 Missing ⚠️
.../incremental/JasmineGraphIncrementalLocalStore.cpp	0.00%	142 Missing ⚠️
.../frontend/core/executor/impl/AgentPlanExecutor.cpp	0.00%	136 Missing ⚠️
.../core/executor/impl/SemanticBeamSearchExecutor.cpp	0.00%	95 Missing ⚠️
src/vectorstore/FaissIndex.cpp	0.00%	72 Missing ⚠️
src/rag/util/LLMUtils.cpp	0.00%	69 Missing ⚠️
...tioner/stream/HDFSMultiThreadedHashPartitioner.cpp	0.00%	61 Missing ⚠️
... and 20 more

Additional details and impacted files

@@            Coverage Diff            @@
##           master    #350      +/-   ##
=========================================
- Coverage    0.95%   0.92%   -0.04%     
=========================================
  Files         108     115       +7     
  Lines       26278   28224    +1946     
  Branches    17306   18615    +1309     
=========================================
+ Hits          250     260      +10     
- Misses      25812   27737    +1925     
- Partials      216     227      +11

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

miyurud · 2026-01-23T17:32:09Z

.clang-format

why do we need clang here?

miyurud · 2026-01-23T17:49:03Z

.clang-format

@@ -0,0 +1,12 @@
+BasedOnStyle: Google


why do we need to change this file?

miyurud · 2026-01-23T18:08:09Z

CMakeLists.txt

    message(WARNING "No ANTLR generated files found in /home/ubuntu/software/antlr/")
    set(GENERATED_SRC ""
+            src/frontend/JasmineGraphFrontEnd.cpp
+            src/localstore/incremental/JasmineGraphIncrementalLocalStore.cpp


why is src/localstore/incremental/JasmineGraphIncrementalLocalStore.cpp mentioned three times here?

miyurud · 2026-01-23T18:09:51Z

conf/jasminegraph-server.properties


 #This parameter holds the maximum label size of Node Block
-org.jasminegraph.nativestore.max.label.size=43
+org.jasminegraph.nativestore.max.label.size=256


why did we increase this value to 256?

miyurud · 2026-01-23T18:10:30Z

conf/jasminegraph-server.properties

+org.jasminegraph.vectorstore.enabled=true
+org.jasminegraph.vectorstore.dimension=512
+org.jasminegraph.vectorstore.embedding.model=jina/jina-embeddings-v2-small-en
+org.jasminegraph.vectorstore.embedding.ollama.endpoint=http://gemma3_container:11441


why do we expose this endpoint?

miyurud · 2026-01-23T18:11:59Z

mock_ollama_server.py

@@ -0,0 +1,121 @@
+"""Copyright 2025 JasmineGraph Team


Suggested change

"""Copyright 2025 JasmineGraph Team

"""Copyright 2026 JasmineGraph Team

ParameswaranSajeenthiran added 30 commits November 29, 2025 14:25

Add fennel partitioning for HDFS stream handler

1b7d081

Update test-docker to specific tests

faff837

Update test-docker to run specific tests

d882826

Fix start-ollama path

22f7f6f

Update build.yml to run specific tests

3c56387

Fix path to kg text files

23a3eb5

Fix path to file upload script

53ce44e

Fine tune parameters for LLM inference

f4feec4

Fix graphRAG test in sending llm runners

ace2648

change llm

a8ce22e

change llm

c736554

Add Semantic Beam Search

8020fd5

Add graph k8s integration test

cef5b54

Pass masterIp to the test-k8s.sh python scriot execution

3bde47d

Update Maximum ephemeral-storage

6018798

Change llm

769cb64

Change llm

1135fea

Removing jasminegraph image to free space

d7c7bba

Revert empherical storage limitation

5dc4240

Fix python syntax error in test.py

6cb6dbe

Change LLM

7f6104f

Change LLM

8166add

Add rm image in test-k8s.sh

411e398

add mock ollama service

274da8d

add mock ollama service

66dc7ae

add mock ollama service

1bdf843

add mock ollama service

a1a5a24

add mock ollama service

41bcba2

Add models fetch api to mock ollama service

bc4ad05

Add models fetch api to mock ollama service

cb3ab71

ParameswaranSajeenthiran and others added 15 commits January 4, 2026 17:36

Remove prom_net in integration test

4d10d9e

Remove prom_net in integration test

a23fe42

Update ip of ollama

47292bd

Add clean up docker

d03fc5e

Add service names for pushgateway and prometheus.yaml

cae8526

Add service names for embedding model

531ab84

Add service names for embedding model

cdbe6b8

Add ip address for jasminegraph integration test

33e1a75

Revert graphRAG integration tests in build.yml

23c4825

sh code linting

b227212

Fix SBS executor util

82655c3

Add sbs integration test

7f80897

Fix python lint issue

83e1e9e

Merge pull request #40 from ParameswaranSajeenthiran/raw-kg-contruction

a860457

Fix NodeManager

Merge pull request #37 from abineyan/feature/agentic-retrieval

58816b6

Feature/agentic retrieval

github-advanced-security bot found potential problems Jan 20, 2026

View reviewed changes

.github/workflows/build.yml Fixed Show fixed Hide fixed

src/frontend/JasmineGraphFrontEnd.cpp Fixed Show fixed Hide fixed

src/rag/util/LLMUtils.cpp Fixed Show fixed Hide fixed

ParameswaranSajeenthiran and others added 4 commits January 20, 2026 17:16

Remove redundant graphRAG github action job

9238464

Remove redundant graphRAG github action job

4e266e9

Add SSL verification for curl connections in agent implementation

0fa4816

Merge pull request #45 from abineyan/saji

a3ce8dd

Remove redundant graphRAG github action job

github-advanced-security bot found potential problems Jan 20, 2026

View reviewed changes

miyurud reviewed Jan 23, 2026

View reviewed changes

.clang-format

Copy link

Owner

miyurud Jan 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why do we need clang here?

miyurud reviewed Jan 23, 2026

View reviewed changes

.clang-format

@@ -0,0 +1,12 @@

BasedOnStyle: Google

Copy link

Owner

miyurud Jan 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why do we need to change this file?

miyurud reviewed Jan 23, 2026

View reviewed changes

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[GraphRAG][v2] Implementation of Agentic Retrieval and Performance enhancement of KG construction & Semantic Beam Search #350

[GraphRAG][v2] Implementation of Agentic Retrieval and Performance enhancement of KG construction & Semantic Beam Search #350

Uh oh!

ParameswaranSajeenthiran commented Jan 20, 2026 •

edited

Loading

Uh oh!

Uh oh!

Uh oh!

Uh oh!

sonarqubecloud bot commented Jan 20, 2026

Uh oh!

Check failure

Check failure

codecov bot commented Jan 20, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

miyurud Jan 23, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

	"""Copyright 2025 JasmineGraph Team
	"""Copyright 2026 JasmineGraph Team

[GraphRAG][v2] Implementation of Agentic Retrieval and Performance enhancement of KG construction & Semantic Beam Search #350

Are you sure you want to change the base?

[GraphRAG][v2] Implementation of Agentic Retrieval and Performance enhancement of KG construction & Semantic Beam Search #350

Uh oh!

Conversation

ParameswaranSajeenthiran commented Jan 20, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Summary

1. Agentic Retrieval

2. Knowledge Graph (KG) Construction

2. Semantic Beam Search & Graph / Query Optimizations

Impact

Notes for Reviewers

Uh oh!

Uh oh!

Uh oh!

Uh oh!

sonarqubecloud bot commented Jan 20, 2026

Quality Gate failed

Uh oh!

Check failure

Check failure

codecov bot commented Jan 20, 2026

Codecov Report

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

miyurud Jan 23, 2026

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

ParameswaranSajeenthiran commented Jan 20, 2026 •

edited

Loading