feat(langchain/createAgent): Simple standalone data analysis experiment

Author: pokey

libs/langchain/src/dataAnalysisExperiment.ts

@@ -0,0 +1,66 @@
+import { ChatAnthropic } from "@langchain/anthropic";
+import { HumanMessage } from "@langchain/core/messages";
+
+import { MemorySaver } from "@langchain/langgraph-checkpoint";
+import { createAgent } from "./agents/middlewareAgent/index.js";
+import { anthropicCodeExecutionMiddleware } from "./agents/middlewareAgent/middleware/anthropicCodeExecution.js";
+import {
+  downloadFileAnthropic,
+  extractGeneratedFilesAnthropic,
+  getFileMetadataAnthropic,
+  uploadFileAnthropic,
+} from "./agents/middlewareAgent/middleware/anthropicHelpers.js";
+
+// Initial setup
+const model = new ChatAnthropic({
+  model: "claude-sonnet-4-20250514",
+});
+const client = model.createClient({});
+const thread = {
+  configurable: {
+    thread_id: "test-123",
+  },
+};
+const agent = createAgent({
+  model,
+  middleware: [anthropicCodeExecutionMiddleware()],
+  checkpointSaver: new MemorySaver(),
+});
+
+// Upload the file to Anthropic
+const uploadedFile = await uploadFileAnthropic(client, "test_data.csv");
+
+// First invocation - should create container and analyze uploaded data
+const response1 = await agent.invoke(
+  {
+    messages: new HumanMessage({
+      content: [
+        {
+          type: "text",
+          text: "Filter to just widget A",
+        },
+        { type: "container_upload", file_id: uploadedFile.fileId },
+      ],
+    }),
+  },
+  thread
+);
+console.log("Response 1:", response1.messages);
+
+// Second invocation - should reuse container and previous analysis
+const response2 = await agent.invoke(
+  {
+    messages: new HumanMessage(
+      "Turn that into a graph of sales and units over time."
+    ),
+  },
+  thread
+);
+console.log("Response 2:", response2.messages);
+
+// Extract and download generated files
+for (const fileId of extractGeneratedFilesAnthropic(response2)) {
+  const metadata = await getFileMetadataAnthropic(client, fileId);
+  await downloadFileAnthropic(client, fileId, metadata.filename);
+  console.log(`Downloaded generated file: ${metadata.filename}`);
+}

libs/langchain/test_data.csv

@@ -0,0 +1,11 @@
+date,region,product,sales,units
+2024-01-15,North,Widget A,1250.50,25
+2024-01-16,South,Widget B,890.25,18
+2024-01-17,East,Widget A,1456.75,29
+2024-01-18,West,Widget C,2100.00,42
+2024-01-19,North,Widget B,765.50,15
+2024-01-20,South,Widget A,1890.00,38
+2024-01-21,East,Widget C,2340.25,47
+2024-01-22,West,Widget B,1123.75,22
+2024-01-23,North,Widget C,1567.50,31
+2024-01-24,South,Widget B,934.25,19