Cleanup and add skipped emit test

src/agent.test.ts

@@ -1,4 +1,4 @@
-import { test, expect } from 'vitest';
+import { test, expect, vi } from 'vitest';
 import { createAgent, type AIAdapter } from './';
 import { createActor, createMachine } from 'xstate';
 
@@ -175,13 +175,30 @@ test('Agents can use a custom adapter', async () => {
   expect(res.text).toEqual('Response');
 });
 
-test.skip('You can listen for emitted agent events', () => {
+test.skip('You can listen for emitted agent events', async () => {
+  const fn = vi.fn();
   const agent = createAgent({
     name: 'test',
     events: {},
     adapter: {} as any,
     model: {} as any,
   });
 
-  agent.on('feedback', (x) => {});
+  const p = new Promise((res) => {
+    agent.on('feedback', (f) => {
+      res(f);
+    });
+  });
+
+  agent.addFeedback({
+    attributes: {
+      score: -1,
+    },
+    goal: 'Win the game',
+    observationId: 'obs-1',
+  });
+
+  await p;
+
+  expect(fn).toHaveBeenCalled();
 });

src/index.ts

@@ -1,4 +1,4 @@
 export { createAgent } from './agent';
-export { fromText, fromTextStream, agentGenerateText } from './text';
-export { fromDecision, agentDecide } from './decision';
+export { fromText, fromTextStream } from './text';
+export { fromDecision } from './decision';
 export * from './types';

src/types.ts

@@ -109,7 +109,7 @@ export type PromptTemplate<TEvents extends EventObject> = (data: {
 
 export type AgentPlanner<T extends Agent<any>> = (
   agent: T['eventTypes'],
-  options: AgentPlanInput<T['eventTypes']>
+  input: AgentPlanInput<T['eventTypes']>
 ) => Promise<AgentPlan<T['eventTypes']> | undefined>;
 
 export type AgentDecideOptions = {
@@ -271,7 +271,7 @@ export type Agent<TEvents extends EventObject> = ActorRefFrom<
    *
    * - The `goal` for the agent to achieve
    * - The observed current `state`
-   * - The `logic` (e.g. a state machine) that specifies what can happen next
+   * - The `machine` (e.g. a state machine) that specifies what can happen next
    * - Additional `context`
    */
   decide: (