illuma-agents 1.0.8 → 1.0.9

package/src/proto/collab.ts

@@ -1,200 +0,0 @@
-// src/collab.ts
-import 'dotenv/config';
-import { BaseMessage, HumanMessage } from '@langchain/core/messages';
-import { END, StateGraphArgs, START, StateGraph } from '@langchain/langgraph';
-import { chartTool, tavilyTool } from '@/tools/example';
-import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
-import {
-    ChatPromptTemplate,
-    MessagesPlaceholder,
-} from '@langchain/core/prompts';
-import { ChatOpenAI } from '@langchain/openai';
-import { Runnable, RunnableConfig } from '@langchain/core/runnables';
-import { JsonOutputToolsParser } from 'langchain/output_parsers';
-import { HandlerRegistry } from '@/events';
-
-interface AgentStateChannels {
-  messages: BaseMessage[];
-  next: string;
-}
-
-export class CollaborativeProcessor {
-    private graph: Runnable | null = null;
-    private handlerRegistry: HandlerRegistry;
-
-    constructor(customHandlers?: Record<string, any>) {
-        this.handlerRegistry = new HandlerRegistry();
-        if (customHandlers) {
-            for (const [eventType, handler] of Object.entries(customHandlers)) {
-                this.handlerRegistry.register(eventType, handler);
-            }
-        }
-    }
-
-    async initialize(): Promise<void> {
-        this.graph = await this.createGraph();
-    }
-
-    private async createGraph(): Promise<Runnable> {
-        const agentStateChannels: StateGraphArgs['channels'] = {
-            messages: {
-                value: (x?: BaseMessage[], y?: BaseMessage[]) => (x ?? []).concat(y ?? []),
-                default: () => [],
-            },
-            next: {
-                value: (x?: string, y?: string) => y ?? x ?? END,
-                default: () => END,
-            },
-        };
-
-        async function createAgent(
-            llm: ChatOpenAI,
-            tools: any[],
-            systemPrompt: string
-        ): Promise<AgentExecutor> {
-            const prompt = await ChatPromptTemplate.fromMessages([
-                ['system', systemPrompt],
-                new MessagesPlaceholder('messages'),
-                new MessagesPlaceholder('agent_scratchpad'),
-            ]);
-            const agent = await createOpenAIToolsAgent({ llm, tools, prompt });
-            return new AgentExecutor({ agent, tools });
-        }
-
-        const members = ['researcher', 'chart_generator'];
-
-        const systemPrompt =
-      'You are a supervisor tasked with managing a conversation between the' +
-      ' following workers: {members}. Given the following user request,' +
-      ' respond with the worker to act next. Each worker will perform a' +
-      ' task and respond with their results and status. When finished,' +
-      ' respond with FINISH.';
-        const options = [END, ...members];
-
-        const functionDef = {
-            name: 'route',
-            description: 'Select the next role.',
-            parameters: {
-                title: 'routeSchema',
-                type: 'object',
-                properties: {
-                    next: {
-                        title: 'Next',
-                        anyOf: [
-                            { enum: options },
-                        ],
-                    },
-                },
-                required: ['next'],
-            },
-        };
-
-        const toolDef = {
-            type: 'function',
-            function: functionDef,
-        } as const;
-
-        const prompt = ChatPromptTemplate.fromMessages([
-            ['system', systemPrompt],
-            new MessagesPlaceholder('messages'),
-            [
-                'system',
-                'Given the conversation above, who should act next?' +
-        ' Or should we FINISH? Select one of: {options}',
-            ],
-        ]);
-
-        const formattedPrompt = await prompt.partial({
-            options: options.join(', '),
-            members: members.join(', '),
-        });
-
-        const llm = new ChatOpenAI({
-            modelName: 'gpt-4',
-            temperature: 0,
-        });
-
-        const supervisorChain = formattedPrompt
-            .pipe(llm.bindTools(
-                [toolDef],
-                {
-                    tool_choice: { 'type': 'function', 'function': { 'name': 'route' } },
-                },
-            ))
-            .pipe(new JsonOutputToolsParser())
-            .pipe((x) => (x[0].args));
-
-        const researcherAgent = await createAgent(
-            llm,
-            [tavilyTool],
-            'You are a web researcher. You may use the Tavily search engine to search the web for' +
-        ' important information, so the Chart Generator in your team can make useful plots.',
-        );
-
-        const researcherNode = async (
-            state: AgentStateChannels,
-            config?: RunnableConfig,
-        ) => {
-            const result = await researcherAgent.invoke(state, config);
-            return {
-                messages: [
-                    new HumanMessage({ content: result.output, name: 'Researcher' }),
-                ],
-            };
-        };
-
-        const chartGenAgent = await createAgent(
-            llm,
-            [chartTool],
-            'You excel at generating bar charts. Use the researcher\'s information to generate the charts.',
-        );
-
-        const chartGenNode = async (
-            state: AgentStateChannels,
-            config?: RunnableConfig,
-        ) => {
-            const result = await chartGenAgent.invoke(state, config);
-            return {
-                messages: [
-                    new HumanMessage({ content: result.output, name: 'ChartGenerator' }),
-                ],
-            };
-        };
-
-        const workflow = new StateGraph({
-            channels: agentStateChannels,
-        })
-            .addNode('researcher', researcherNode)
-            .addNode('chart_generator', chartGenNode)
-            .addNode('supervisor', supervisorChain);
-
-        members.forEach((member) => {
-            workflow.addEdge(member, 'supervisor');
-        });
-
-        workflow.addConditionalEdges(
-            'supervisor',
-            (x: AgentStateChannels) => x.next,
-        );
-
-        workflow.addEdge(START, 'supervisor');
-
-        return workflow.compile();
-    }
-
-    async processStream(
-        inputs: { messages: BaseMessage[] },
-        config: Partial<RunnableConfig> & { version: 'v1' | 'v2' },
-    ) {
-        if (!this.graph) {
-            throw new Error('CollaborativeProcessor not initialized. Call initialize() first.');
-        }
-        const stream = await this.graph.streamEvents(inputs, config);
-        for await (const event of stream) {
-            const handler = this.handlerRegistry.getHandler(event.event);
-            if (handler) {
-                handler.handle(event.event, event.data);
-            }
-        }
-    }
-}

