Initial push

packages/components/nodes/agents/MRLKAgentLLM/MRLKAgentLLM.ts

@@ -0,0 +1,58 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+
+class MRLKAgentLLM implements INode {
+    label: string
+    name: string
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'MRLK Agent for LLMs'
+        this.name = 'mrlkAgentLLM'
+        this.type = 'AgentExecutor'
+        this.category = 'Agents'
+        this.icon = 'agent.svg'
+        this.description = 'Agent that uses the ReAct Framework to decide what action to take, optimized to be used with LLMs'
+        this.inputs = [
+            {
+                label: 'Allowed Tools',
+                name: 'tools',
+                type: 'Tool',
+                list: true
+            },
+            {
+                label: 'LLM Model',
+                name: 'model',
+                type: 'BaseLanguageModel'
+            }
+        ]
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        return ['AgentExecutor']
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const { initializeAgentExecutor } = await import('langchain/agents')
+
+        const model = nodeData.inputs?.model
+        const tools = nodeData.inputs?.tools
+
+        const executor = await initializeAgentExecutor(tools, model, 'zero-shot-react-description', true)
+
+        return executor
+    }
+
+    async run(nodeData: INodeData, input: string): Promise<string> {
+        const executor = nodeData.instance
+        const result = await executor.call({ input })
+
+        return result?.output
+    }
+}
+
+module.exports = { nodeClass: MRLKAgentLLM }

packages/components/nodes/agents/MRLKAgentLLM/agent.svg

@@ -0,0 +1,9 @@
+<svg xmlns="http://www.w3.org/2000/svg" class="icon icon-tabler icon-tabler-robot" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round">
+   <path stroke="none" d="M0 0h24v24H0z" fill="none"></path>
+   <path d="M7 7h10a2 2 0 0 1 2 2v1l1 1v3l-1 1v3a2 2 0 0 1 -2 2h-10a2 2 0 0 1 -2 -2v-3l-1 -1v-3l1 -1v-1a2 2 0 0 1 2 -2z"></path>
+   <path d="M10 16h4"></path>
+   <circle cx="8.5" cy="11.5" r=".5" fill="currentColor"></circle>
+   <circle cx="15.5" cy="11.5" r=".5" fill="currentColor"></circle>
+   <path d="M9 7l-1 -4"></path>
+   <path d="M15 7l1 -4"></path>
+</svg>

packages/components/nodes/chains/LLMChain/LLMChain.ts

@@ -0,0 +1,61 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses } from '../../../src/utils'
+
+class LLMChain_Chains implements INode {
+    label: string
+    name: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    description: string
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'LLM Chain'
+        this.name = 'llmChain'
+        this.type = 'LLMChain'
+        this.icon = 'chain.svg'
+        this.category = 'Chains'
+        this.description = 'Chain to run queries against LLMs'
+        this.inputs = [
+            {
+                label: 'LLM',
+                name: 'llm',
+                type: 'BaseLanguageModel'
+            },
+            {
+                label: 'Prompt',
+                name: 'prompt',
+                type: 'BasePromptTemplate'
+            }
+        ]
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        const { LLMChain } = await import('langchain/chains')
+        return getBaseClasses(LLMChain)
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const { LLMChain } = await import('langchain/chains')
+
+        const llm = nodeData.inputs?.llm
+        const prompt = nodeData.inputs?.prompt
+
+        const chain = new LLMChain({ llm, prompt })
+        return chain
+    }
+
+    async run(nodeData: INodeData, input: string): Promise<string> {
+        const prompt = nodeData.instance.prompt.inputVariables // ["product"]
+        if (prompt.length > 1) throw new Error('Prompt can only contains 1 literal string {}. Multiples are found')
+
+        const chain = nodeData.instance
+        const res = await chain.run(input)
+
+        return res
+    }
+}
+
+module.exports = { nodeClass: LLMChain_Chains }

