@@ -757,8 +756,23 @@ def startup_diagnostics():
else:
print("β PyTorch Geometric not available")
+ # Check PyGraphistry credentials
+ print("Checking PyGraphistry credentials...")
+ personal_key = os.getenv('GRAPHISTRY_PERSONAL_KEY')
+ secret_key = os.getenv('GRAPHISTRY_SECRET_KEY')
+ api_key = os.getenv('GRAPHISTRY_API_KEY')
+
+ if personal_key and secret_key:
+ print("β PyGraphistry personal key/secret found")
+ elif api_key:
+ print("β PyGraphistry API key found")
+ else:
+ print("β No PyGraphistry credentials found - cloud mode will be limited")
+ print(" Set GRAPHISTRY_PERSONAL_KEY + GRAPHISTRY_SECRET_KEY for full cloud features")
+
print("")
print("π― Available Processing Modes:")
+ print(" βοΈ PyGraphistry Cloud - Interactive GPU embeds (requires credentials)")
print(" π Local GPU (cuGraph) - Full local GPU processing")
print(" π» Local CPU - NetworkX fallback")
print("")
@@ -772,4 +786,4 @@ def startup_diagnostics():
if __name__ == "__main__":
startup_diagnostics()
- uvicorn.run(app, host="0.0.0.0", port=8080)
+ uvicorn.run(app, host="0.0.0.0", port=8080)
\ No newline at end of file
diff --git a/nvidia/txt2kg/assets/deploy/services/gpu-viz/webgl_rendering_enhancement.py b/nvidia/txt2kg/assets/deploy/services/gpu-viz/webgl_rendering_enhancement.py
index 0b49a9c..c887783 100644
--- a/nvidia/txt2kg/assets/deploy/services/gpu-viz/webgl_rendering_enhancement.py
+++ b/nvidia/txt2kg/assets/deploy/services/gpu-viz/webgl_rendering_enhancement.py
@@ -1,3 +1,19 @@
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# WebGL-Enhanced Remote GPU Rendering Service
# Using Three.js for GPU-accelerated visualization
diff --git a/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile b/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile
index e3e76d8..efdad5f 100644
--- a/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile
+++ b/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile
@@ -1,4 +1,4 @@
-FROM ollama/ollama:latest
+FROM ollama/ollama:0.12.6
# Copy the entrypoint script
COPY entrypoint.sh /entrypoint.sh
diff --git a/nvidia/txt2kg/assets/deploy/services/ollama/clear_cache_and_restart.sh b/nvidia/txt2kg/assets/deploy/services/ollama/clear_cache_and_restart.sh
index 9721e92..dc17a69 100755
--- a/nvidia/txt2kg/assets/deploy/services/ollama/clear_cache_and_restart.sh
+++ b/nvidia/txt2kg/assets/deploy/services/ollama/clear_cache_and_restart.sh
@@ -1,5 +1,21 @@
#!/bin/bash
#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+#
# Clear buffer cache and restart Ollama to fix unified memory detection
# This script addresses the issue where Ollama can't see full GPU memory
# due to buffer cache not being reclaimable in unified memory systems
diff --git a/nvidia/txt2kg/assets/deploy/services/ollama/entrypoint.sh b/nvidia/txt2kg/assets/deploy/services/ollama/entrypoint.sh
index 6959cf3..425ff96 100644
--- a/nvidia/txt2kg/assets/deploy/services/ollama/entrypoint.sh
+++ b/nvidia/txt2kg/assets/deploy/services/ollama/entrypoint.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
set -e
# Start Ollama server in the background
diff --git a/nvidia/txt2kg/assets/deploy/services/sentence-transformers/app.py b/nvidia/txt2kg/assets/deploy/services/sentence-transformers/app.py
index 1f3cbc0..de9f914 100644
--- a/nvidia/txt2kg/assets/deploy/services/sentence-transformers/app.py
+++ b/nvidia/txt2kg/assets/deploy/services/sentence-transformers/app.py
@@ -1,3 +1,19 @@
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
from flask import Flask, request, jsonify
from sentence_transformers import SentenceTransformer
import os
diff --git a/nvidia/txt2kg/assets/deploy/services/vllm/build_image.sh b/nvidia/txt2kg/assets/deploy/services/vllm/build_image.sh
index 4c902d8..fc50dcc 100755
--- a/nvidia/txt2kg/assets/deploy/services/vllm/build_image.sh
+++ b/nvidia/txt2kg/assets/deploy/services/vllm/build_image.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# Use latest stable vLLM release for better compute capability 12.1 support
# Clone the vLLM GitHub repo and use latest stable release.
diff --git a/nvidia/txt2kg/assets/deploy/services/vllm/launch_server.sh b/nvidia/txt2kg/assets/deploy/services/vllm/launch_server.sh
index e89640c..a9a09ae 100755
--- a/nvidia/txt2kg/assets/deploy/services/vllm/launch_server.sh
+++ b/nvidia/txt2kg/assets/deploy/services/vllm/launch_server.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# Launch vLLM with NVIDIA Triton Inference Server optimized build
# This should have proper support for compute capability 12.1 (DGX Spark)
diff --git a/nvidia/txt2kg/assets/deploy/services/vllm/run_benchmark.sh b/nvidia/txt2kg/assets/deploy/services/vllm/run_benchmark.sh
index 43208e7..795d287 100755
--- a/nvidia/txt2kg/assets/deploy/services/vllm/run_benchmark.sh
+++ b/nvidia/txt2kg/assets/deploy/services/vllm/run_benchmark.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# vLLM Llama3 8B Benchmark Runner
# Uses NVIDIA vLLM container for optimal performance
diff --git a/nvidia/txt2kg/assets/deploy/services/vllm/run_container.sh b/nvidia/txt2kg/assets/deploy/services/vllm/run_container.sh
index a56cb18..feb81bf 100755
--- a/nvidia/txt2kg/assets/deploy/services/vllm/run_container.sh
+++ b/nvidia/txt2kg/assets/deploy/services/vllm/run_container.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# Follow the official vLLM tutorial run_container.sh exactly
docker run -e HF_TOKEN="$HF_TOKEN" -e HF_HOME="$HF_HOME" --ipc=host --gpus all --entrypoint "/bin/bash" --rm -it vllm/vllm-openai:deploy
diff --git a/nvidia/txt2kg/assets/deploy/services/vllm/start-vllm.sh b/nvidia/txt2kg/assets/deploy/services/vllm/start-vllm.sh
index 4b27093..f0140a8 100755
--- a/nvidia/txt2kg/assets/deploy/services/vllm/start-vllm.sh
+++ b/nvidia/txt2kg/assets/deploy/services/vllm/start-vllm.sh
@@ -1,4 +1,20 @@
#!/bin/bash
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
# vLLM startup script with NVFP4 quantization support for Llama 4 Scout
# Optimized for NVIDIA Blackwell and Hopper architectures
diff --git a/nvidia/txt2kg/assets/examples/download_biorxiv_dataset.py b/nvidia/txt2kg/assets/examples/download_biorxiv_dataset.py
index a7cf2da..8a4605b 100644
--- a/nvidia/txt2kg/assets/examples/download_biorxiv_dataset.py
+++ b/nvidia/txt2kg/assets/examples/download_biorxiv_dataset.py
@@ -1,4 +1,20 @@
#!/usr/bin/env python3
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
"""
Download and process the MTEB raw_biorxiv dataset for txt2kg demo.
Filter for genetics/genomics categories and create individual txt files.
diff --git a/nvidia/txt2kg/assets/examples/download_cc_biorxiv_dataset.py b/nvidia/txt2kg/assets/examples/download_cc_biorxiv_dataset.py
index bfa76b4..645b046 100644
--- a/nvidia/txt2kg/assets/examples/download_cc_biorxiv_dataset.py
+++ b/nvidia/txt2kg/assets/examples/download_cc_biorxiv_dataset.py
@@ -1,4 +1,20 @@
#!/usr/bin/env python3
+#
+# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
"""
Download and process the marianna13/biorxiv dataset for txt2kg demo.
Filter for Creative Commons licensed papers and create individual txt files.
diff --git a/nvidia/txt2kg/assets/frontend/app/.static/3d-force-graph-mock.js b/nvidia/txt2kg/assets/frontend/app/.static/3d-force-graph-mock.js
index 218be2b..5990b7d 100644
--- a/nvidia/txt2kg/assets/frontend/app/.static/3d-force-graph-mock.js
+++ b/nvidia/txt2kg/assets/frontend/app/.static/3d-force-graph-mock.js
@@ -1 +1,17 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
console.log('This is mock data to avoid SSR issues')
diff --git a/nvidia/txt2kg/assets/frontend/app/api/backend/route.ts b/nvidia/txt2kg/assets/frontend/app/api/backend/route.ts
index 178002a..0103ff7 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/backend/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/backend/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import remoteBackend from '@/lib/remote-backend';
import type { Triple } from '@/types/graph';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/config/route.ts b/nvidia/txt2kg/assets/frontend/app/api/config/route.ts
index d94e619..0f80a92 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/config/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/config/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextResponse } from "next/server";
export async function GET() {
diff --git a/nvidia/txt2kg/assets/frontend/app/api/embeddings/route.ts b/nvidia/txt2kg/assets/frontend/app/api/embeddings/route.ts
index caf7bb3..e6038e5 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/embeddings/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/embeddings/route.ts
@@ -1,9 +1,25 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { EmbeddingsService } from '@/lib/embeddings';
-import { PineconeService } from '@/lib/pinecone';
+import { QdrantService } from '@/lib/qdrant';
/**
- * Generate embeddings for text chunks and store them in Pinecone
+ * Generate embeddings for text chunks and store them in Qdrant
*/
export async function POST(request: NextRequest) {
try {
@@ -38,15 +54,15 @@ export async function POST(request: NextRequest) {
console.log('Generating embeddings for chunks...');
const embeddings = await embeddingsService.encode(chunks);
console.log(`Generated ${embeddings.length} embeddings`);
-
- // Initialize PineconeService
- const pineconeService = PineconeService.getInstance();
-
- // Check if Pinecone server is running
- const isPineconeRunning = await pineconeService.isPineconeRunning();
+
+ // Initialize QdrantService
+ const pineconeService = QdrantService.getInstance();
+
+ // Check if Qdrant server is running
+ const isPineconeRunning = await pineconeService.isQdrantRunning();
if (!isPineconeRunning) {
return NextResponse.json(
- { error: 'Pinecone server is not available. Please make sure it is running.' },
+ { error: 'Qdrant server is not available. Please make sure it is running.' },
{ status: 503 }
);
}
diff --git a/nvidia/txt2kg/assets/frontend/app/api/enhanced-query/route.ts b/nvidia/txt2kg/assets/frontend/app/api/enhanced-query/route.ts
index 7533849..7841ad8 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/enhanced-query/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/enhanced-query/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { RemoteBackendService } from '@/lib/remote-backend';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/extract-triples/route.ts b/nvidia/txt2kg/assets/frontend/app/api/extract-triples/route.ts
index da1382b..0e82e80 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/extract-triples/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/extract-triples/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { processDocument, TextProcessor } from '@/lib/text-processor';
import { llmService } from '@/lib/llm-service';
@@ -29,7 +45,8 @@ export async function POST(req: NextRequest) {
ollamaModel,
ollamaBaseUrl,
vllmModel,
- vllmBaseUrl
+ vllmBaseUrl,
+ nvidiaModel
} = body;
if (!text || typeof text !== 'string') {
@@ -136,7 +153,8 @@ export async function POST(req: NextRequest) {
ollamaModel: ollamaModel,
ollamaBaseUrl: ollamaBaseUrl,
vllmModel: vllmModel,
- vllmBaseUrl: vllmBaseUrl
+ vllmBaseUrl: vllmBaseUrl,
+ nvidiaModel: nvidiaModel
});
}
diff --git a/nvidia/txt2kg/assets/frontend/app/api/fix-query-logs/route.ts b/nvidia/txt2kg/assets/frontend/app/api/fix-query-logs/route.ts
index 525cc67..a318846 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/fix-query-logs/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/fix-query-logs/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import queryLoggerService, { QueryLogEntry } from '@/lib/query-logger';
import fs from 'fs';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-data/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-data/route.ts
index 26c9304..c9a5200 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/graph-data/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-data/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { type NextRequest, NextResponse } from "next/server"
// Utility function to generate UUID with fallback
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-db/clear/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-db/clear/route.ts
index 3434760..6a144ba 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/graph-db/clear/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-db/clear/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { getGraphDbService } from '@/lib/graph-db-util';
import { getGraphDbType } from '../../settings/route';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-db/disconnect/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-db/disconnect/route.ts
index 5eac33b..098b114 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/graph-db/disconnect/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-db/disconnect/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { getGraphDbService } from '@/lib/graph-db-util';
import { getGraphDbType } from '../../settings/route';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-db/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-db/route.ts
index 690b919..91bb57f 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/graph-db/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-db/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { getGraphDbService } from '@/lib/graph-db-util';
import { getGraphDbType } from '../settings/route';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-db/triples/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-db/triples/route.ts
index f777a4d..3142133 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/graph-db/triples/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-db/triples/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { getGraphDbService } from '@/lib/graph-db-util';
import { getGraphDbType } from '../../settings/route';
@@ -160,6 +176,16 @@ export async function POST(req: NextRequest) {
// Store triples in the graph database
await graphDbService.importTriples(validTriples);
+ // Mark document as processed if documentName is provided (only for ArangoDB)
+ if (graphDbType === 'arangodb' && documentName) {
+ try {
+ await (graphDbService as any).markDocumentAsProcessed(documentName, validTriples.length);
+ } catch (error) {
+ console.error('Error marking document as processed:', error);
+ // Don't fail the request if marking fails
+ }
+ }
+
// Return success response
return NextResponse.json({
success: true,
diff --git a/nvidia/txt2kg/assets/frontend/app/api/graph-query-llm/route.ts b/nvidia/txt2kg/assets/frontend/app/api/graph-query-llm/route.ts
new file mode 100644
index 0000000..b0250e8
--- /dev/null
+++ b/nvidia/txt2kg/assets/frontend/app/api/graph-query-llm/route.ts
@@ -0,0 +1,66 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+import { NextRequest, NextResponse } from 'next/server';
+import backendService from '@/lib/backend-service';
+import { getGraphDbType } from '../settings/route';
+
+/**
+ * API endpoint for LLM-enhanced graph query
+ * This retrieves triples using graph search and generates an answer using LLM
+ * Makes traditional graph search comparable to RAG for fair benchmarking
+ * POST /api/graph-query-llm
+ */
+export async function POST(request: NextRequest) {
+ try {
+ const { query, topK = 5, useTraditional = true, llmModel, llmProvider } = await request.json();
+
+ if (!query) {
+ return NextResponse.json({ error: 'Query is required' }, { status: 400 });
+ }
+
+ // Initialize backend if needed with the selected graph DB type
+ if (!backendService.isInitialized) {
+ const graphDbType = getGraphDbType();
+ console.log(`Initializing backend with graph DB type: ${graphDbType}`);
+ await backendService.initialize(graphDbType);
+ }
+
+ console.log(`Graph query with LLM: "${query}", topK=${topK}, traditional=${useTraditional}, model=${llmModel || 'default'}, provider=${llmProvider || 'default'}`);
+
+ // Query the backend with LLM enhancement
+ const result = await backendService.queryWithLLM(query, topK, useTraditional, llmModel, llmProvider);
+
+ // DEBUG: Log first triple in API route to verify depth/pathLength
+ if (result.triples && result.triples.length > 0) {
+ console.log('API route - first triple:', JSON.stringify(result.triples[0], null, 2));
+ }
+
+ // Return results
+ return NextResponse.json({
+ query,
+ answer: result.answer,
+ triples: result.triples,
+ count: result.count,
+ message: `Retrieved ${result.count} triples and generated answer using ${useTraditional ? 'traditional' : 'vector'} graph search + LLM`
+ });
+ } catch (error) {
+ console.error('Error in graph query with LLM:', error);
+ const errorMessage = error instanceof Error ? error.message : 'Unknown error';
+ return NextResponse.json({ error: errorMessage }, { status: 500 });
+ }
+}
+
diff --git a/nvidia/txt2kg/assets/frontend/app/api/metrics/route.ts b/nvidia/txt2kg/assets/frontend/app/api/metrics/route.ts
index 8f024c2..66dbe1f 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/metrics/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/metrics/route.ts
@@ -1,8 +1,24 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import remoteBackendInstance from '@/lib/remote-backend';
-import { Neo4jService } from '@/lib/neo4j';
-import neo4jService from '@/lib/neo4j';
-import { PineconeService } from '@/lib/pinecone';
+import { getGraphDbService } from '@/lib/graph-db-util';
+import { getGraphDbType } from '../settings/route';
+import { QdrantService } from '@/lib/qdrant';
import RAGService from '@/lib/rag';
import queryLoggerService, { QueryLogSummary } from '@/lib/query-logger';
@@ -11,16 +27,31 @@ import queryLoggerService, { QueryLogSummary } from '@/lib/query-logger';
*/
export async function GET(request: NextRequest) {
try {
- // Initialize services
- const neo4j = Neo4jService.getInstance();
- const pineconeService = PineconeService.getInstance();
+ // Initialize services with the correct graph database type
+ const graphDbType = getGraphDbType();
+ const graphDbService = getGraphDbService(graphDbType);
+ const pineconeService = QdrantService.getInstance();
- if (!neo4j.isInitialized()) {
- neo4j.initialize();
+ // Initialize graph database if needed
+ if (!graphDbService.isInitialized()) {
+ if (graphDbType === 'arangodb') {
+ await graphDbService.initialize(
+ process.env.ARANGODB_URL,
+ process.env.ARANGODB_DB,
+ process.env.ARANGODB_USER,
+ process.env.ARANGODB_PASSWORD
+ );
+ } else if (graphDbType === 'neo4j') {
+ graphDbService.initialize(
+ process.env.NEO4J_URI,
+ process.env.NEO4J_USER || process.env.NEO4J_USERNAME,
+ process.env.NEO4J_PASSWORD
+ );
+ }
}
- // Get graph stats from Neo4j
- const graphData = await neo4j.getGraphData();
+ // Get graph stats from the active graph database
+ const graphData = await graphDbService.getGraphData();
// Get unique entities (nodes)
const uniqueEntities = new Set();
@@ -51,11 +82,12 @@ export async function GET(request: NextRequest) {
// Get real query logs instead of mock data
let queryLogs: QueryLogSummary[] = [];
- let precision = 0;
+ let precision = 0;
let recall = 0;
let f1Score = 0;
let avgQueryTime = vectorStats.avgQueryTime || 0;
let avgRelevance = 0;
+ let queryTimesByMode: Record = {};
// Get query logs from file-based logger instead of Neo4j
try {
@@ -72,25 +104,42 @@ export async function GET(request: NextRequest) {
// Calculate metrics from the query logs
if (queryLogs.length > 0) {
// Calculate metrics from logs with actual data
- const logsWithMetrics = queryLogs.filter(log =>
- log.metrics.avgPrecision > 0 ||
- log.metrics.avgRecall > 0 ||
+ const logsWithMetrics = queryLogs.filter(log =>
+ log.metrics.avgPrecision > 0 ||
+ log.metrics.avgRecall > 0 ||
log.metrics.avgExecutionTimeMs > 0
);
-
+
const logsWithRelevance = queryLogs.filter(log => log.metrics.avgRelevanceScore > 0);
-
+
if (logsWithMetrics.length > 0) {
precision = logsWithMetrics.reduce((sum, log) => sum + (log.metrics.avgPrecision || 0), 0) / logsWithMetrics.length;
recall = logsWithMetrics.reduce((sum, log) => sum + (log.metrics.avgRecall || 0), 0) / logsWithMetrics.length;
avgQueryTime = logsWithMetrics.reduce((sum, log) => sum + (log.metrics.avgExecutionTimeMs || 0), 0) / logsWithMetrics.length;
f1Score = precision > 0 && recall > 0 ? 2 * (precision * recall) / (precision + recall) : 0;
}
-
+
if (logsWithRelevance.length > 0) {
avgRelevance = logsWithRelevance.reduce((sum, log) => sum + (log.metrics.avgRelevanceScore || 0), 0) / logsWithRelevance.length;
}
}
+
+ // Calculate per-mode query times
+ const logsByMode = queryLogs.reduce((acc, log) => {
+ const mode = log.queryMode || 'traditional';
+ if (!acc[mode]) acc[mode] = [];
+ acc[mode].push(log);
+ return acc;
+ }, {} as Record);
+
+ Object.entries(logsByMode).forEach(([mode, logs]) => {
+ const logsWithTime = logs.filter(log => log.metrics.avgExecutionTimeMs > 0);
+ if (logsWithTime.length > 0) {
+ queryTimesByMode[mode] = logsWithTime.reduce((sum, log) =>
+ sum + log.metrics.avgExecutionTimeMs, 0
+ ) / logsWithTime.length;
+ }
+ });
} catch (error) {
console.warn('Error getting query logs from file:', error);
// Keep values at 0 instead of using defaults
@@ -117,6 +166,7 @@ export async function GET(request: NextRequest) {
recall,
f1Score,
topQueries,
+ queryTimesByMode: queryTimesByMode || {},
// Add metadata about query logs
queryLogStats: {
totalQueryLogs: queryLogs.length,
diff --git a/nvidia/txt2kg/assets/frontend/app/api/neo4j/disconnect/route.ts b/nvidia/txt2kg/assets/frontend/app/api/neo4j/disconnect/route.ts
index 64efe5f..b43bd4c 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/neo4j/disconnect/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/neo4j/disconnect/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
/**
diff --git a/nvidia/txt2kg/assets/frontend/app/api/neo4j/route.ts b/nvidia/txt2kg/assets/frontend/app/api/neo4j/route.ts
index cbd66da..972e888 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/neo4j/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/neo4j/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { Neo4jService } from '@/lib/neo4j';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/neo4j/triples/route.ts b/nvidia/txt2kg/assets/frontend/app/api/neo4j/triples/route.ts
index 3082a31..471433c 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/neo4j/triples/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/neo4j/triples/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
/**
diff --git a/nvidia/txt2kg/assets/frontend/app/api/ollama/batch/route.ts b/nvidia/txt2kg/assets/frontend/app/api/ollama/batch/route.ts
index da59cd0..047ae5e 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/ollama/batch/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/ollama/batch/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { llmService, LLMMessage } from '@/lib/llm-service';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/ollama/route.ts b/nvidia/txt2kg/assets/frontend/app/api/ollama/route.ts
index 499e81e..819d3c3 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/ollama/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/ollama/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { llmService } from '@/lib/llm-service';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/ollama/test/route.ts b/nvidia/txt2kg/assets/frontend/app/api/ollama/test/route.ts
index f0545d9..a5c24e0 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/ollama/test/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/ollama/test/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
/**
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/clear/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/clear/route.ts
index f73d7cf..d2491b7 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/clear/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/clear/route.ts
@@ -1,5 +1,21 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
-import { PineconeService } from '@/lib/pinecone';
+import { QdrantService } from '@/lib/qdrant';
/**
* Clear all data from the Pinecone vector database
@@ -7,7 +23,7 @@ import { PineconeService } from '@/lib/pinecone';
*/
export async function POST() {
// Get the Pinecone service instance
- const pineconeService = PineconeService.getInstance();
+ const pineconeService = QdrantService.getInstance();
// Clear all vectors from the database
const deleteSuccess = await pineconeService.deleteAllEntities();
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/create-index/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/create-index/route.ts
index 7ce0f5c..749dacc 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/create-index/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/create-index/route.ts
@@ -1,5 +1,21 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextResponse } from 'next/server';
-import { PineconeService } from '@/lib/pinecone';
+import { QdrantService } from '@/lib/qdrant';
/**
* Create Pinecone index API endpoint
@@ -8,7 +24,7 @@ import { PineconeService } from '@/lib/pinecone';
export async function POST() {
try {
// Get the Pinecone service instance
- const pineconeService = PineconeService.getInstance();
+ const pineconeService = QdrantService.getInstance();
// Force re-initialization to create the index
(pineconeService as any).initialized = false;
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/stats/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/stats/route.ts
index a1aa129..e9159e0 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/stats/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pinecone-diag/stats/route.ts
@@ -1,5 +1,21 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
-import { PineconeService } from '@/lib/pinecone';
+import { QdrantService } from '@/lib/qdrant';
/**
* Get Pinecone vector database stats
@@ -7,7 +23,7 @@ import { PineconeService } from '@/lib/pinecone';
export async function GET() {
try {
// Initialize Pinecone service
- const pineconeService = PineconeService.getInstance();
+ const pineconeService = QdrantService.getInstance();
// We can now directly call getStats() which handles initialization and error recovery
const stats = await pineconeService.getStats();
diff --git a/nvidia/txt2kg/assets/frontend/app/api/process-document/route.ts b/nvidia/txt2kg/assets/frontend/app/api/process-document/route.ts
index 9182ff5..f18ba9c 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/process-document/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/process-document/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { RemoteBackendService } from '@/lib/remote-backend';
import { EmbeddingsService } from '@/lib/embeddings';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/[taskId]/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/[taskId]/route.ts
index 38c6536..1924744 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/[taskId]/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/[taskId]/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const PYGRAPHISTRY_SERVICE_URL = process.env.PYGRAPHISTRY_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/route.ts
index 5405d2b..37caf86 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/generate/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const PYGRAPHISTRY_SERVICE_URL = process.env.PYGRAPHISTRY_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/health/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/health/route.ts
index 4110185..ce98920 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/health/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/health/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const PYGRAPHISTRY_SERVICE_URL = process.env.PYGRAPHISTRY_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/stats/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/stats/route.ts
index 163a5b7..1ab0710 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/stats/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/stats/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const PYGRAPHISTRY_SERVICE_URL = process.env.PYGRAPHISTRY_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/visualize/route.ts b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/visualize/route.ts
index 9c916ee..31c8bb5 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/visualize/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/pygraphistry/visualize/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const PYGRAPHISTRY_SERVICE_URL = process.env.PYGRAPHISTRY_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/query-log/add/route.ts b/nvidia/txt2kg/assets/frontend/app/api/query-log/add/route.ts
index b1283bd..af24b5b 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/query-log/add/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/query-log/add/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import neo4jService from '@/lib/neo4j';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/query-log/route.ts b/nvidia/txt2kg/assets/frontend/app/api/query-log/route.ts
index c9ed02d..6019426 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/query-log/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/query-log/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import queryLoggerService from '@/lib/query-logger';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/query-log/test/route.ts b/nvidia/txt2kg/assets/frontend/app/api/query-log/test/route.ts
index b22b31c..7cec9a3 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/query-log/test/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/query-log/test/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import neo4jService from '@/lib/neo4j';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/query/route.ts b/nvidia/txt2kg/assets/frontend/app/api/query/route.ts
index ede7853..9619364 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/query/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/query/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import backendService from '@/lib/backend-service';
import type { Triple } from '@/types/graph';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/rag-query/route.ts b/nvidia/txt2kg/assets/frontend/app/api/rag-query/route.ts
index 00fec6b..474544f 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/rag-query/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/rag-query/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import RAGService from '@/lib/rag';
@@ -20,17 +36,20 @@ export async function POST(req: NextRequest) {
const ragService = RAGService;
await ragService.initialize();
- console.log(`Processing RAG query: "${query}" with topK=${topK}`);
+ console.log(`Processing Pure RAG query: "${query}" with topK=${topK}`);
// Retrieve documents and generate answer
- const answer = await ragService.retrievalQA(query, topK);
+ const result = await ragService.retrievalQA(query, topK);
// Check if this is a fallback response
- const isGeneralKnowledgeFallback = answer.startsWith('[Note: No specific information was found');
+ const isGeneralKnowledgeFallback = result.answer.startsWith('[Note: No specific information was found');
+
+ console.log(`β Pure RAG query completed. Retrieved ${result.documentCount} document chunks`);
// Return the results
return NextResponse.json({
- answer,
+ answer: result.answer,
+ documentCount: result.documentCount,
usedFallback: isGeneralKnowledgeFallback,
success: true
});
diff --git a/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu-stream/[sessionId]/route.ts b/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu-stream/[sessionId]/route.ts
index cd57fe0..119f029 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu-stream/[sessionId]/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu-stream/[sessionId]/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
// Proxy route specifically for WebRTC streaming frames
diff --git a/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu/[...path]/route.ts b/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu/[...path]/route.ts
index 109b0fc..78d2c2f 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu/[...path]/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/remote-webgpu/[...path]/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
// Proxy route for remote WebGPU clustering service
diff --git a/nvidia/txt2kg/assets/frontend/app/api/sentence-embeddings/route.ts b/nvidia/txt2kg/assets/frontend/app/api/sentence-embeddings/route.ts
index c23c0bb..7fc1905 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/sentence-embeddings/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/sentence-embeddings/route.ts
@@ -1,6 +1,22 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { processSentenceEmbeddings, SentenceEmbedding } from '@/lib/text-processor';
-import { PineconeService } from '@/lib/pinecone';
+import { QdrantService } from '@/lib/qdrant';
/**
* API endpoint for splitting text into sentences and generating embeddings
@@ -49,7 +65,7 @@ export async function POST(req: NextRequest) {
});
// Store in Pinecone
- const pineconeService = PineconeService.getInstance();
+ const pineconeService = QdrantService.getInstance();
await pineconeService.storeEmbeddingsWithMetadata(
embeddingsMap,
textContentMap,
diff --git a/nvidia/txt2kg/assets/frontend/app/api/settings/route.ts b/nvidia/txt2kg/assets/frontend/app/api/settings/route.ts
index e091f38..5875395 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/settings/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/settings/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { GraphDBType } from '@/lib/graph-db-service';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/stop-embeddings/route.ts b/nvidia/txt2kg/assets/frontend/app/api/stop-embeddings/route.ts
index f8b026b..55cb1af 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/stop-embeddings/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/stop-embeddings/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
// Global flag to track if embeddings generation should be stopped
diff --git a/nvidia/txt2kg/assets/frontend/app/api/stop-processing/route.ts b/nvidia/txt2kg/assets/frontend/app/api/stop-processing/route.ts
index c5b3128..807df2b 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/stop-processing/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/stop-processing/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
// Global flag to track if processing should be stopped
diff --git a/nvidia/txt2kg/assets/frontend/app/api/store-documents/route.ts b/nvidia/txt2kg/assets/frontend/app/api/store-documents/route.ts
index af4a8c3..0e3e2b2 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/store-documents/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/store-documents/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import RAGService from '@/lib/rag';
diff --git a/nvidia/txt2kg/assets/frontend/app/api/test-proxy/route.ts b/nvidia/txt2kg/assets/frontend/app/api/test-proxy/route.ts
index 5e4c77c..77bad52 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/test-proxy/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/test-proxy/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
// Simple test endpoint to verify proxy connectivity
diff --git a/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/capabilities/route.ts b/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/capabilities/route.ts
index ff01d58..dc10b4a 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/capabilities/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/capabilities/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const UNIFIED_GPU_SERVICE_URL = process.env.UNIFIED_GPU_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/visualize/route.ts b/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/visualize/route.ts
index fb4fbd1..2193362 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/visualize/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/unified-gpu/visualize/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server'
const UNIFIED_GPU_SERVICE_URL = process.env.UNIFIED_GPU_SERVICE_URL || 'http://localhost:8080'
diff --git a/nvidia/txt2kg/assets/frontend/app/api/vllm/route.ts b/nvidia/txt2kg/assets/frontend/app/api/vllm/route.ts
index 4301a53..0a72b3e 100644
--- a/nvidia/txt2kg/assets/frontend/app/api/vllm/route.ts
+++ b/nvidia/txt2kg/assets/frontend/app/api/vllm/route.ts
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { NextRequest, NextResponse } from 'next/server';
import { LLMService } from '@/lib/llm-service';
diff --git a/nvidia/txt2kg/assets/frontend/app/components/documents-list.tsx b/nvidia/txt2kg/assets/frontend/app/components/documents-list.tsx
index e89b83e..f535b25 100644
--- a/nvidia/txt2kg/assets/frontend/app/components/documents-list.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/components/documents-list.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client";
import { useState, useEffect } from "react";
diff --git a/nvidia/txt2kg/assets/frontend/app/document-data/page.tsx b/nvidia/txt2kg/assets/frontend/app/document-data/page.tsx
index b17fcff..f5073cf 100644
--- a/nvidia/txt2kg/assets/frontend/app/document-data/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/document-data/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import { DocumentsTable } from "@/components/documents-table";
import { DocumentProcessor } from "@/components/document-processor";
diff --git a/nvidia/txt2kg/assets/frontend/app/graph/page.tsx b/nvidia/txt2kg/assets/frontend/app/graph/page.tsx
index 1458303..d3eb3a4 100644
--- a/nvidia/txt2kg/assets/frontend/app/graph/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/graph/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import { useEffect, useState } from "react"
diff --git a/nvidia/txt2kg/assets/frontend/app/graph3d/page.tsx b/nvidia/txt2kg/assets/frontend/app/graph3d/page.tsx
index 2e6dfc2..96b5196 100644
--- a/nvidia/txt2kg/assets/frontend/app/graph3d/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/graph3d/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
/**
diff --git a/nvidia/txt2kg/assets/frontend/app/layout.tsx b/nvidia/txt2kg/assets/frontend/app/layout.tsx
index 1de1542..e28b19d 100644
--- a/nvidia/txt2kg/assets/frontend/app/layout.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/layout.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import type React from "react"
import type { Metadata } from "next"
import { Inter } from "next/font/google"
@@ -43,9 +59,6 @@ export default function RootLayout({
txt2kg
-
- Powered by NVIDIA AI
-
diff --git a/nvidia/txt2kg/assets/frontend/app/page.tsx b/nvidia/txt2kg/assets/frontend/app/page.tsx
index 741415e..2289d69 100644
--- a/nvidia/txt2kg/assets/frontend/app/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import { useState, useEffect } from "react"
diff --git a/nvidia/txt2kg/assets/frontend/app/rag/metrics/page.tsx b/nvidia/txt2kg/assets/frontend/app/rag/metrics/page.tsx
index 304ad72..43986e8 100644
--- a/nvidia/txt2kg/assets/frontend/app/rag/metrics/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/rag/metrics/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client";
import { useState, useEffect } from "react";
@@ -17,6 +33,7 @@ interface MetricsData {
recall: number;
f1Score: number;
topQueries: { query: string; count: number }[];
+ queryTimesByMode?: Record;
queryLogStats?: {
totalQueryLogs: number;
totalExecutions: number;
diff --git a/nvidia/txt2kg/assets/frontend/app/rag/page.tsx b/nvidia/txt2kg/assets/frontend/app/rag/page.tsx
index 3de4886..f5d7186 100644
--- a/nvidia/txt2kg/assets/frontend/app/rag/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/rag/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client";
import { useState, useEffect } from "react";
@@ -12,6 +28,7 @@ import { ArrowLeft, BarChart2, Search as SearchIcon } from "lucide-react";
export default function RagPage() {
const router = useRouter();
const [results, setResults] = useState(null);
+ const [llmAnswer, setLlmAnswer] = useState(null);
const [isLoading, setIsLoading] = useState(false);
const [errorMessage, setErrorMessage] = useState(null);
const [vectorEnabled, setVectorEnabled] = useState(false);
@@ -20,6 +37,7 @@ export default function RagPage() {
avgRelevance: number;
precision: number;
recall: number;
+ queryTimesByMode?: Record;
} | null>(null);
const [currentParams, setCurrentParams] = useState({
kNeighbors: 4096,
@@ -64,7 +82,8 @@ export default function RagPage() {
avgQueryTime: data.avgQueryTime,
avgRelevance: data.avgRelevance,
precision: data.precision,
- recall: data.recall
+ recall: data.recall,
+ queryTimesByMode: data.queryTimesByMode
});
}
} catch (error) {
@@ -84,12 +103,20 @@ export default function RagPage() {
let resultCount = 0;
let relevanceScore = 0;
+ // Debug logging
+ console.log('π Query params:', {
+ usePureRag: params.usePureRag,
+ useVectorSearch: params.useVectorSearch,
+ vectorEnabled,
+ queryMode: params.queryMode
+ });
+
try {
// If using pure RAG (Pinecone + LangChain) without graph search
if (params.usePureRag) {
queryMode = 'pure-rag';
try {
- console.log('Using pure RAG with just Pinecone and LangChain for query:', query);
+ console.log('Using pure RAG with Qdrant and NVIDIA LLM for query:', query);
const ragResponse = await fetch('/api/rag-query', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
@@ -101,17 +128,22 @@ export default function RagPage() {
if (ragResponse.ok) {
const data = await ragResponse.json();
+ console.log('π₯ RAG Response data:', {
+ hasAnswer: !!data.answer,
+ answerLength: data.answer?.length,
+ documentCount: data.documentCount
+ });
// Handle the answer - we might need to display differently than triples
if (data.answer) {
- // Special UI handling for text answer rather than triples
- setResults([{
- subject: 'Answer',
- predicate: '',
- object: data.answer,
- usedFallback: data.usedFallback
- }]);
-
- resultCount = 1;
+ console.log('β Setting answer in results:', data.answer.substring(0, 100) + '...');
+
+ // Set the LLM answer for display (same as traditional mode)
+ setLlmAnswer(data.answer);
+
+ // Set empty results array since Pure RAG doesn't return triples
+ setResults([]);
+
+ resultCount = data.documentCount || 0;
relevanceScore = data.relevanceScore || 0;
// Log the query with performance metrics
@@ -121,7 +153,7 @@ export default function RagPage() {
resultCount
});
- console.log('Pure RAG query completed successfully');
+ console.log(`β Pure RAG query completed. Retrieved ${resultCount} document chunks`);
setIsLoading(false);
return;
}
@@ -136,8 +168,8 @@ export default function RagPage() {
}
}
- // If we have vector embeddings, use enhanced query with metadata
- if (vectorEnabled && params.useVectorSearch) {
+ // If we have vector embeddings AND explicitly selected vector search, use enhanced query with metadata
+ if (vectorEnabled && params.useVectorSearch && !params.usePureRag) {
queryMode = 'vector-search';
try {
console.log('Using enhanced RAG with LangChain for query:', query);
@@ -184,35 +216,71 @@ export default function RagPage() {
}
}
- // Call the traditional backend API as fallback or if explicitly selected
+ // Call the LLM-enhanced graph query API
+ console.log('β Using Graph Search + LLM approach');
queryMode = 'traditional';
- const response = await fetch(`/api/query`, {
+
+ // Get selected LLM model from localStorage
+ let llmModel = undefined;
+ let llmProvider = undefined;
+ try {
+ const savedModel = localStorage.getItem("selectedModelForRAG");
+ if (savedModel) {
+ const modelData = JSON.parse(savedModel);
+ llmModel = modelData.model;
+ llmProvider = modelData.provider;
+ console.log(`Using LLM: ${llmModel} (${llmProvider})`);
+ }
+ } catch (e) {
+ console.warn("Could not load selected LLM model, using default");
+ }
+
+ const response = await fetch(`/api/graph-query-llm`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({
query,
- kNeighbors: params.kNeighbors,
- fanout: params.fanout,
- numHops: params.numHops,
- topK: params.topK,
- queryMode: queryMode, // Explicitly pass the query mode
- useTraditional: true // Force use of the direct pattern matching approach
+ topK: params.topK || 5,
+ useTraditional: true,
+ llmModel,
+ llmProvider
}),
});
if (!response.ok) {
const errorData = await response.json();
- throw new Error(errorData.error || 'Failed to query the RAG backend');
+ throw new Error(errorData.error || 'Failed to query with LLM');
}
const data = await response.json();
+
+ // Log sample of retrieved triples for debugging
+ console.log('π Retrieved Triples (sample):', data.triples.slice(0, 3));
+ console.log('π€ LLM-Generated Answer (preview):', data.answer?.substring(0, 200) + '...');
+ console.log('π Triple Count:', data.count);
+
+ // DEBUG: Check if depth/pathLength are present in received data
+ if (data.triples && data.triples.length > 0) {
+ console.log('π First triple structure:', JSON.stringify(data.triples[0], null, 2));
+ console.log('π Has depth?', 'depth' in data.triples[0]);
+ console.log('π Has pathLength?', 'pathLength' in data.triples[0]);
+ }
- // Update the results
- setResults(data.relevantTriples || []);
+ // Update the results with the triples (for display)
+ setResults(data.triples || []);
resultCount = data.count || 0;
- relevanceScore = data.relevanceScore || 0;
+ relevanceScore = 0; // No relevance score for traditional search
+
+ // Store the LLM answer for display
+ if (data.answer) {
+ console.log('β Setting llmAnswer state (length:', data.answer.length, 'chars)');
+ setLlmAnswer(data.answer);
+ } else {
+ console.log('β οΈ No answer in response');
+ setLlmAnswer(null);
+ }
// Log the query with performance metrics
logQuery(query, queryMode, {
@@ -279,6 +347,7 @@ export default function RagPage() {
const clearResults = () => {
setResults(null);
+ setLlmAnswer(null);
setErrorMessage(null);
};
@@ -318,22 +387,34 @@ export default function RagPage() {
diff --git a/nvidia/txt2kg/assets/frontend/app/test-large-graph/page.tsx b/nvidia/txt2kg/assets/frontend/app/test-large-graph/page.tsx
index bbebef3..20ff789 100644
--- a/nvidia/txt2kg/assets/frontend/app/test-large-graph/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/test-large-graph/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import React, { useState, useEffect } from "react"
diff --git a/nvidia/txt2kg/assets/frontend/app/test-million-nodes/page.tsx b/nvidia/txt2kg/assets/frontend/app/test-million-nodes/page.tsx
index aec3ba3..2ae800b 100644
--- a/nvidia/txt2kg/assets/frontend/app/test-million-nodes/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/test-million-nodes/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import React, { useState, useEffect, useCallback, useMemo } from "react"
diff --git a/nvidia/txt2kg/assets/frontend/app/test-webgpu-clustering/page.tsx b/nvidia/txt2kg/assets/frontend/app/test-webgpu-clustering/page.tsx
index 3863f9e..8b4be39 100644
--- a/nvidia/txt2kg/assets/frontend/app/test-webgpu-clustering/page.tsx
+++ b/nvidia/txt2kg/assets/frontend/app/test-webgpu-clustering/page.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
// @ts-nocheck
diff --git a/nvidia/txt2kg/assets/frontend/components/advanced-options.tsx b/nvidia/txt2kg/assets/frontend/components/advanced-options.tsx
index 5922a0a..167179f 100644
--- a/nvidia/txt2kg/assets/frontend/components/advanced-options.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/advanced-options.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
import React, { useState } from "react";
import { ChevronDown, ChevronRight } from "lucide-react";
import { cn } from "@/lib/utils";
diff --git a/nvidia/txt2kg/assets/frontend/components/api-key-prompt.tsx b/nvidia/txt2kg/assets/frontend/components/api-key-prompt.tsx
index 486aa3c..008fc61 100644
--- a/nvidia/txt2kg/assets/frontend/components/api-key-prompt.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/api-key-prompt.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import type React from "react"
diff --git a/nvidia/txt2kg/assets/frontend/components/client-init.tsx b/nvidia/txt2kg/assets/frontend/components/client-init.tsx
index 5e7fe85..625cf3d 100644
--- a/nvidia/txt2kg/assets/frontend/components/client-init.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/client-init.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import { useEffect } from "react"
diff --git a/nvidia/txt2kg/assets/frontend/components/database-connection.tsx b/nvidia/txt2kg/assets/frontend/components/database-connection.tsx
index e1ddc80..4ddaa35 100644
--- a/nvidia/txt2kg/assets/frontend/components/database-connection.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/database-connection.tsx
@@ -1,7 +1,23 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import { useState, useEffect } from "react"
-import { Network, Database, Zap, AlertCircle, RefreshCw, ChevronDown, ChevronUp, InfoIcon, Trash2 } from "lucide-react"
+import { Network, Database, Zap, AlertCircle, RefreshCw, ChevronDown, ChevronUp, InfoIcon, Trash2, LogOut } from "lucide-react"
import { Badge } from '@/components/ui/badge'
import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@/components/ui/tooltip'
import { Button } from "@/components/ui/button"
@@ -157,15 +173,20 @@ export function DatabaseConnection({ className }: DatabaseConnectionProps) {
try {
const response = await fetch('/api/pinecone-diag/stats');
const data = await response.json();
-
+
if (response.ok) {
setVectorStats({
nodes: typeof data.totalVectorCount === 'number' ? data.totalVectorCount : 0,
relationships: 0, // Vector DB doesn't store relationships
source: data.source || 'unknown',
- httpHealthy: data.httpHealthy
- });
-
+ httpHealthy: data.httpHealthy,
+ // Store additional Qdrant stats
+ ...(data.status && { status: data.status }),
+ ...(data.vectorSize && { vectorSize: data.vectorSize }),
+ ...(data.distance && { distance: data.distance }),
+ ...(data.url && { url: data.url }),
+ } as any);
+
// If we have a healthy HTTP connection, we're connected
if (data.httpHealthy) {
setVectorConnectionStatus("connected");
@@ -364,39 +385,65 @@ export function DatabaseConnection({ className }: DatabaseConnectionProps) {
)}
- No graph data available for PyGraphistry visualization
-
-
- Please load graph data to enable GPU-accelerated visualization
-
-
-
- )}
-
-
-
+
+
+
+
diff --git a/nvidia/txt2kg/assets/frontend/components/fallback-graph.tsx b/nvidia/txt2kg/assets/frontend/components/fallback-graph.tsx
index 7c96df3..e5414c8 100644
--- a/nvidia/txt2kg/assets/frontend/components/fallback-graph.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/fallback-graph.tsx
@@ -1,3 +1,19 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import type React from "react"
diff --git a/nvidia/txt2kg/assets/frontend/components/force-graph-wrapper.tsx b/nvidia/txt2kg/assets/frontend/components/force-graph-wrapper.tsx
index 6b94b65..c74bae3 100644
--- a/nvidia/txt2kg/assets/frontend/components/force-graph-wrapper.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/force-graph-wrapper.tsx
@@ -1,10 +1,25 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
import React, { useEffect, useRef, useState, useCallback } from "react"
import type { Triple } from "@/utils/text-processing"
import { Maximize2, Minimize2, Pause, Play, RefreshCw, ZoomIn, X, LayoutGrid } from "lucide-react"
import { WebGPUClusteringEngine } from "@/utils/webgpu-clustering"
-import { EnhancedWebGPUClusteringEngine } from "@/utils/remote-webgpu-clustering"
import * as d3 from 'd3'
import * as THREE from 'three'
@@ -389,7 +404,7 @@ export function ForceGraphWrapper({
const [isInitialized, setIsInitialized] = useState(false);
// Add WebGPU clustering engine ref
- const clusteringEngineRef = useRef(null); // Can be WebGPUClusteringEngine or EnhancedWebGPUClusteringEngine
+ const clusteringEngineRef = useRef(null);
// Track if WebGPU clustering is available
const [isClusteringAvailable, setIsClusteringAvailable] = useState(false);
// Track if clustering is enabled
@@ -1498,7 +1513,7 @@ export function ForceGraphWrapper({
}
}, [graphRef.current, jsonData]);
- // Enhanced WebGPU clustering initialization for 3D views with remote fallback
+ // WebGPU clustering initialization for 3D views
useEffect(() => {
async function initClustering() {
try {
@@ -1506,36 +1521,22 @@ export function ForceGraphWrapper({
if (layoutType === '3d' && !clusteringEngineRef.current) {
console.log("π§ Initializing clustering engine for 3D view...");
- // Use enhanced clustering engine for 3D views with remote fallback
- const enhancedEngine = new EnhancedWebGPUClusteringEngine([32, 18, 24], 'http://localhost:8083');
+ // Use WebGPU clustering engine for 3D views
+ const engine = new WebGPUClusteringEngine([32, 18, 24]);
console.log("β³ Waiting for engine initialization...");
- // Wait longer for proper initialization (remote service check takes time)
- await new Promise(resolve => setTimeout(resolve, 2000)); // Increased timeout even more
+ await new Promise(resolve => setTimeout(resolve, 1000));
- console.log("π Checking enhanced engine availability...");
- console.log("Engine available:", enhancedEngine.isAvailable());
- console.log("Engine using remote:", enhancedEngine.isUsingRemote());
- console.log("Engine capabilities:", enhancedEngine.getCapabilities());
+ console.log("π Checking engine availability...");
+ console.log("Engine available:", engine.isAvailable());
// Store the engine reference regardless of availability for debugging
- clusteringEngineRef.current = enhancedEngine;
+ clusteringEngineRef.current = engine;
- if (enhancedEngine.isAvailable()) {
+ if (engine.isAvailable()) {
setIsClusteringAvailable(true);
-
- if (enhancedEngine.isUsingRemote()) {
- console.log("β Using remote WebGPU clustering service for 3D view");
- setUsingCpuFallback(false); // Remote GPU is available
-
- // Set up event listeners for remote clustering updates
- enhancedEngine.on('clusteringComplete', (result: any) => {
- console.log(`π Remote clustering completed in ${result.processingTime}s`);
- });
- } else {
- console.log("β Local WebGPU clustering engine initialized for 3D view");
- setUsingCpuFallback(false); // Local WebGPU is also not CPU fallback
- }
+ console.log("β Local WebGPU clustering engine initialized for 3D view");
+ setUsingCpuFallback(false);
// Auto-enable clustering for large 3D graphs
if (graphData && graphData.nodes && graphData.nodes.length > 200) {
@@ -2357,7 +2358,7 @@ export function ForceGraphWrapper({
try {
console.log("π Applying GPU clustering to", graphData.nodes.length, "nodes");
- // Use the correct updateNodePositions method from EnhancedWebGPUClusteringEngine
+ // Use the updateNodePositions method from WebGPUClusteringEngine
const success = await clusteringEngineRef.current.updateNodePositions(
graphData.nodes,
graphData.links || []
diff --git a/nvidia/txt2kg/assets/frontend/components/graph-actions.tsx b/nvidia/txt2kg/assets/frontend/components/graph-actions.tsx
index 71bc34a..c3fb862 100644
--- a/nvidia/txt2kg/assets/frontend/components/graph-actions.tsx
+++ b/nvidia/txt2kg/assets/frontend/components/graph-actions.tsx
@@ -1,16 +1,27 @@
+//
+// SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
"use client"
-import { Network, Zap, HelpCircle } from "lucide-react"
+import { Network, Zap } from "lucide-react"
import { useDocuments } from "@/contexts/document-context"
import { Loader2 } from "lucide-react"
-import { useState } from "react"
-import { Switch } from "@/components/ui/switch"
-import { Label } from "@/components/ui/label"
-import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip"
export function GraphActions() {
const { documents, processDocuments, isProcessing, openGraphVisualization } = useDocuments()
- const [useLangChain, setUseLangChain] = useState(false)
const hasNewDocuments = documents.some((doc) => doc.status === "New")
const hasProcessedDocuments = documents.some(
@@ -30,7 +41,7 @@ export function GraphActions() {
}
await processDocuments(newDocumentIds, {
- useLangChain,
+ useLangChain: false,
useGraphTransformer: false,
promptConfigs: undefined
});
@@ -41,34 +52,6 @@ export function GraphActions() {
return (