From 9de397bf2000e2e96115f27afd29903f06a00f51 Mon Sep 17 00:00:00 2001
From: Pranay Prakash <pranay.gp@gmail.com>
Date: Tue, 4 Nov 2025 15:00:52 -0800
Subject: [PATCH] Use top-level DurableAgent import from @workflow/ai
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Update imports to use @workflow/ai instead of @workflow/ai/agent following the new preferred import pattern.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 flight-booking-app/workflows/chat/index.ts | 40 +++++++++++-----------
 rag-agent/workflows/chat/index.ts          |  2 +-
 2 files changed, 21 insertions(+), 21 deletions(-)

diff --git a/flight-booking-app/workflows/chat/index.ts b/flight-booking-app/workflows/chat/index.ts
index ebb5ddb..431a4fc 100644
--- a/flight-booking-app/workflows/chat/index.ts
+++ b/flight-booking-app/workflows/chat/index.ts
@@ -1,32 +1,32 @@
 import {
-	convertToModelMessages,
-	type UIMessage,
-	type UIMessageChunk,
-} from "ai";
-import { getWritable } from "workflow";
-import { DurableAgent } from "@workflow/ai/agent";
-import { FLIGHT_ASSISTANT_PROMPT, flightBookingTools } from "./steps/tools";
+  convertToModelMessages,
+  type UIMessage,
+  type UIMessageChunk,
+} from 'ai';
+import { getWritable } from 'workflow';
+import { DurableAgent } from '@workflow/ai';
+import { FLIGHT_ASSISTANT_PROMPT, flightBookingTools } from './steps/tools';
 
 /**
  * The main chat workflow
  */
 export async function chat(messages: UIMessage[]) {
-	"use workflow";
+  'use workflow';
 
-	console.log("Starting workflow");
+  console.log('Starting workflow');
 
-	const writable = getWritable<UIMessageChunk>();
+  const writable = getWritable<UIMessageChunk>();
 
-	const agent = new DurableAgent({
-		model: "bedrock/claude-4-sonnet-20250514-v1",
-		system: FLIGHT_ASSISTANT_PROMPT,
-		tools: flightBookingTools,
-	});
+  const agent = new DurableAgent({
+    model: 'bedrock/claude-4-sonnet-20250514-v1',
+    system: FLIGHT_ASSISTANT_PROMPT,
+    tools: flightBookingTools,
+  });
 
-	await agent.stream({
-		messages: convertToModelMessages(messages),
-		writable,
-	});
+  await agent.stream({
+    messages: convertToModelMessages(messages),
+    writable,
+  });
 
-	console.log("Finished workflow");
+  console.log('Finished workflow');
 }
diff --git a/rag-agent/workflows/chat/index.ts b/rag-agent/workflows/chat/index.ts
index 3ff5ea8..ad1eb78 100644
--- a/rag-agent/workflows/chat/index.ts
+++ b/rag-agent/workflows/chat/index.ts
@@ -4,7 +4,7 @@ import {
 	type UIMessageChunk,
 } from "ai";
 import { getWritable } from "workflow";
-import { DurableAgent } from "@workflow/ai/agent";
+import { DurableAgent } from "@workflow/ai";
 import { z } from "zod";
 import { createResource } from "./createResource";
 import { findRelevant } from "./findRelevant";