package/src/proto/collab_design.ts

@@ -1,184 +0,0 @@
-// src/collab.ts
-import 'dotenv/config';
-import { BaseMessage, HumanMessage } from '@langchain/core/messages';
-import { END, StateGraphArgs, START, StateGraph } from '@langchain/langgraph';
-import { chartTool, tavilyTool } from '@/tools/example';
-import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
-import {
-    ChatPromptTemplate,
-    MessagesPlaceholder,
-} from '@langchain/core/prompts';
-import { ChatOpenAI } from '@langchain/openai';
-import { Runnable, RunnableConfig } from '@langchain/core/runnables';
-import { JsonOutputToolsParser } from 'langchain/output_parsers';
-import { HandlerRegistry } from '@/events';
-
-interface AgentStateChannels {
-  messages: BaseMessage[];
-  next: string;
-}
-
-interface Member {
-  name: string;
-  systemPrompt: string;
-  tools: any[];
-}
-
-export class CollaborativeProcessor {
-    private graph: Runnable | null = null;
-    private handlerRegistry: HandlerRegistry;
-    private members: Member[];
-
-    constructor(members: Member[], customHandlers?: Record<string, any>) {
-        this.members = members;
-        this.handlerRegistry = new HandlerRegistry();
-        if (customHandlers) {
-            for (const [eventType, handler] of Object.entries(customHandlers)) {
-                this.handlerRegistry.register(eventType, handler);
-            }
-        }
-    }
-
-    async initialize(): Promise<void> {
-        this.graph = await this.createGraph();
-    }
-
-    private async createGraph(): Promise<Runnable> {
-        const agentStateChannels: StateGraphArgs['channels'] = {
-            messages: {
-                value: (x?: BaseMessage[], y?: BaseMessage[]) => (x ?? []).concat(y ?? []),
-                default: () => [],
-            },
-            next: {
-                value: (x?: string, y?: string) => y ?? x ?? END,
-                default: () => END,
-            },
-        };
-
-        async function createAgent(
-            llm: ChatOpenAI,
-            tools: any[],
-            systemPrompt: string
-        ): Promise<AgentExecutor> {
-            const prompt = await ChatPromptTemplate.fromMessages([
-                ['system', systemPrompt],
-                new MessagesPlaceholder('messages'),
-                new MessagesPlaceholder('agent_scratchpad'),
-            ]);
-            const agent = await createOpenAIToolsAgent({ llm, tools, prompt });
-            return new AgentExecutor({ agent, tools });
-        }
-
-        const memberNames = this.members.map(member => member.name);
-
-        const systemPrompt =
-      'You are a supervisor tasked with managing a conversation between the' +
-      ' following workers: {members}. Given the following user request,' +
-      ' respond with the worker to act next. Each worker will perform a' +
-      ' task and respond with their results and status. When finished,' +
-      ' respond with FINISH.';
-        const options = [END, ...memberNames];
-
-        const functionDef = {
-            name: 'route',
-            description: 'Select the next role.',
-            parameters: {
-                title: 'routeSchema',
-                type: 'object',
-                properties: {
-                    next: {
-                        title: 'Next',
-                        anyOf: [
-                            { enum: options },
-                        ],
-                    },
-                },
-                required: ['next'],
-            },
-        };
-
-        const toolDef = {
-            type: 'function',
-            function: functionDef,
-        } as const;
-
-        const prompt = ChatPromptTemplate.fromMessages([
-            ['system', systemPrompt],
-            new MessagesPlaceholder('messages'),
-            [
-                'system',
-                'Given the conversation above, who should act next?' +
-        ' Or should we FINISH? Select one of: {options}',
-            ],
-        ]);
-
-        const formattedPrompt = await prompt.partial({
-            options: options.join(', '),
-            members: memberNames.join(', '),
-        });
-
-        const llm = new ChatOpenAI({
-            modelName: 'gpt-4',
-            temperature: 0,
-        });
-
-        const supervisorChain = formattedPrompt
-            .pipe(llm.bindTools(
-                [toolDef],
-                {
-                    tool_choice: { 'type': 'function', 'function': { 'name': 'route' } },
-                },
-            ))
-            .pipe(new JsonOutputToolsParser())
-            .pipe((x) => (x[0].args));
-
-        const workflow = new StateGraph({
-            channels: agentStateChannels,
-        });
-
-        // Dynamically create agents and add nodes for each member
-        for (const member of this.members) {
-            const agent = await createAgent(llm, member.tools, member.systemPrompt);
-            const node = async (
-                state: AgentStateChannels,
-                config?: RunnableConfig,
-            ) => {
-                const result = await agent.invoke(state, config);
-                return {
-                    messages: [
-                        new HumanMessage({ content: result.output, name: member.name }),
-                    ],
-                };
-            };
-            workflow.addNode(member.name, node);
-            workflow.addEdge(member.name, 'supervisor');
-        }
-
-        workflow.addNode('supervisor', supervisorChain);
-
-        workflow.addConditionalEdges(
-            'supervisor',
-            (x: AgentStateChannels) => x.next,
-        );
-
-        workflow.addEdge(START, 'supervisor');
-
-        return workflow.compile();
-    }
-
-    async processStream(
-        inputs: { messages: BaseMessage[] },
-        config: Partial<RunnableConfig> & { version: 'v1' | 'v2' },
-    ) {
-        if (!this.graph) {
-            throw new Error('CollaborativeProcessor not initialized. Call initialize() first.');
-        }
-        const stream = this.graph.streamEvents(inputs, config);
-        for await (const event of stream) {
-            const handler = this.handlerRegistry.getHandler(event.event);
-            if (handler) {
-                handler.handle(event.event, event.data);
-            }
-        }
-    }
-}

