refactor: Minor updates

Author: tjtanjin

src/config.ts

@@ -39,10 +39,10 @@ export const config = {
   // Server Configuration
   port: process.env.PORT || 8080,
 
+  ragConversationWindowSize: parseInt(process.env.RAG_CONVERSATION_WINDOW_SIZE || '', 10) || 1,
+
   // RAG Management API Key (protects management endpoints)
   ragManagementApiKey: process.env.RAG_MANAGEMENT_API_KEY!,
 
   ragQueryApiKey: process.env.RAG_QUERY_API_KEY!,
-
-  ragConversationWindowSize: parseInt(process.env.RAG_CONVERSATION_WINDOW_SIZE || '', 10) || 1,
 };

src/controllers/geminiQuery.ts

@@ -128,8 +128,8 @@ export const handleGeminiBatch = async (req: Request, res: Response) => {
 
     try {
       const llmResponse = (await generateText({
-        model: model,
-        contents: contentsForLlm, // Pass the (potentially RAG-augmented) GeminiContent[]
+        contents: contentsForLlm,
+        model: model, // Pass the (potentially RAG-augmented) GeminiContent[]
         stream: false,
       })) as LLMChatResponse;
       res.status(200).json(llmResponse);
@@ -298,11 +298,11 @@ export const handleGeminiStream = async (req: Request, res: Response) => {
 
     try {
       await generateText({
+        contents: contentsForLlm,
         model: model,
         onChunk: (rawSseLine: string) => {
           res.write(`${rawSseLine}\n`);
-        },
-        contents: contentsForLlm, // Pass the (potentially RAG-augmented) GeminiContent[]
+        }, // Pass the (potentially RAG-augmented) GeminiContent[]
         stream: true,
       });
       res.end();

src/index.ts

@@ -1,3 +1,13 @@
+import { logs as apiLogs } from '@opentelemetry/api-logs';
+import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-grpc';
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-grpc';
+import { ExpressInstrumentation } from '@opentelemetry/instrumentation-express';
+import { HttpInstrumentation } from '@opentelemetry/instrumentation-http';
+import { Resource } from '@opentelemetry/resources';
+import { BatchLogRecordProcessor, LoggerProvider } from '@opentelemetry/sdk-logs';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-node';
+import { ATTR_SERVICE_NAME } from '@opentelemetry/semantic-conventions';
 import bodyParser from 'body-parser';
 import cors from 'cors';
 import express from 'express';
@@ -10,16 +20,6 @@ import Logger from './logger';
 import { geminiRouter } from './routers/geminiQuery';
 import { ragManagementRouter } from './routers/ragManagement';
 import swaggerDocument from './swagger';
-import { logs as apiLogs } from '@opentelemetry/api-logs';
-import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-grpc';
-import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-grpc';
-import { ExpressInstrumentation } from '@opentelemetry/instrumentation-express';
-import { HttpInstrumentation } from '@opentelemetry/instrumentation-http';
-import { Resource } from '@opentelemetry/resources';
-import { BatchLogRecordProcessor, LoggerProvider } from '@opentelemetry/sdk-logs';
-import { NodeSDK } from '@opentelemetry/sdk-node';
-import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-node';
-import { ATTR_SERVICE_NAME } from '@opentelemetry/semantic-conventions';
 
 // OpenTelemetry SDK Initialization