packages/components/nodes/chains/LLMChain/chain.svg

@@ -0,0 +1,6 @@
+<svg xmlns="http://www.w3.org/2000/svg" class="icon icon-tabler icon-tabler-dna" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round">
+   <path stroke="none" d="M0 0h24v24H0z" fill="none"></path>
+   <path d="M14.828 14.828a4 4 0 1 0 -5.656 -5.656a4 4 0 0 0 5.656 5.656z"></path>
+   <path d="M9.172 20.485a4 4 0 1 0 -5.657 -5.657"></path>
+   <path d="M14.828 3.515a4 4 0 0 0 5.657 5.657"></path>
+</svg>

packages/components/nodes/llms/OpenAI/OpenAI.ts

@@ -0,0 +1,83 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses } from '../../../src/utils'
+
+class OpenAI_LLMs implements INode {
+    label: string
+    name: string
+    type: string
+    icon: string
+    category: string
+    description: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'OpenAI'
+        this.name = 'openAI'
+        this.type = 'OpenAI'
+        this.icon = 'openai.png'
+        this.category = 'LLMs'
+        this.description = 'Wrapper around OpenAI large language models'
+        this.inputs = [
+            {
+                label: 'OpenAI Api Key',
+                name: 'openAIApiKey',
+                type: 'password'
+            },
+            {
+                label: 'Model Name',
+                name: 'modelName',
+                type: 'options',
+                options: [
+                    {
+                        label: 'text-davinci-003',
+                        name: 'text-davinci-003'
+                    },
+                    {
+                        label: 'text-davinci-002',
+                        name: 'text-davinci-002'
+                    },
+                    {
+                        label: 'text-curie-001',
+                        name: 'text-curie-001'
+                    },
+                    {
+                        label: 'text-babbage-001',
+                        name: 'text-babbage-001'
+                    }
+                ],
+                default: 'text-davinci-003',
+                optional: true
+            },
+            {
+                label: 'Temperature',
+                name: 'temperature',
+                type: 'number',
+                default: 0.7,
+                optional: true
+            }
+        ]
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        const { OpenAI } = await import('langchain/llms')
+        return getBaseClasses(OpenAI)
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const { OpenAI } = await import('langchain/llms')
+
+        const temperature = nodeData.inputs?.temperature as string
+        const modelName = nodeData.inputs?.modelName as string
+        const openAIApiKey = nodeData.inputs?.openAIApiKey as string
+
+        const model = new OpenAI({
+            temperature: parseInt(temperature, 10),
+            modelName,
+            openAIApiKey
+        })
+        return model
+    }
+}
+
+module.exports = { nodeClass: OpenAI_LLMs }

packages/components/nodes/prompts/PromptTemplate/PromptTemplate.ts

@@ -0,0 +1,56 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses, getInputVariables } from '../../../src/utils'
+
+class PromptTemplate_Prompts implements INode {
+    label: string
+    name: string
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'Prompt Template'
+        this.name = 'promptTemplate'
+        this.type = 'PromptTemplate'
+        this.icon = 'prompt.svg'
+        this.category = 'Prompts'
+        this.description = 'Schema to represent a basic prompt for an LLM. Template can only contains 1 literal string {}'
+        this.inputs = [
+            {
+                label: 'Template',
+                name: 'template',
+                type: 'string',
+                rows: 5,
+                default: 'What is a good name for a company that makes {product}?',
+                placeholder: 'What is a good name for a company that makes {product}?'
+            }
+        ]
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        const { PromptTemplate } = await import('langchain/prompts')
+        return getBaseClasses(PromptTemplate)
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const { PromptTemplate } = await import('langchain/prompts')
+
+        const template = nodeData.inputs?.template as string
+        const inputVariables = getInputVariables(template)
+
+        try {
+            const prompt = new PromptTemplate({
+                template,
+                inputVariables: inputVariables
+            })
+            return prompt
+        } catch (e) {
+            throw new Error(e)
+        }
+    }
+}
+
+module.exports = { nodeClass: PromptTemplate_Prompts }

