openai-api-mock 0.1.33 → 0.2.0

package/dist/index.js

@@ -2186,16 +2186,129 @@ function getImageResponce(requestBody) {
     data
   };
 }
+const RESPONSE_TEMPLATES = {
+  SIMPLE_CHAT: {
+    choices: [
+      {
+        finish_reason: "stop",
+        index: 0,
+        message: {
+          content: "This is a consistent test response.",
+          role: "assistant"
+        },
+        logprobs: null
+      }
+    ],
+    created: 1640995200,
+    id: "chatcmpl-test123456789",
+    model: "gpt-3.5-mock",
+    object: "chat.completion",
+    usage: {
+      completion_tokens: 10,
+      prompt_tokens: 20,
+      total_tokens: 30
+    }
+  },
+  FUNCTION_CALL: {
+    id: "chatcmpl-test123456789",
+    object: "chat.completion",
+    created: 1640995200,
+    model: "gpt-3.5-mock",
+    choices: [
+      {
+        index: 0,
+        message: {
+          role: "assistant",
+          content: null,
+          function_call: {
+            name: "test_function",
+            arguments: '{"param1": "test_value", "param2": 42}'
+          }
+        },
+        finish_reason: "function_call"
+      }
+    ],
+    usage: {
+      prompt_tokens: 81,
+      completion_tokens: 19,
+      total_tokens: 100
+    }
+  },
+  TOOL_CALL: {
+    id: "chatcmpl-test123456789",
+    object: "chat.completion",
+    created: 1640995200,
+    model: "gpt-3.5-mock",
+    choices: [
+      {
+        index: 0,
+        message: {
+          role: "assistant",
+          content: null,
+          tool_calls: [
+            {
+              id: "call_test123456789",
+              type: "function",
+              function: {
+                name: "test_function",
+                arguments: '{"param1": "test_value", "param2": 42}'
+              }
+            }
+          ]
+        },
+        finish_reason: "tool_calls"
+      }
+    ],
+    usage: {
+      prompt_tokens: 81,
+      completion_tokens: 19,
+      total_tokens: 100
+    }
+  },
+  IMAGE_GENERATION: {
+    created: 1640995200,
+    data: [
+      {
+        url: "https://example.com/test-image.png"
+      }
+    ]
+  }
+};
+function createResponseTemplate(templateType, overrides = {}) {
+  const template = RESPONSE_TEMPLATES[templateType];
+  if (!template) {
+    throw new Error(`Unknown template type: ${templateType}`);
+  }
+  return deepMerge(template, overrides);
+}
+function deepMerge(target, source) {
+  const result = { ...target };
+  for (const key in source) {
+    if (source[key] && typeof source[key] === "object" && !Array.isArray(source[key])) {
+      result[key] = deepMerge(result[key] || {}, source[key]);
+    } else {
+      result[key] = source[key];
+    }
+  }
+  return result;
+}
 const OPEN_AI_BASE_URL = "https://api.openai.com";
 const CHAT_COMPLETIONS_ENDPOINT = "/v1/chat/completions";
 const IMAGE_GENERATIONS_ENDPOINT = "/v1/images/generations";
-const customScopes = [];
 function mockOpenAIResponse(force = false, options = {}) {
   const {
     includeErrors = false,
     latency = 0,
-    logRequests = false
+    logRequests = false,
+    seed = null,
+    useFixedResponses = false
   } = options;
+  if (seed !== null) {
+    f.seed(seed);
+    if (logRequests) {
+      console.log(`[openai-api-mock] Using seed for consistent outputs: ${seed}`);
+    }
+  }
   const env = process.env.NODE_ENV || "development";
   if (env !== "development" && !force) {
     return { isActive: false, stopMocking };
@@ -2216,12 +2329,21 @@ function mockOpenAIResponse(force = false, options = {}) {
         const stream = createChatStream(requestBody);
         return [200, stream];
       }
+      if (useFixedResponses) {
+        if (requestBody.tools) {
+          return [200, createResponseTemplate("TOOL_CALL")];
+        } else if (requestBody.functions) {
+          return [200, createResponseTemplate("FUNCTION_CALL")];
+        } else {
+          return [200, createResponseTemplate("SIMPLE_CHAT")];
+        }
+      }
       return [200, getChatResponce(requestBody)];
     } catch (error) {
       console.error("[openai-api-mock] Error processing chat request:", error);
       return [500, { error: { message: "Internal server error in mock" } }];
     }
-  });
+  }).persist();
   nock(OPEN_AI_BASE_URL).post(IMAGE_GENERATIONS_ENDPOINT).delay(latency).reply(function(uri, requestBody) {
     if (logRequests) {
       console.log(`[openai-api-mock] Image request:`, JSON.stringify(requestBody, null, 2));
@@ -2233,16 +2355,52 @@ function mockOpenAIResponse(force = false, options = {}) {
       if (includeErrors && Math.random() < 0.05) {
         return [400, { error: { message: "Your request was rejected as a result of our safety system." } }];
       }
+      if (useFixedResponses) {
+        return [200, createResponseTemplate("IMAGE_GENERATION")];
+      }
       return [200, getImageResponce(requestBody)];
     } catch (error) {
       console.error("[openai-api-mock] Error processing image request:", error);
       return [500, { error: { message: "Internal server error in mock" } }];
     }
-  });
+  }).persist();
   nock.enableNetConnect((host) => host !== "api.openai.com");
   return {
     isActive: true,
     stopMocking,
+    /**
+     * Resets the faker seed to ensure consistent outputs for subsequent requests
+     * @param {number|string} newSeed - New seed value to use
+     */
+    setSeed(newSeed) {
+      f.seed(newSeed);
+      if (logRequests) {
+        console.log(`[openai-api-mock] Seed updated to: ${newSeed}`);
+      }
+    },
+    /**
+     * Resets faker to use random values (removes deterministic behavior)
+     */
+    resetSeed() {
+      f.seed();
+      if (logRequests) {
+        console.log(`[openai-api-mock] Seed reset - using random values`);
+      }
+    },
+    /**
+     * Get available response templates
+     * @returns {Object} Available response templates
+     */
+    getResponseTemplates() {
+      return RESPONSE_TEMPLATES;
+    },
+    /**
+     * Create a custom response template
+     * @param {string} templateType - Type of template to use
+     * @param {Object} overrides - Values to override in the template
+     * @returns {Object} Response template with overrides applied
+     */
+    createResponseTemplate,
     /**
      * Adds a custom endpoint mock
      * @param {string} method - HTTP method (e.g., 'GET', 'POST')
@@ -2251,14 +2409,12 @@ function mockOpenAIResponse(force = false, options = {}) {
      */
     addCustomEndpoint(method, path, handler) {
       const methodLower = method.toLowerCase();
-      const scope = nock(OPEN_AI_BASE_URL)[methodLower](path).reply(handler).persist();
-      customScopes.push(scope);
+      nock(OPEN_AI_BASE_URL)[methodLower](path).reply(handler).persist();
     }
   };
 }
 function stopMocking() {
   nock.cleanAll();
-  customScopes.forEach((scope) => scope.persist(false));
 }
 export {
   mockOpenAIResponse,