Merge pull request #226 from Ironclad/use-assemblyai-sdk

Use AssemblyAI Node SDK

packages/core/package.json

@@ -45,6 +45,7 @@
   "dependencies": {
     "@gentrace/core": "^2.2.5",
     "@huggingface/inference": "^2.6.4",
+    "assemblyai": "^3.0.1",
     "autoevals": "^0.0.26",
     "crypto-js": "^4.1.1",
     "emittery": "^1.0.1",

packages/core/src/plugins/assemblyAi/LemurActionItemsNode.ts

@@ -1,5 +1,6 @@
 import { nanoid } from 'nanoid/non-secure';
 import { dedent } from 'ts-dedent';
+import { AssemblyAI, type LemurActionItemsParameters } from 'assemblyai';
 import {
   type ChartNode,
   type EditorDefinition,
@@ -16,7 +17,6 @@ import {
 import { pluginNodeDefinition } from '../../model/NodeDefinition.js';
 import {
   type LemurNodeData,
-  type LemurParams,
   getApiKey,
   getLemurParams,
   lemurEditorDefinitions,
@@ -25,9 +25,7 @@ import {
 
 export type LemurActionItemsNode = ChartNode<'assemblyAiLemurActionItems', LemurActionItemsNodeData>;
 
-export type LemurActionItemsNodeData = LemurNodeData & {
-  answer_format?: string;
-};
+export type LemurActionItemsNodeData = LemurNodeData;
 
 export const LemurActionItemsNodeImpl: PluginNodeImpl<LemurActionItemsNode> = {
   create(): LemurActionItemsNode {
@@ -95,15 +93,11 @@ export const LemurActionItemsNodeImpl: PluginNodeImpl<LemurActionItemsNode> = {
 
   async process(data, inputs: Inputs, context: InternalProcessContext): Promise<Outputs> {
     const apiKey = getApiKey(context);
-    const params: LemurParams & {
-      answer_format?: string;
-    } = getLemurParams(inputs, data);
+    const client = new AssemblyAI({ apiKey });
 
-    if (data.answer_format) {
-      params.answer_format = data.answer_format;
-    }
+    const params: LemurActionItemsParameters = getLemurParams(inputs, data);
 
-    const { response } = await runLemurActionItems(apiKey, params);
+    const { response } = await client.lemur.actionItems(params);
 
     return {
       ['response' as PortId]: {
@@ -114,21 +108,4 @@ export const LemurActionItemsNodeImpl: PluginNodeImpl<LemurActionItemsNode> = {
   },
 };
 
-async function runLemurActionItems(apiToken: string, params: object) {
-  const response = await fetch('https://api.assemblyai.com/lemur/v3/generate/action-items', {
-    method: 'POST',
-    body: JSON.stringify(params),
-    headers: {
-      authorization: apiToken,
-    },
-  });
-  const body = await response.json();
-  if (response.status !== 200) {
-    if ('error' in body) throw new Error(body.error);
-    throw new Error(`LeMUR Action Items failed with status ${response.status}`);
-  }
-
-  return body as { response: string };
-}
-
 export const lemurActionItemsNode = pluginNodeDefinition(LemurActionItemsNodeImpl, 'LeMUR Action Items');

packages/core/src/plugins/assemblyAi/LemurQaNode.ts

@@ -20,14 +20,14 @@ import {
 } from '../../index.js';
 import {
   type LemurNodeData,
-  type LemurParams,
   getApiKey,
   getLemurParams,
   lemurEditorDefinitions,
   lemurTranscriptIdsInputDefinition,
 } from './lemurHelpers.js';
 import { coerceType } from '../../utils/coerceType.js';
 import { pluginNodeDefinition } from '../../model/NodeDefinition.js';
+import { AssemblyAI, type LemurQuestion, type LemurQuestionAnswerParameters } from 'assemblyai';
 
 export type LemurQaNode = ChartNode<'assemblyAiLemurQa', LemurQaNodeData>;
 
@@ -123,17 +123,16 @@ export const LemurQaNodeImpl = {
 
   async process(data, inputs: Inputs, context: InternalProcessContext): Promise<Outputs> {
     const apiKey = getApiKey(context);
+    const client = new AssemblyAI({ apiKey });
 
     const questions = getQuestions(inputs).map((question) => applyQuestionEditors(data, question));
 
-    const params: LemurParams & {
-      questions: Question[];
-    } = {
+    const params: LemurQuestionAnswerParameters = {
       questions,
       ...getLemurParams(inputs, data),
     };
 
-    const { response } = await runLemurQa(apiKey, params);
+    const { response } = await client.lemur.questionAnswer(params);
     return {
       ['response' as PortId]: {
         type: 'object[]',
@@ -143,7 +142,7 @@ export const LemurQaNodeImpl = {
   },
 } satisfies PluginNodeImpl<LemurQaNode>;
 
-function getQuestions(inputs: Inputs): Question[] {
+function getQuestions(inputs: Inputs): LemurQuestion[] {
   const input = inputs['questions' as PortId] as
     | StringDataValue
     | StringArrayDataValue
@@ -163,21 +162,21 @@ function getQuestions(inputs: Inputs): Question[] {
   } else if (input.type === 'string[]') {
     return coerceType(input, 'string[]').map((question) => ({ question }));
   } else if (input.type === 'object') {
-    return [coerceType(input, 'object')] as Question[];
+    return [coerceType(input, 'object')] as LemurQuestion[];
   } else if (input.type === 'object[]') {
-    return coerceType(input, 'object[]') as unknown as Question[];
+    return coerceType(input, 'object[]') as unknown as LemurQuestion[];
   } else if (input.type === 'any' && typeof input.value === 'string') {
     return [
       {
         question: coerceType(input, 'string'),
       },
     ];
   } else if ((input.type === 'any' && Array.isArray(input.value)) || input.type === 'any[]') {
-    return (input.value as any[]).map<Question>((question: any) => {
+    return (input.value as any[]).map<LemurQuestion>((question: any) => {
       if (typeof question === 'string') {
         return { question };
       } else if (typeof question === 'object') {
-        return question as Question;
+        return question as LemurQuestion;
       } else {
         throw new Error('Question must be a string or object.');
       }
@@ -186,7 +185,7 @@ function getQuestions(inputs: Inputs): Question[] {
   throw new Error('Questions must be a string, string[], a question object, or an array of question objects.');
 }
 
-function applyQuestionEditors(data: LemurQaNodeData, question: Question): Question {
+function applyQuestionEditors(data: LemurQaNodeData, question: LemurQuestion): LemurQuestion {
   if (!('answer_format' in question) && data.questions_answer_format) {
     question.answer_format = data.questions_answer_format;
   }
@@ -200,32 +199,4 @@ function applyQuestionEditors(data: LemurQaNodeData, question: Question): Questi
   return question;
 }
 
-async function runLemurQa(apiToken: string, params: object) {
-  const response = await fetch('https://api.assemblyai.com/lemur/v3/generate/question-answer', {
-    method: 'POST',
-    body: JSON.stringify(params),
-    headers: {
-      authorization: apiToken,
-    },
-  });
-  const body = await response.json();
-  if (response.status !== 200) {
-    if ('error' in body) throw new Error(body.error);
-    throw new Error(`LeMUR QA failed with status ${response.status}`);
-  }
-
-  return body as { response: QuestionAnswer[] };
-}
-
-type Question = {
-  question: string;
-  context?: string;
-  answer_format?: string;
-  answer_options?: string[];
-};
-type QuestionAnswer = {
-  question: string;
-  answer: string;
-};
-
 export const lemurQaNode = pluginNodeDefinition(LemurQaNodeImpl, 'LeMUR Q&A');

packages/core/src/plugins/assemblyAi/LemurSummaryNode.ts

@@ -1,5 +1,6 @@
 import { nanoid } from 'nanoid/non-secure';
 import { dedent } from 'ts-dedent';
+import { AssemblyAI, type LemurSummaryParameters } from 'assemblyai';
 import {
   type ChartNode,
   type EditorDefinition,
@@ -15,7 +16,6 @@ import {
 } from '../../index.js';
 import {
   type LemurNodeData,
-  type LemurParams,
   getApiKey,
   getLemurParams,
   lemurEditorDefinitions,
@@ -95,15 +95,14 @@ export const LemurSummaryNodeImpl: PluginNodeImpl<LemurSummaryNode> = {
 
   async process(data, inputs: Inputs, context: InternalProcessContext): Promise<Outputs> {
     const apiKey = getApiKey(context);
-    const params: LemurParams & {
-      answer_format?: string;
-    } = getLemurParams(inputs, data);
+    const client = new AssemblyAI({ apiKey });
+    const params: LemurSummaryParameters = getLemurParams(inputs, data);
 
     if (data.answer_format) {
       params.answer_format = data.answer_format;
     }
 
-    const { response } = await runLemurSummary(apiKey, params);
+    const { response } = await client.lemur.summary(params);
 
     return {
       ['response' as PortId]: {
@@ -114,21 +113,4 @@ export const LemurSummaryNodeImpl: PluginNodeImpl<LemurSummaryNode> = {
   },
 };
 
-async function runLemurSummary(apiToken: string, params: object) {
-  const response = await fetch('https://api.assemblyai.com/lemur/v3/generate/summary', {
-    method: 'POST',
-    body: JSON.stringify(params),
-    headers: {
-      authorization: apiToken,
-    },
-  });
-  const body = await response.json();
-  if (response.status !== 200) {
-    if ('error' in body) throw new Error(body.error);
-    throw new Error(`LeMUR Summary failed with status ${response.status}`);
-  }
-
-  return body as { response: string };
-}
-
 export const lemurSummaryNode = pluginNodeDefinition(LemurSummaryNodeImpl, 'LeMUR Summary');