mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-23 18:33:54 +00:00
71 lines
2.5 KiB
TypeScript
71 lines
2.5 KiB
TypeScript
//
|
|
// 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';
|
|
|
|
/**
|
|
* API endpoint to create a test query log
|
|
* This is for debugging purposes only
|
|
*/
|
|
export async function GET(request: NextRequest) {
|
|
try {
|
|
console.log('[Test] Creating test query log');
|
|
|
|
// Initialize Neo4j if not already
|
|
if (!neo4jService.isInitialized()) {
|
|
console.log('[Test] Initializing Neo4j service');
|
|
neo4jService.initialize();
|
|
}
|
|
|
|
// Get query text from URL parameters or use a default
|
|
const query = request.nextUrl.searchParams.get('query') || 'Test query for debugging';
|
|
const queryMode = (request.nextUrl.searchParams.get('mode') || 'traditional') as 'traditional' | 'vector-search' | 'pure-rag';
|
|
const executionTime = parseInt(request.nextUrl.searchParams.get('time') || '300');
|
|
const resultCount = parseInt(request.nextUrl.searchParams.get('count') || '5');
|
|
|
|
console.log(`[Test] Adding test query: "${query}" (${queryMode})`);
|
|
|
|
// Log the query with some test metrics
|
|
await neo4jService.logQuery(
|
|
query,
|
|
queryMode,
|
|
{
|
|
executionTimeMs: executionTime,
|
|
relevanceScore: 0,
|
|
precision: 0,
|
|
recall: 0,
|
|
resultCount: resultCount
|
|
}
|
|
);
|
|
|
|
// Get current query logs to verify
|
|
const logs = await neo4jService.getQueryLogs(10);
|
|
|
|
return NextResponse.json({
|
|
success: true,
|
|
message: `Test query "${query}" added successfully`,
|
|
logs: logs.slice(0, 3) // Return top 3 logs for verification
|
|
});
|
|
} catch (error) {
|
|
console.error('[Test] Error creating test query log:', error);
|
|
const errorMessage = error instanceof Error ? error.message : 'Unknown error';
|
|
return NextResponse.json(
|
|
{ error: errorMessage },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|