Merge pull request #162 from bettyblocks/feat/extra-params-generative-ai

feat: added new parameters, these can be used to connect to azure in …

__tests__/ai-classifier/1.0/index.test.js

@@ -14,6 +14,7 @@ const input = {
   maxTokens: 100,
   temperature: 50,
   model: 'text-davinci-002',
+  parameters: [],
 };
 
 const createPaths = (paths) => {
@@ -79,6 +80,7 @@ describe('Classifier function', () => {
       variables: {
         question: 'What tool is the best to ask for the current weather?',
       },
+      parameters: {},
     },
   };
 

functions/ai-anonymizer/1.0/function.json

@@ -76,6 +76,15 @@
         "default": 1024
       }
     },
+    {
+      "name": "parameters",
+      "label": "Extra parameters",
+      "info": "Define extra parameters that will be passed to the LLM, for example if you want to use azure openai",
+      "advanced": true,
+      "meta": {
+        "type": "Map"
+      }
+    },
     {
       "name": "result",
       "label": "Result",

functions/ai-anonymizer/1.0/index.js

@@ -1,6 +1,21 @@
-const anonymizer = async ({ textToAnonymize, apiKey, maxTokens, model }) => {
+const anonymizer = async ({
+  textToAnonymize,
+  apiKey,
+  maxTokens,
+  model,
+  parameters = [],
+}) => {
   const prompt = `Please review the following document and redact any Personal Identifiable Information (PII) such as names, addresses, phone numbers, email addresses, social security numbers, and any other details that could be used to identify an individual. Ensure the document remains coherent and retains its essential information while ensuring privacy and confidentiality. ${textToAnonymize}`;
 
+  // Convert parameters list to a single object
+  const parameterMap = parameters.reduce(
+    (previousValue, currentValue) => ({
+      ...previousValue,
+      [currentValue.key]: currentValue.value,
+    }),
+    {},
+  );
+
   const { result } = await generativeAI({
     authorization: {
       apiKey,
@@ -12,6 +27,7 @@ const anonymizer = async ({ textToAnonymize, apiKey, maxTokens, model }) => {
         model,
         settings: { maxNewTokens: maxTokens },
       },
+      parameters: parameterMap,
     },
   });
 

functions/ai-classifier/1.0/function.json

@@ -76,6 +76,15 @@
         "type": "Map"
       }
     },
+    {
+      "name": "parameters",
+      "label": "Extra parameters",
+      "info": "Define extra parameters that will be passed to the LLM, for example if you want to use azure openai",
+      "advanced": true,
+      "meta": {
+        "type": "Map"
+      }
+    },
     {
       "info": "The result produced by the classifier",
       "name": "result",

functions/ai-classifier/1.0/index.js

@@ -1,4 +1,7 @@
-const classifier = async ({ prompt, apiKey, model, variables }, paths) => {
+const classifier = async (
+  { prompt, apiKey, model, variables, parameters = [] },
+  paths,
+) => {
   const variableMap = variables.reduce(
     (previousValue, currentValue) => ({
       ...previousValue,
@@ -7,6 +10,15 @@ const classifier = async ({ prompt, apiKey, model, variables }, paths) => {
     {},
   );
 
+  // Convert parameters list to a single object
+  const parameterMap = parameters.reduce(
+    (previousValue, currentValue) => ({
+      ...previousValue,
+      [currentValue.key]: currentValue.value,
+    }),
+    {},
+  );
+
   const choices = paths.map(({ label, value }) => ({
     choice: label,
     description:
@@ -27,6 +39,7 @@ const classifier = async ({ prompt, apiKey, model, variables }, paths) => {
         model,
       },
       variables: variableMap,
+      parameters: parameterMap,
     },
   });
 

functions/ai-prompt/1.0/function.json

@@ -98,6 +98,15 @@
         }
       }
     },
+    {
+      "name": "parameters",
+      "label": "Extra parameters",
+      "info": "Define extra parameters that will be passed to the LLM, for example if you want to use azure openai",
+      "advanced": true,
+      "meta": {
+        "type": "Map"
+      }
+    },
     {
       "name": "result",
       "label": "Result",

functions/ai-prompt/1.0/index.js

@@ -5,6 +5,7 @@ const agent = async ({
   maxTokens,
   temperature,
   model,
+  parameters = [],
 }) => {
   const variableMap = variables.reduce(
     (previousValue, currentValue) => ({
@@ -14,6 +15,15 @@ const agent = async ({
     {},
   );
 
+  // Convert parameters list to a single object
+  const parameterMap = parameters.reduce(
+    (previousValue, currentValue) => ({
+      ...previousValue,
+      [currentValue.key]: currentValue.value,
+    }),
+    {},
+  );
+
   const { result } = await generativeAI({
     authorization: {
       apiKey,
@@ -26,6 +36,7 @@ const agent = async ({
         model,
         settings: { maxNewTokens: maxTokens, temperature: temperature / 100 },
       },
+      parameters: parameterMap,
     },
   });
 

functions/ai-query-generator/1.0/function.json

@@ -95,6 +95,15 @@
         "default": 256
       }
     },
+    {
+      "name": "parameters",
+      "label": "Extra parameters",
+      "info": "Define extra parameters that will be passed to the LLM, for example if you want to use azure openai",
+      "advanced": true,
+      "meta": {
+        "type": "Map"
+      }
+    },
     {
       "name": "result",
       "label": "Result",

functions/ai-query-generator/1.0/index.js

@@ -8,7 +8,14 @@ const regexQuery = (text) =>
 const booleanQuery = (text) =>
   `Generate a BM25 compatible boolean query for the following text: ${text}`;
 
-const queryGenerator = async ({ query, style, apiKey, maxTokens, model }) => {
+const queryGenerator = async ({
+  query,
+  style,
+  apiKey,
+  maxTokens,
+  model,
+  parameters = [],
+}) => {
   let prompt;
 
   switch (style) {
@@ -19,6 +26,15 @@ const queryGenerator = async ({ query, style, apiKey, maxTokens, model }) => {
       prompt = regexQuery(query);
   }
 
+  // Convert parameters list to a single object
+  const parameterMap = parameters.reduce(
+    (previousValue, currentValue) => ({
+      ...previousValue,
+      [currentValue.key]: currentValue.value,
+    }),
+    {},
+  );
+
   const { result } = await generativeAI({
     authorization: {
       apiKey,
@@ -30,6 +46,7 @@ const queryGenerator = async ({ query, style, apiKey, maxTokens, model }) => {
         model,
         settings: { maxNewTokens: maxTokens },
       },
+      parameters: parameterMap,
     },
   });
 

functions/ai-summarizer/1.0/function.json

@@ -98,6 +98,15 @@
         "default": 1024
       }
     },
+    {
+      "name": "parameters",
+      "label": "Extra parameters",
+      "info": "Define extra parameters that will be passed to the LLM, for example if you want to use azure openai",
+      "advanced": true,
+      "meta": {
+        "type": "Map"
+      }
+    },
     {
       "name": "result",
       "label": "Result",

functions/ai-summarizer/1.0/index.js

@@ -10,6 +10,7 @@ const summarizer = async ({
   apiKey,
   maxTokens,
   model,
+  parameters = [],
 }) => {
   let prompt;
 
@@ -21,6 +22,15 @@ const summarizer = async ({
       prompt = conciseSummary(textToSummarize);
   }
 
+  // Convert parameters list to a single object
+  const parameterMap = parameters.reduce(
+    (previousValue, currentValue) => ({
+      ...previousValue,
+      [currentValue.key]: currentValue.value,
+    }),
+    {},
+  );
+
   const { result } = await generativeAI({
     authorization: {
       apiKey,
@@ -32,6 +42,7 @@ const summarizer = async ({
         model,
         settings: { maxNewTokens: maxTokens },
       },
+      parameters: parameterMap,
     },
   });