package/src/proto/collab_design_v2.ts

@@ -1,224 +0,0 @@
-// src/collab.ts
-import 'dotenv/config';
-import { BaseMessage, HumanMessage } from '@langchain/core/messages';
-import { END, StateGraphArgs, START, StateGraph } from '@langchain/langgraph';
-import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
-import {
-    ChatPromptTemplate,
-    MessagesPlaceholder,
-} from '@langchain/core/prompts';
-import { Runnable, RunnableConfig } from '@langchain/core/runnables';
-import { JsonOutputToolsParser } from 'langchain/output_parsers';
-import { HandlerRegistry } from '@/events';
-import { ChatOpenAI } from '@langchain/openai';
-import { ChatBedrockConverse } from '@langchain/aws';
-import { ChatAnthropic } from '@langchain/anthropic';
-import { ChatMistralAI } from '@langchain/mistralai';
-import { ChatVertexAI } from '@langchain/google-vertexai';
-import { BedrockChat } from '@langchain/community/chat_models/bedrock/web';
-import { Providers } from '@/common';
-
-interface AgentStateChannels {
-  messages: BaseMessage[];
-  next: string;
-}
-
-export interface Member {
-  name: string;
-  systemPrompt: string;
-  tools: any[];
-  llmConfig: LLMConfig;
-}
-
-interface LLMConfig {
-  provider: Providers;
-  [key: string]: any;
-}
-
-interface SupervisorConfig {
-  systemPrompt?: string;
-  llmConfig: LLMConfig;
-}
-
-const llmProviders: Record<Providers, any> = {
-    [Providers.OPENAI]: ChatOpenAI,
-    [Providers.VERTEXAI]: ChatVertexAI,
-    [Providers.BEDROCK_LEGACY]: BedrockChat,
-    [Providers.MISTRALAI]: ChatMistralAI,
-    [Providers.BEDROCK]: ChatBedrockConverse,
-    [Providers.ANTHROPIC]: ChatAnthropic,
-};
-
-export class CollaborativeProcessor {
-    private graph: Runnable | null = null;
-    private handlerRegistry: HandlerRegistry;
-    private members: Member[];
-    private supervisorConfig: SupervisorConfig;
-
-    constructor(
-        members: Member[],
-        supervisorConfig: SupervisorConfig,
-        customHandlers?: Record<string, any>
-    ) {
-        this.members = members;
-        this.supervisorConfig = supervisorConfig;
-        this.handlerRegistry = new HandlerRegistry();
-        if (customHandlers) {
-            for (const [eventType, handler] of Object.entries(customHandlers)) {
-                this.handlerRegistry.register(eventType, handler);
-            }
-        }
-    }
-
-    async initialize(): Promise<void> {
-        this.graph = await this.createGraph();
-    }
-
-    private async createGraph(): Promise<Runnable> {
-        const agentStateChannels: StateGraphArgs['channels'] = {
-            messages: {
-                value: (x?: BaseMessage[], y?: BaseMessage[]) => (x ?? []).concat(y ?? []),
-                default: () => [],
-            },
-            next: {
-                value: (x?: string, y?: string) => y ?? x ?? END,
-                default: () => END,
-            },
-        };
-
-        async function createAgent(
-            llmConfig: LLMConfig,
-            tools: any[],
-            systemPrompt: string
-        ): Promise<AgentExecutor> {
-            const { provider, ...clientOptions } = llmConfig;
-            const LLMClass = llmProviders[provider];
-            if (!LLMClass) {
-                throw new Error(`Unsupported LLM provider: ${provider}`);
-            }
-            const llm = new LLMClass(clientOptions);
-
-            const prompt = await ChatPromptTemplate.fromMessages([
-                ['system', systemPrompt],
-                new MessagesPlaceholder('messages'),
-                new MessagesPlaceholder('agent_scratchpad'),
-            ]);
-            const agent = await createOpenAIToolsAgent({ llm, tools, prompt });
-            return new AgentExecutor({ agent, tools });
-        }
-
-        const memberNames = this.members.map(member => member.name);
-
-        const systemPrompt = this.supervisorConfig.systemPrompt ||
-      'You are a supervisor tasked with managing a conversation between the' +
-      ' following workers: {members}. Given the following user request,' +
-      ' respond with the worker to act next. Each worker will perform a' +
-      ' task and respond with their results and status. When finished,' +
-      ' respond with FINISH.';
-        const options = [END, ...memberNames];
-
-        const functionDef = {
-            name: 'route',
-            description: 'Select the next role.',
-            parameters: {
-                title: 'routeSchema',
-                type: 'object',
-                properties: {
-                    next: {
-                        title: 'Next',
-                        anyOf: [
-                            { enum: options },
-                        ],
-                    },
-                },
-                required: ['next'],
-            },
-        };
-
-        const toolDef = {
-            type: 'function',
-            function: functionDef,
-        } as const;
-
-        const prompt = ChatPromptTemplate.fromMessages([
-            ['system', systemPrompt],
-            new MessagesPlaceholder('messages'),
-            [
-                'system',
-                'Given the conversation above, who should act next?' +
-        ' Or should we FINISH? Select one of: {options}',
-            ],
-        ]);
-
-        const formattedPrompt = await prompt.partial({
-            options: options.join(', '),
-            members: memberNames.join(', '),
-        });
-
-        const { provider, ...clientOptions } = this.supervisorConfig.llmConfig;
-        const LLMClass = llmProviders[provider];
-        if (!LLMClass) {
-            throw new Error(`Unsupported LLM provider for supervisor: ${provider}`);
-        }
-        const llm = new LLMClass(clientOptions);
-
-        const supervisorChain = formattedPrompt
-            .pipe(llm.bindTools(
-                [toolDef],
-                {
-                    tool_choice: { 'type': 'function', 'function': { 'name': 'route' } },
-                },
-            ))
-            .pipe(new JsonOutputToolsParser())
-            .pipe((x) => (x[0].args));
-
-        const workflow = new StateGraph({
-            channels: agentStateChannels,
-        });
-
-        // Dynamically create agents and add nodes for each member
-        for (const member of this.members) {
-            const agent = await createAgent(member.llmConfig, member.tools, member.systemPrompt);
-            const node = async (
-                state: AgentStateChannels,
-                config?: RunnableConfig,
-            ) => {
-                const result = await agent.invoke(state, config);
-                return {
-                    messages: [
-                        new HumanMessage({ content: result.output, name: member.name }),
-                    ],
-                };
-            };
-            workflow.addNode(member.name, node);
-            workflow.addEdge(member.name, 'supervisor');
-        }
-
-        workflow.addNode('supervisor', supervisorChain);
-
-        workflow.addConditionalEdges(
-            'supervisor',
-            (x: AgentStateChannels) => x.next,
-        );
-
-        workflow.addEdge(START, 'supervisor');
-
-        return workflow.compile();
-    }
-
-    async processStream(
-        inputs: { messages: BaseMessage[] },
-        config: Partial<RunnableConfig> & { version: 'v1' | 'v2' },
-    ) {
-        if (!this.graph) {
-            throw new Error('CollaborativeProcessor not initialized. Call initialize() first.');
-        }
-        const stream = this.graph.streamEvents(inputs, config);
-        for await (const event of stream) {
-            const handler = this.handlerRegistry.getHandler(event.event);
-            if (handler) {
-                handler.handle(event.event, event.data);
-            }
-        }
-    }
-}