@statelyai/agent 1.1.5 → 2.0.0-next.0

package/examples/verify.ts

@@ -1,5 +1,5 @@
 import { assign, createActor, setup, log } from 'xstate';
-import { getFromTerminal } from './helpers/helpers';
+import { fromTerminal } from './helpers/helpers';
 import { createAgent, fromDecision } from '../src';
 import { z } from 'zod';
 import { openai } from '@ai-sdk/openai';
@@ -38,7 +38,7 @@ const machine = setup({
     events: agent.types.events,
   },
   actors: {
-    getFromTerminal,
+    getFromTerminal: fromTerminal,
     agent: fromDecision(agent),
   },
 }).createMachine({

package/examples/weather.ts

@@ -1,6 +1,6 @@
 import { createAgent, fromDecision } from '../src';
 import { assign, createActor, fromPromise, log, setup } from 'xstate';
-import { getFromTerminal } from './helpers/helpers';
+import { fromTerminal } from './helpers/helpers';
 import { z } from 'zod';
 import { openai } from '@ai-sdk/openai';
 
@@ -50,7 +50,7 @@ const getWeather = fromPromise(async ({ input }: { input: string }) => {
 
 const agent = createAgent({
   name: 'weather',
-  model: openai('gpt-4-1106-preview'),
+  model: openai('gpt-4o-mini'),
   events: {
     'agent.getWeather': z.object({
       location: z.string().describe('The location to get the weather for'),
@@ -83,7 +83,7 @@ const machine = setup({
   actors: {
     agent: fromDecision(agent),
     getWeather,
-    getFromTerminal,
+    getFromTerminal: fromTerminal,
   },
 }).createMachine({
   initial: 'getLocation',
@@ -149,6 +149,7 @@ const machine = setup({
         src: 'agent',
         input: ({ context }) => ({
           goal: 'Report the weather', // TODO
+          context,
         }),
       },
       on: {
@@ -167,11 +168,7 @@ const machine = setup({
   },
 });
 
-const actor = createActor(machine, {
-  input: {
-    location: 'New York',
-  },
-});
+const actor = createActor(machine);
 actor.subscribe((s) => {
   console.log(s.value);
 });

package/examples/wiki.ts

@@ -1,30 +1,49 @@
 import { z } from 'zod';
 import { createAgent } from '../src';
 import { openai } from '@ai-sdk/openai';
+import { CoreMessage, generateText, streamText } from 'ai';
 
 const agent = createAgent({
   name: 'wiki',
-  model: openai('gpt-4-turbo'),
+  model: openai('gpt-4o-mini'),
   events: {
     provideAnswer: z.object({
       answer: z.string().describe('The answer'),
     }),
+    researchTopic: z.object({
+      topic: z.string().describe('The topic to research'),
+    }),
   },
 });
 
+agent.onMessage((msg) => {
+  console.log(msg);
+});
+
 async function main() {
-  const response1 = await agent.generateText({
+  const response = await generateText({
+    model: agent.model,
     prompt: 'When was Deadpool 2 released?',
   });
 
-  console.log(response1.text);
+  for (const msg of await response.responseMessages) {
+    agent.addMessage(msg);
+  }
 
-  const response2 = await agent.generateText({
-    messages: (x) => x.select((ctx) => ctx.messages),
-    prompt: 'What about the first one?',
+  const response2 = await streamText({
+    model: agent.model,
+    messages: (agent.getMessages() as CoreMessage[]).concat({
+      role: 'user',
+      content: 'What about the first one?',
+    }),
   });
 
-  console.log(response2.text);
+  let text = '';
+
+  for await (const t of response2.textStream) {
+    text += t;
+    console.log(text);
+  }
 }
 
 main();

package/examples/word.ts

@@ -1,18 +1,17 @@
 import { assign, createActor, log, setup } from 'xstate';
-import { getFromTerminal } from './helpers/helpers';
+import { fromTerminal } from './helpers/helpers';
 import { createAgent, fromDecision } from '../src';
 import { z } from 'zod';
 import { openai } from '@ai-sdk/openai';
 
-const context = {
-  word: null as string | null,
-  guessedWord: null as string | null,
-  lettersGuessed: [] as string[],
-};
-
 const agent = createAgent({
   name: 'word',
-  model: openai('gpt-4o'),
+  model: openai('gpt-4o-mini'),
+  context: {
+    word: z.string().nullable().describe('The word to guess'),
+    guessedWord: z.string().nullable().describe('The guessed word'),
+    lettersGuessed: z.array(z.string()).describe('The letters guessed'),
+  },
   events: {
     'agent.guessLetter': z.object({
       letter: z.string().min(1).max(1).describe('The letter guessed'),
@@ -33,14 +32,20 @@ const agent = createAgent({
   },
 });
 
+const context = {
+  word: null,
+  guessedWord: null,
+  lettersGuessed: [],
+} satisfies typeof agent.types.context;
+
 const wordGuesserMachine = setup({
   types: {
-    context: {} as typeof context,
+    context: agent.types.context,
     events: agent.types.events,
   },
   actors: {
     agent: fromDecision(agent),
-    getFromTerminal,
+    getFromTerminal: fromTerminal,
   },
   actions: {
     resetContext: assign(context),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@statelyai/agent",
-  "version": "1.1.5",
+  "version": "2.0.0-next.0",
   "description": "Stateful agents that make decisions based on finite-state machine models",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -12,22 +12,24 @@
     "rl",
     "reinforcement learning"
   ],
-  "author": "",
+  "author": "David Khourshid <david@stately.ai>",
   "license": "MIT",
   "devDependencies": {
+    "@ai-sdk/openai": "^0.0.40",
     "@changesets/changelog-github": "^0.5.0",
-    "@changesets/cli": "^2.27.7",
+    "@changesets/cli": "^2.27.9",
     "@langchain/community": "^0.0.53",
     "@langchain/core": "^0.1.63",
     "@langchain/openai": "^0.0.28",
-    "@types/node": "^20.14.14",
+    "@tavily/core": "^0.0.2",
+    "@types/node": "^20.17.1",
     "@types/object-hash": "^3.0.6",
     "dotenv": "^16.4.5",
-    "json-schema-to-ts": "^3.1.0",
+    "json-schema-to-ts": "^3.1.1",
     "ts-node": "^10.9.2",
-    "tsup": "^8.2.4",
-    "typescript": "^5.5.4",
-    "vitest": "^2.0.5",
+    "tsup": "^8.3.5",
+    "typescript": "^5.6.3",
+    "vitest": "^2.1.3",
     "wikipedia": "^2.1.2",
     "zod": "^3.23.8"
   },
@@ -35,11 +37,12 @@
     "access": "public"
   },
   "dependencies": {
-    "@ai-sdk/openai": "^0.0.40",
-    "@xstate/graph": "^2.0.0",
-    "ai": "^3.3.2",
+    "@xstate/graph": "^2.0.1",
+    "ai": "^3.4.20",
+    "ajv": "^8.17.1",
     "object-hash": "^3.0.0",
-    "xstate": "^5.17.1"
+    "xstate": "^5.18.2",
+    "zod-to-json-schema": "^3.23.5"
   },
   "scripts": {
     "build": "tsup src/index.ts --format cjs,esm --dts",

package/readme.md

@@ -5,6 +5,6 @@ Stately Agent is a flexible framework for building AI agents using state machine
 - Using state machines to guide the agent's behavior, powered by [XState](https://stately.ai/docs/xstate)
 - Incorporating **observations**, **message history**, and **feedback** to the agent decision-making and text-generation processes, as needed
 - Enabling custom **planning** abilities for agents to achieve specific goals based on state machine logic, observations, and feedback
-- Wrapping the [Vercel AI SDK](https://sdk.vercel.ai/) to easily support multiple model providers, such as OpenAI, Anthropic, Google, Mistral, Groq, Perplexity, and more
+- First-class integration with the [Vercel AI SDK](https://sdk.vercel.ai/) to easily support multiple model providers, such as OpenAI, Anthropic, Google, Mistral, Groq, Perplexity, and more
 
 **Read the documentation: [stately.ai/docs/agents](https://stately.ai/docs/agents)**

package/src/agent-experimental.ts

@@ -127,7 +127,7 @@
 //     //     openai,
 //     //     goal,
 //     //     state,
-//     //     'gpt-4-1106-preview',
+//     //     'gpt-4o-mini',
 //     //     (eventType) => eventType.startsWith('agent.'),
 //     //     schemas
 //     //   );