@statelyai/agent 1.0.0-beta.0 → 1.0.0

package/src/utils.ts

@@ -1,18 +1,50 @@
-import { AnyMachineSnapshot, AnyStateNode } from 'xstate';
+import { AnyMachineSnapshot, AnyStateMachine, AnyStateNode } from 'xstate';
+import hash from 'object-hash';
 import { TransitionData } from './types';
 
 export function getAllTransitions(state: AnyMachineSnapshot): TransitionData[] {
   const nodes = state._nodes;
   const transitions = (nodes as AnyStateNode[])
     .map((node) => [...(node as AnyStateNode).transitions.values()])
-    .flat(2)
-    .map((transition) => ({
-      ...transition,
-      guard:
-        typeof transition.guard === 'string'
-          ? { type: transition.guard }
-          : (transition.guard as any), // TODO: fix
-    }));
+    .map((nodeTransitions) => {
+      return nodeTransitions.map((nodeEventTransitions) => {
+        return nodeEventTransitions.map((transition) => {
+          return {
+            ...transition,
+            guard:
+              typeof transition.guard === 'string'
+                ? { type: transition.guard }
+                : (transition.guard as any), // TODO: fix
+          };
+        });
+      });
+    })
+    .flat(2);
+
+  return transitions;
+}
+
+export function getAllMachineTransitions(
+  stateNode: AnyStateNode
+): TransitionData[] {
+  const transitions: TransitionData[] = [...stateNode.transitions.values()]
+    .map((nodeTransitions) => {
+      return nodeTransitions.map((transition) => {
+        return {
+          ...transition,
+          guard:
+            typeof transition.guard === 'string'
+              ? { type: transition.guard }
+              : (transition.guard as any), // TODO: fix
+        };
+      });
+    })
+    .flat(2);
+
+  for (const s of Object.values(stateNode.states)) {
+    const stateTransitions = getAllMachineTransitions(s);
+    transitions.push(...stateTransitions);
+  }
 
   return transitions;
 }
@@ -20,3 +52,21 @@ export function getAllTransitions(state: AnyMachineSnapshot): TransitionData[] {
 export function wrapInXml(tagName: string, content: string): string {
   return `<${tagName}>${content}</${tagName}>`;
 }
+
+export function randomId() {
+  const timestamp = Date.now().toString(36);
+  const random = Math.random().toString(36).substring(2, 9);
+  return timestamp + random;
+}
+
+const machineHashes: WeakMap<AnyStateMachine, string> = new WeakMap();
+/**
+ * Returns a string hash representing only the transitions in the state machine.
+ */
+export function getMachineHash(machine: AnyStateMachine): string {
+  if (machineHashes.has(machine)) return machineHashes.get(machine)!;
+  const transitions = getAllMachineTransitions(machine.root);
+  const machineHash = hash(transitions);
+  machineHashes.set(machine, machineHash);
+  return machineHash;
+}

package/src/templates/defaultToolCall.ts

@@ -1,10 +0,0 @@
-import { PromptTemplate } from '../types';
-import { defaultTextTemplate } from './defaultText';
-
-export const defaultToolCallTemplate: PromptTemplate<any> = (data) => {
-  return `
-${defaultTextTemplate(data)}
-
-Only make a single tool call to achieve the above goal.
-  `.trim();
-};