@tonyclaw/agent-inspector 2.0.1 → 2.0.3

package/src/routes/api/providers.$providerId.test.log.ts

@@ -1,7 +1,7 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { getProvider, getModelUsageName } from "../../proxy/providers";
-import { appendLogEntry } from "../../proxy/logger";
 import { addTestLogEntry } from "../../proxy/store";
+import { buildProviderTestRequestBody } from "../../lib/providerTestPrompt";
 import {
   ProviderTestResultsSchema,
   type ProviderTestResult as TestResult,
@@ -12,38 +12,6 @@ function hasSuccessField(result: ProviderTestState): result is TestResult {
   return Object.prototype.hasOwnProperty.call(result, "success");
 }
 
-function createTestLogEntry(
-  providerName: string,
-  path: string,
-  body: string,
-  upstreamUrl: string,
-  result: TestResult,
-  isTest: boolean,
-): Record<string, unknown> {
-  return {
-    timestamp: new Date().toISOString(),
-    id: `test-${Date.now()}`,
-    method: "POST",
-    path,
-    model: isTest ? result.model : undefined,
-    sessionId: null,
-    rawRequestBody: body,
-    responseStatus: result.success ? 200 : 500,
-    responseText: result.rawResponse ?? JSON.stringify(result),
-    inputTokens: result.inputTokens ?? null,
-    outputTokens: result.outputTokens ?? null,
-    elapsedMs: result.latencyMs ?? 0,
-    streaming: result.streaming ?? false,
-    userAgent: "provider-test",
-    origin: null,
-    upstreamUrl,
-    error: result.success ? null : (result.error?.message ?? String(result.error)),
-    isTest: true,
-    providerName,
-    headers: result.requestHeaders ?? {},
-  };
-}
-
 async function logModelResults(
   displayName: string,
   providerName: string,
@@ -63,12 +31,7 @@ async function logModelResults(
       const nonStreamingResult = nsResult;
       const streamingResult = sResult;
 
-      const requestBody = JSON.stringify({
-        model: usageModel,
-        messages: [{ role: "user", content: "say hello and briefly introduce yourself" }],
-        max_tokens: 1024,
-      });
-      const upstreamUrl = `${anthropicUrl}/v1/messages`;
+      const requestBody = JSON.stringify(buildProviderTestRequestBody(usageModel, "non-streaming"));
 
       await addTestLogEntry({
         timestamp: new Date().toISOString(),
@@ -93,24 +56,10 @@ async function logModelResults(
         headers: nonStreamingResult.requestHeaders ?? {},
       });
 
-      appendLogEntry(
-        createTestLogEntry(
-          providerName,
-          "/v1/messages",
-          requestBody,
-          upstreamUrl,
-          nonStreamingResult,
-          true,
-        ),
+      const streamingRequestBody = JSON.stringify(
+        buildProviderTestRequestBody(usageModel, "streaming"),
       );
 
-      const streamingRequestBody = JSON.stringify({
-        model: usageModel,
-        messages: [{ role: "user", content: "say hello" }],
-        max_tokens: 256,
-        stream: true,
-      });
-
       await addTestLogEntry({
         timestamp: new Date().toISOString(),
         method: "POST",
@@ -134,17 +83,6 @@ async function logModelResults(
         providerName,
         headers: streamingResult.requestHeaders ?? {},
       });
-
-      appendLogEntry(
-        createTestLogEntry(
-          providerName,
-          "/v1/messages",
-          streamingRequestBody,
-          upstreamUrl,
-          streamingResult,
-          true,
-        ),
-      );
     }
   }
 
@@ -155,12 +93,7 @@ async function logModelResults(
       const nonStreamingResult = nsResult;
       const streamingResult = sResult;
 
-      const requestBody = JSON.stringify({
-        model: usageModel,
-        messages: [{ role: "user", content: "say hello and briefly introduce yourself" }],
-        max_tokens: 1024,
-      });
-      const upstreamUrl = `${openaiUrl}/v1/chat/completions`;
+      const requestBody = JSON.stringify(buildProviderTestRequestBody(usageModel, "non-streaming"));
 
       await addTestLogEntry({
         timestamp: new Date().toISOString(),
@@ -185,24 +118,10 @@ async function logModelResults(
         headers: nonStreamingResult.requestHeaders ?? {},
       });
 
-      appendLogEntry(
-        createTestLogEntry(
-          providerName,
-          "/v1/chat/completions",
-          requestBody,
-          upstreamUrl,
-          nonStreamingResult,
-          true,
-        ),
+      const streamingRequestBody = JSON.stringify(
+        buildProviderTestRequestBody(usageModel, "streaming"),
       );
 
-      const streamingRequestBody = JSON.stringify({
-        model: usageModel,
-        messages: [{ role: "user", content: "say hello" }],
-        max_tokens: 256,
-        stream: true,
-      });
-
       await addTestLogEntry({
         timestamp: new Date().toISOString(),
         method: "POST",
@@ -226,17 +145,6 @@ async function logModelResults(
         providerName,
         headers: streamingResult.requestHeaders ?? {},
       });
-
-      appendLogEntry(
-        createTestLogEntry(
-          providerName,
-          "/v1/chat/completions",
-          streamingRequestBody,
-          upstreamUrl,
-          streamingResult,
-          true,
-        ),
-      );
     }
   }
 }