packages/components/nodes/prompts/PromptTemplate/prompt.svg

@@ -0,0 +1,6 @@
+<svg xmlns="http://www.w3.org/2000/svg" class="icon icon-tabler icon-tabler-terminal-2" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round">
+   <path stroke="none" d="M0 0h24v24H0z" fill="none"></path>
+   <path d="M8 9l3 3l-3 3"></path>
+   <path d="M13 15l3 0"></path>
+   <path d="M3 4m0 2a2 2 0 0 1 2 -2h14a2 2 0 0 1 2 2v12a2 2 0 0 1 -2 2h-14a2 2 0 0 1 -2 -2z"></path>
+</svg>

packages/components/nodes/tools/Calculator/Calculator.ts

@@ -0,0 +1,33 @@
+import { INode } from '../../../src/Interface'
+import { getBaseClasses } from '../../../src/utils'
+
+class Calculator implements INode {
+    label: string
+    name: string
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+
+    constructor() {
+        this.label = 'Calculator'
+        this.name = 'calculator'
+        this.type = 'Calculator'
+        this.icon = 'calculator.svg'
+        this.category = 'Tools'
+        this.description = 'Perform calculations on response'
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        const { Calculator } = await import('langchain/tools')
+        return getBaseClasses(Calculator)
+    }
+
+    async init(): Promise<any> {
+        const { Calculator } = await import('langchain/tools')
+        return new Calculator()
+    }
+}
+
+module.exports = { nodeClass: Calculator }

packages/components/nodes/tools/Calculator/calculator.svg

@@ -0,0 +1,11 @@
+<svg xmlns="http://www.w3.org/2000/svg" class="icon icon-tabler icon-tabler-calculator" width="24" height="24" viewBox="0 0 24 24" stroke-width="2" stroke="currentColor" fill="none" stroke-linecap="round" stroke-linejoin="round">
+   <path stroke="none" d="M0 0h24v24H0z" fill="none"></path>
+   <path d="M4 3m0 2a2 2 0 0 1 2 -2h12a2 2 0 0 1 2 2v14a2 2 0 0 1 -2 2h-12a2 2 0 0 1 -2 -2z"></path>
+   <path d="M8 7m0 1a1 1 0 0 1 1 -1h6a1 1 0 0 1 1 1v1a1 1 0 0 1 -1 1h-6a1 1 0 0 1 -1 -1z"></path>
+   <path d="M8 14l0 .01"></path>
+   <path d="M12 14l0 .01"></path>
+   <path d="M16 14l0 .01"></path>
+   <path d="M8 17l0 .01"></path>
+   <path d="M12 17l0 .01"></path>
+   <path d="M16 17l0 .01"></path>
+</svg>

packages/components/nodes/tools/SerpAPI/SerpAPI.ts

@@ -0,0 +1,42 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses } from '../../../src/utils'
+
+class SerpAPI implements INode {
+    label: string
+    name: string
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'Serp API'
+        this.name = 'serpAPI'
+        this.type = 'SerpAPI'
+        this.icon = 'serp.png'
+        this.category = 'Tools'
+        this.description = 'Wrapper around SerpAPI - a real-time API to access Google search results'
+        this.inputs = [
+            {
+                label: 'Serp Api Key',
+                name: 'apiKey',
+                type: 'password'
+            }
+        ]
+    }
+
+    async getBaseClasses(): Promise<string[]> {
+        const { SerpAPI } = await import('langchain/tools')
+        return getBaseClasses(SerpAPI)
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const { SerpAPI } = await import('langchain/tools')
+        const apiKey = nodeData.inputs?.apiKey as string
+        return new SerpAPI(apiKey)
+    }
+}
+
+module.exports = { nodeClass: SerpAPI }