import { z as zodV3 } from "zod/v3";
import {
  ChatMessageRole,
  ChatMessageType,
  LLMApiKeySchema,
  type ModelConfig,
} from "./types";
import { decrypt } from "../../encryption";
import { fetchLLMCompletion } from "./fetchLLMCompletion";
import { decryptAndParseExtraHeaders } from "./utils";
import z from "zod/v4";

export const testModelCall = async ({
  provider,
  model,
  apiKey,
  prompt,
  modelConfig,
}: {
  provider: string;
  model: string;
  apiKey: z.infer<typeof LLMApiKeySchema>;
  prompt?: string;
  modelConfig?: ModelConfig | null;
}) => {
  (
    await fetchLLMCompletion({
      streaming: false,
      apiKey: decrypt(apiKey.secretKey), // decrypt the secret key
      extraHeaders: decryptAndParseExtraHeaders(apiKey.extraHeaders),
      baseURL: apiKey.baseURL ?? undefined,
      messages: [
        {
          role: ChatMessageRole.User,
          content: prompt ?? "mock content",
          type: ChatMessageType.User,
        },
      ],
      modelParams: {
        provider: provider,
        model: model,
        adapter: apiKey.adapter,
        ...modelConfig,
      },
      structuredOutputSchema: zodV3.object({
        score: zodV3.string(),
        reasoning: zodV3.string(),
      }),
      config: apiKey.config,
    })
  ).completion;
};