illuma-agents 1.0.2 → 1.0.3

package/src/scripts/cli3.ts

@@ -1,178 +1,184 @@
-/* eslint-disable no-console */
-// src/scripts/cli.ts
-import { config } from 'dotenv';
-config();
-import { z } from "zod";
-import { HumanMessage, BaseMessage } from '@langchain/core/messages';
-import type { RunnableConfig } from '@langchain/core/runnables';
-import { TavilySearchResults } from '@langchain/community/tools/tavily_search';
-import { tool } from "@langchain/core/tools";
-import type * as t from '@/types';
-import { ModelEndHandler, ToolEndHandler } from '@/events';
-import { ChatModelStreamHandler } from '@/stream';
-
-
-import { getArgs } from '@/scripts/args';
-import { Run } from '@/run';
-import { GraphEvents, Callback } from '@/common';
-import { getLLMConfig } from '@/utils/llmConfig';
-
-const conversationHistory: BaseMessage[] = [];
-async function testStandardStreaming(): Promise<void> {
-  const { userName, location, provider, currentDate } = await getArgs();
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
-    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_RUN_STEP ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_RUN_STEP_DELTA ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_MESSAGE_DELTA ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.TOOL_START]: {
-      handle: (_event: string, data: t.StreamEventData, metadata?: Record<string, unknown>): void => {
-        console.log('====== TOOL_START ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    // [GraphEvents.LLM_STREAM]: new LLMStreamHandler(),
-    // [GraphEvents.LLM_START]: {
-    //   handle: (_event: string, data: t.StreamEventData): void => {
-    //     console.log('====== LLM_START ======');
-    //     console.dir(data, { depth: null });
-    //   }
-    // },
-    // [GraphEvents.LLM_END]: {
-    //   handle: (_event: string, data: t.StreamEventData): void => {
-    //     console.log('====== LLM_END ======');
-    //     console.dir(data, { depth: null });
-    //   }
-    // },
-    /*
-    [GraphEvents.CHAIN_START]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== CHAIN_START ======');
-        // console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.CHAIN_END]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== CHAIN_END ======');
-        // console.dir(data, { depth: null });
-      }
-    },
-    */
-    // [GraphEvents.CHAT_MODEL_START]: {
-    //   handle: (_event: string, _data: t.StreamEventData): void => {
-    //     console.log('====== CHAT_MODEL_START ======');
-    //     console.dir(_data, { depth: null });
-    //     // Intentionally left empty
-    //   }
-    // },
-  };
-
-  const llmConfig = getLLMConfig(provider);
-
-  const getWeather = tool(async ({ location }) => {
-    if (location === "SAN FRANCISCO") {
-      return "It's 60 degrees and foggy";
-    } else if (location.toLowerCase() === "san francisco") {
-      throw new Error("Input queries must be all capitals");
-    } else {
-      throw new Error("Invalid input.");
-    }
-  }, {
-    name: "get_weather",
-    description: "Call to get the current weather",
-    schema: z.object({
-      location: z.string(),
-    }),
-  });
-
-  const run = await Run.create<t.IState>({
-    runId: 'test-run-id',
-    graphConfig: {
-      type: 'standard',
-      llmConfig,
-      tools: [getWeather],
-      // tools: [new TavilySearchResults()],
-    },
-    customHandlers,
-  });
-
-  const config = {
-    configurable: {
-      provider,
-      thread_id: 'conversation-num-1',
-      instructions: 'You are a friendly AI assistant. Always address the user by their name.',
-      additional_instructions: `The user's name is ${userName} and they are located in ${location}.`
-    },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  console.log(' Test 1: Initial greeting');
-
-  conversationHistory.push(new HumanMessage(`Hi I'm ${userName}.`));
-  let inputs = {
-    messages: conversationHistory,
-  };
-  const contentParts = await run.processStream(inputs, config, {
-    callbacks: {
-    [Callback.TOOL_ERROR]: (graph, error, toolId) => {
-      console.error(`Tool ${toolId} failed with error: ${error.message}`);
-    },
-  },
-  });
-  const finalMessages = run.getRunMessages();
-  if (finalMessages) {
-    conversationHistory.push(...finalMessages);
-  }
-
-  console.log(' Test 2: Weather query');
-
-  const userMessage = `
-  Make a search for the weather in ${location} today, which is ${currentDate}.
-  Make sure to always refer to me by name.
-  After giving me a thorough summary, tell me a joke about the weather forecast we went over.
-  `;
-
-  conversationHistory.push(new HumanMessage(userMessage));
-
-  inputs = {
-    messages: conversationHistory,
-  };
-  const contentParts2 = await run.processStream(inputs, config);
-  const finalMessages2 = run.getRunMessages();
-  if (finalMessages2) {
-    conversationHistory.push(...finalMessages2);
-    console.dir(conversationHistory, { depth: null });
-  }
-}
-
-process.on('unhandledRejection', (reason, promise) => {
-  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-  console.log('Conversation history:');
-  console.dir(conversationHistory, { depth: null });
-  process.exit(1);
-});
-
-testStandardStreaming().catch((err) => {
-  console.error(err);
-  console.log('Conversation history:');
-  console.dir(conversationHistory, { depth: null });
-  process.exit(1);
-});
+/* eslint-disable no-console */
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { z } from 'zod';
+import { HumanMessage, BaseMessage } from '@langchain/core/messages';
+import type { RunnableConfig } from '@langchain/core/runnables';
+import { tool } from '@langchain/core/tools';
+import type * as t from '@/types';
+import { ModelEndHandler, ToolEndHandler } from '@/events';
+import { ChatModelStreamHandler } from '@/stream';
+
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents, Callback } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+
+const conversationHistory: BaseMessage[] = [];
+async function testStandardStreaming(): Promise<void> {
+  const { userName, location, provider, currentDate } = await getArgs();
+  const customHandlers = {
+    [GraphEvents.TOOL_END]: new ToolEndHandler(),
+    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+    [GraphEvents.ON_RUN_STEP]: {
+      handle: (_event: string, data: t.StreamEventData): void => {
+        console.log('====== ON_RUN_STEP ======');
+        console.dir(data, { depth: null });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP_DELTA]: {
+      handle: (_event: string, data: t.StreamEventData): void => {
+        console.log('====== ON_RUN_STEP_DELTA ======');
+        console.dir(data, { depth: null });
+      },
+    },
+    [GraphEvents.ON_MESSAGE_DELTA]: {
+      handle: (_event: string, data: t.StreamEventData): void => {
+        console.log('====== ON_MESSAGE_DELTA ======');
+        console.dir(data, { depth: null });
+      },
+    },
+    [GraphEvents.TOOL_START]: {
+      handle: (
+        _event: string,
+        data: t.StreamEventData,
+        metadata?: Record<string, unknown>
+      ): void => {
+        console.log('====== TOOL_START ======');
+        console.dir(data, { depth: null });
+      },
+    },
+    // [GraphEvents.LLM_STREAM]: new LLMStreamHandler(),
+    // [GraphEvents.LLM_START]: {
+    //   handle: (_event: string, data: t.StreamEventData): void => {
+    //     console.log('====== LLM_START ======');
+    //     console.dir(data, { depth: null });
+    //   }
+    // },
+    // [GraphEvents.LLM_END]: {
+    //   handle: (_event: string, data: t.StreamEventData): void => {
+    //     console.log('====== LLM_END ======');
+    //     console.dir(data, { depth: null });
+    //   }
+    // },
+    /*
+    [GraphEvents.CHAIN_START]: {
+      handle: (_event: string, data: t.StreamEventData): void => {
+        console.log('====== CHAIN_START ======');
+        // console.dir(data, { depth: null });
+      }
+    },
+    [GraphEvents.CHAIN_END]: {
+      handle: (_event: string, data: t.StreamEventData): void => {
+        console.log('====== CHAIN_END ======');
+        // console.dir(data, { depth: null });
+      }
+    },
+    */
+    // [GraphEvents.CHAT_MODEL_START]: {
+    //   handle: (_event: string, _data: t.StreamEventData): void => {
+    //     console.log('====== CHAT_MODEL_START ======');
+    //     console.dir(_data, { depth: null });
+    //     // Intentionally left empty
+    //   }
+    // },
+  };
+
+  const llmConfig = getLLMConfig(provider);
+
+  const getWeather = tool(
+    async ({ location }) => {
+      if (location === 'SAN FRANCISCO') {
+        return "It's 60 degrees and foggy";
+      } else if (location.toLowerCase() === 'san francisco') {
+        throw new Error('Input queries must be all capitals');
+      } else {
+        throw new Error('Invalid input.');
+      }
+    },
+    {
+      name: 'get_weather',
+      description: 'Call to get the current weather',
+      schema: z.object({
+        location: z.string(),
+      }),
+    }
+  );
+
+  const run = await Run.create<t.IState>({
+    runId: 'test-run-id',
+    graphConfig: {
+      type: 'standard',
+      llmConfig,
+      tools: [getWeather],
+      // tools: [],
+    },
+    customHandlers,
+  });
+
+  const config = {
+    configurable: {
+      provider,
+      thread_id: 'conversation-num-1',
+      instructions:
+        'You are a friendly AI assistant. Always address the user by their name.',
+      additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
+    },
+    streamMode: 'values',
+    version: 'v2' as const,
+  };
+
+  console.log(' Test 1: Initial greeting');
+
+  conversationHistory.push(new HumanMessage(`Hi I'm ${userName}.`));
+  let inputs = {
+    messages: conversationHistory,
+  };
+  const contentParts = await run.processStream(inputs, config, {
+    callbacks: {
+      [Callback.TOOL_ERROR]: (graph, error, toolId) => {
+        console.error(`Tool ${toolId} failed with error: ${error.message}`);
+      },
+    },
+  });
+  const finalMessages = run.getRunMessages();
+  if (finalMessages) {
+    conversationHistory.push(...finalMessages);
+  }
+
+  console.log(' Test 2: Weather query');
+
+  const userMessage = `
+  Make a search for the weather in ${location} today, which is ${currentDate}.
+  Make sure to always refer to me by name.
+  After giving me a thorough summary, tell me a joke about the weather forecast we went over.
+  `;
+
+  conversationHistory.push(new HumanMessage(userMessage));
+
+  inputs = {
+    messages: conversationHistory,
+  };
+  const contentParts2 = await run.processStream(inputs, config);
+  const finalMessages2 = run.getRunMessages();
+  if (finalMessages2) {
+    conversationHistory.push(...finalMessages2);
+    console.dir(conversationHistory, { depth: null });
+  }
+}
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+  console.log('Conversation history:');
+  console.dir(conversationHistory, { depth: null });
+  process.exit(1);
+});
+
+testStandardStreaming().catch((err) => {
+  console.error(err);
+  console.log('Conversation history:');
+  console.dir(conversationHistory, { depth: null });
+  process.exit(1);
+});