tjs-lang 0.8.2 → 0.8.3

package/src/vm/atom-effects.test.ts

@@ -0,0 +1,72 @@
+import { describe, it, expect } from 'bun:test'
+import { coreAtoms, EFFECTFUL_CORE_OPS, type AtomDef } from './runtime'
+import { batteryAtoms } from './atoms/index'
+
+const allAtoms = { ...coreAtoms, ...batteryAtoms } as Record<string, AtomDef>
+
+describe('atom effects classification (predicate-safety keystone)', () => {
+  it('every atom is classified pure | io', () => {
+    for (const [op, atom] of Object.entries(allAtoms)) {
+      expect(atom.effects, `${op} has no effects tag`).toMatch(/^(pure|io)$/)
+    }
+  })
+
+  it('the declared effectful core ops are all tagged io', () => {
+    for (const op of EFFECTFUL_CORE_OPS) {
+      expect(coreAtoms[op as keyof typeof coreAtoms]?.effects, op).toBe('io')
+    }
+  })
+
+  it('every battery atom is io (network calls)', () => {
+    for (const atom of Object.values(batteryAtoms) as AtomDef[]) {
+      expect(atom.effects, atom.op).toBe('io')
+    }
+  })
+
+  it('the pure substrate a predicate relies on is tagged pure', () => {
+    // data / control-flow atoms predicates compose from
+    const pure = [
+      'map',
+      'filter',
+      'reduce',
+      'find',
+      'len',
+      'split',
+      'join',
+      'keys',
+      'pick',
+      'omit',
+      'merge',
+      'regexMatch',
+      'jsonParse',
+      'callLocal',
+      'if',
+      'while',
+      'return',
+      'scope',
+    ]
+    for (const op of pure) {
+      expect(coreAtoms[op as keyof typeof coreAtoms]?.effects, op).toBe('pure')
+    }
+  })
+
+  it('the IO atoms are exactly the ones that may not appear in a predicate', () => {
+    const io = Object.values(allAtoms)
+      .filter((a) => a.effects === 'io')
+      .map((a) => a.op)
+    // sanity: the capability-touching atoms are all present
+    expect(io).toEqual(
+      expect.arrayContaining([
+        'httpFetch',
+        'storeGet',
+        'llmPredict',
+        'agentRun',
+        'runCode',
+        'llmVision',
+      ])
+    )
+    // and none of the pure data ops leaked in
+    expect(io).not.toContain('map')
+    expect(io).not.toContain('jsonParse')
+  })
+})

package/src/vm/atoms/batteries.ts

@@ -229,3 +229,15 @@ export const llmVision = defineAtom(
   },
   { docs: 'Analyze images using a vision model', timeoutMs: 120000, cost: 150 }
 )
+
+// Every battery atom is IO (embedding/LLM/vector-store network calls).
+for (const atom of [
+  storeVectorize,
+  storeCreateCollection,
+  storeVectorAdd,
+  storeSearch,
+  llmPredictBattery,
+  llmVision,
+]) {
+  atom.effects = 'io'
+}

package/src/vm/runtime.ts

@@ -180,6 +180,18 @@ export interface RuntimeContext {
 
 export type AtomExec = (step: any, ctx: RuntimeContext) => Promise<void>
 
+/**
+ * Effect classification of an atom.
+ * - `'pure'`: deterministic, no IO, no capability access, no observable side
+ *   effects — safe inside a synchronous predicate (see the predicate verifier).
+ * - `'io'`: touches `ctx.capabilities` (fetch/store/llm/agent/code), or is
+ *   nondeterministic (random/uuid), or has side effects (console). Not allowed
+ *   in a predicate.
+ * Defaults to `'pure'`; effectful atoms must opt into `'io'`. The invariant
+ * "anything touching ctx.capabilities is tagged 'io'" is guarded by a test.
+ */
+export type AtomEffects = 'pure' | 'io'
+
 export interface AtomDef {
   op: OpCode
   inputSchema: any
@@ -188,6 +200,7 @@ export interface AtomDef {
   docs?: string
   timeoutMs?: number
   cost?: number | ((input: any, ctx: RuntimeContext) => number)
+  effects?: AtomEffects
 }
 
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
@@ -199,6 +212,8 @@ export interface AtomOptions {
   docs?: string
   timeoutMs?: number
   cost?: number | ((input: any, ctx: RuntimeContext) => number)
+  /** Effect class — defaults to `'pure'`; effectful atoms set `'io'`. */
+  effects?: AtomEffects
 }
 
 export interface RunResult {
@@ -1347,6 +1362,7 @@ export function defineAtom<I extends Record<string, any>, O = any>(
     docs = '',
     timeoutMs = 1000,
     cost = 1,
+    effects = 'pure',
   } = typeof options === 'string' ? { docs: options } : options
 
   const exec: AtomExec = async (step: any, ctx: RuntimeContext) => {
@@ -1446,6 +1462,7 @@ export function defineAtom<I extends Record<string, any>, O = any>(
     docs,
     timeoutMs,
     cost,
+    effects,
     create: (input: I) => ({ op, ...input }),
   }
 }
@@ -3114,3 +3131,37 @@ export const coreAtoms = {
   releaseProcedure,
   clearExpiredProcedures,
 }
+
+/**
+ * Effectful core atoms — anything that touches `ctx.capabilities`
+ * (fetch/store/llm/agent/code), is nondeterministic (random/uuid), or has
+ * observable side effects (console). Tagged centrally so the list reads as one
+ * audit surface; a test asserts every capability-touching atom is in here.
+ * Everything else defaults to `effects: 'pure'`.
+ */
+export const EFFECTFUL_CORE_OPS = [
+  'httpFetch',
+  'storeGet',
+  'storeSet',
+  'storeQuery',
+  'storeVectorSearch',
+  'llmPredict',
+  'agentRun',
+  'transpileCode',
+  'runCode',
+  'random',
+  'uuid',
+  'consoleLog',
+  'consoleWarn',
+  'consoleError',
+  'storeProcedure',
+  'releaseProcedure',
+  'clearExpiredProcedures',
+  'cache',
+  'memoize',
+] as const
+
+for (const op of EFFECTFUL_CORE_OPS) {
+  const atom = (coreAtoms as Record<string, AtomDef>)[op]
+  if (atom) atom.effects = 'io'
+}