openai-api-mock 0.1.10 → 0.1.12

package/main.test.js

@@ -89,6 +89,49 @@ describe('Mock OpenAI Chat & Image generation API', () => {
     });
 
 
+    it('should mock the chat completion with email function call (tools)', async () => {
+        try{
+            const response = await openai.chat.completions.create({
+                model: "gpt-3.5",
+                messages: [
+                    { role: 'system', content: "You'r an expert chef" },
+                    { role: 'user', content: "Suggest at least 5 recipes" },
+                ],
+                tools: [
+                    {
+                    type: "function",
+                    function: {
+                        name: 'send_email',
+                        description: 'Send an email to the user with the recipes',
+                        parameters: {
+                            type: "object",
+                            properties: {
+                                email: {
+                                    type: "string",
+                                    description: "The email address of the user",
+                                }
+                            }
+                        },
+                        required: ["email"],
+                    }
+
+                }],
+
+                tool_choice: { name: 'send_email' },
+            });
+            expect(response.model).toEqual('gpt-3.5-mock');
+            expect(response.choices[0]).toHaveProperty('finish_reason', 'tool_calls');
+            expect(response.choices[0].message.tool_calls[0].function).toHaveProperty('name', 'send_email');
+            expect(response.choices[0].message.tool_calls[0].function).toHaveProperty('arguments');
+            expect(JSON.parse(response.choices[0].message.tool_calls[0].function.arguments)).toHaveProperty('email');
+         
+        }catch(error){
+            throw new Error(error);
+        }
+
+    });
+
+
     it('should mock the chat completion with function call (tools)', async () => {
             const response = await openai.chat.completions.create({
                 model: "gpt-3.5",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openai-api-mock",
-  "version": "0.1.10",
+  "version": "0.1.12",
   "description": "A Node.js module for mocking OpenAI API responses in a development environment",
   "repository": {
     "type": "git",

package/src/chat.js

@@ -92,7 +92,8 @@ function generateToolCallArguments(requestBody) {
   Object.entries(parameters.properties).forEach(([paramName, paramDetails]) => {
     argumentsObject[paramName] = generateFakeData(
       paramDetails.type,
-      paramDetails
+      paramDetails,
+      paramName
     );
   });
 
@@ -102,19 +103,7 @@ function generateToolCallArguments(requestBody) {
 function generateFakeData(type, properties, name) {
   switch (type) {
     case "string":
-      if (name === "name") {
-        return faker.person.fullName();
-      } else if (name === "email") {
-        return faker.internet.email();
-      } else if (name === "price") {
-        return faker.commerce.price({ min: 100 })
-      } else if (name === "company") {
-        return faker.company.bs()
-      } else if (name === "phone") {
-        return faker.phone.number()
-      } else {
-        return faker.lorem.words(5);
-      }
+      return generateFakeStringData(name);
     case "number":
       return faker.number.int({ max: 100 });
     case "array":
@@ -135,7 +124,7 @@ function generateFakeArray(properties) {
       return faker.lorem.words(1);
     } else if (arrayItemsType === "object") {
       const itemProperties = properties.items.properties;
-      return generateFakeData("object", { properties: itemProperties });
+      return generateFakeData("object", { properties: itemProperties }, "item");
     }
   });
 }
@@ -147,7 +136,6 @@ function generateFakeObject(properties) {
   });
   return itemObject;
 }
-
 function generateFunctionCallArguments(requestBody) {
   const { parameters } = requestBody.functions[0];
   const argumentsObject = {};
@@ -155,13 +143,45 @@ function generateFunctionCallArguments(requestBody) {
   Object.entries(parameters.properties).forEach(([paramName, paramDetails]) => {
     argumentsObject[paramName] = generateFakeData(
       paramDetails.type,
-      paramDetails
+      paramDetails,
+      paramName
     );
   });
 
   return JSON.stringify(argumentsObject, null, 2);
 }
 
+function generateFakeStringData(name) {
+  if (name === "name") {
+    return faker.person.fullName();
+  } else if (name === "email") {
+    return faker.internet.email();
+  } else if (name === "price") {
+    return faker.commerce.price();
+  } else if (name === "company") {
+    return faker.company.companyName();
+  } else if (name === "phone") {
+    return faker.phone.phoneNumber();
+  } else if (name === "address") {
+    return faker.address.streetAddress();
+  } else if (name === "date") {
+    return faker.date.past();
+  } else if (name === "jobTitle") {
+    return faker.name.jobTitle();
+  } else if (name === "creditCardNumber") {
+    return faker.finance.creditCardNumber();
+  } else if (name === "currencyCode") {
+    return faker.finance.currencyCode();
+  } else if (name === "productName") {
+    return faker.commerce.productName();
+  } else if (name === "uuid") {
+    return faker.datatype.uuid();
+  } else {
+    return faker.lorem.words(5);
+  }
+}
+
+
 module.exports = {
   getChatResponce,
-};
+};