ocpipe 0.2.1 → 0.3.1

package/src/predict.ts

@@ -1,5 +1,5 @@
 /**
- * DSTS SDK Predict class.
+ * ocpipe Predict class.
  *
  * Executes a signature by generating a prompt, calling OpenCode, and parsing the response.
  */
@@ -47,8 +47,13 @@ export interface PredictConfig {
   correction?: CorrectionConfig | false
 }
 
+type AnySignature = SignatureDef<
+  Record<string, FieldConfig>,
+  Record<string, FieldConfig>
+>
+
 /** Predict executes a signature by calling an LLM and parsing the response. */
-export class Predict<S extends SignatureDef<any, any>> {
+export class Predict<S extends AnySignature> {
   constructor(
     public readonly sig: S,
     public readonly config: PredictConfig = {},
@@ -90,7 +95,11 @@ export class Predict<S extends SignatureDef<any, any>> {
     }
 
     // Parsing failed - attempt correction if enabled
-    if (this.config.correction !== false && parseResult.errors && parseResult.json) {
+    if (
+      this.config.correction !== false &&
+      parseResult.errors &&
+      parseResult.json
+    ) {
       const corrected = await this.correctFields(
         parseResult.json,
         parseResult.errors,
@@ -111,9 +120,19 @@ export class Predict<S extends SignatureDef<any, any>> {
 
     // Correction failed or disabled - throw SchemaValidationError (non-retryable)
     const errors = parseResult.errors ?? []
-    const errorMessages = errors.map((e) => `${e.path}: ${e.message}`).join('; ') || 'Unknown error'
-    const correctionAttempts = this.config.correction !== false ? (typeof this.config.correction === 'object' ? this.config.correction.maxRounds ?? 3 : 3) : 0
-    throw new SchemaValidationError(`Schema validation failed: ${errorMessages}`, errors, correctionAttempts)
+    const errorMessages =
+      errors.map((e) => `${e.path}: ${e.message}`).join('; ') || 'Unknown error'
+    const correctionAttempts =
+      this.config.correction !== false ?
+        typeof this.config.correction === 'object' ?
+          (this.config.correction.maxRounds ?? 3)
+        : 3
+      : 0
+    throw new SchemaValidationError(
+      `Schema validation failed: ${errorMessages}`,
+      errors,
+      correctionAttempts,
+    )
   }
 
   /** correctFields attempts to fix field errors using same-session patches with retries. */
@@ -123,32 +142,39 @@ export class Predict<S extends SignatureDef<any, any>> {
     ctx: ExecutionContext,
     sessionId: string,
   ): Promise<InferOutputs<S> | null> {
-    const correctionConfig = typeof this.config.correction === 'object' ? this.config.correction : {}
+    const correctionConfig =
+      typeof this.config.correction === 'object' ? this.config.correction : {}
     const method: CorrectionMethod = correctionConfig.method ?? 'json-patch'
     const maxFields = correctionConfig.maxFields ?? 5
     const maxRounds = correctionConfig.maxRounds ?? 3
     const correctionModel = correctionConfig.model
 
-    let currentJson = JSON.parse(JSON.stringify(json)) as Record<string, unknown>
+    let currentJson = JSON.parse(JSON.stringify(json)) as Record<
+      string,
+      unknown
+    >
     let currentErrors = initialErrors
 
     for (let round = 1; round <= maxRounds; round++) {
       const errorsToFix = currentErrors.slice(0, maxFields)
-      
+
       if (errorsToFix.length === 0) {
         break
       }
 
-      console.error(`\n>>> Correction round ${round}/${maxRounds} [${method}]: fixing ${errorsToFix.length} field(s)...`)
+      console.error(
+        `\n>>> Correction round ${round}/${maxRounds} [${method}]: fixing ${errorsToFix.length} field(s)...`,
+      )
 
       // Build prompt based on correction method
-      const patchPrompt = method === 'jq'
-        ? (errorsToFix.length === 1
-            ? buildPatchPrompt(errorsToFix[0]!, currentJson, this.sig.outputs)
-            : buildBatchPatchPrompt(errorsToFix, currentJson))
-        : (errorsToFix.length === 1
-            ? buildJsonPatchPrompt(errorsToFix[0]!, currentJson, this.sig.outputs)
-            : buildBatchJsonPatchPrompt(errorsToFix, currentJson))
+      const patchPrompt =
+        method === 'jq' ?
+          errorsToFix.length === 1 ?
+            buildPatchPrompt(errorsToFix[0]!, currentJson, this.sig.outputs)
+          : buildBatchPatchPrompt(errorsToFix, currentJson)
+        : errorsToFix.length === 1 ?
+          buildJsonPatchPrompt(errorsToFix[0]!, currentJson, this.sig.outputs)
+        : buildBatchJsonPatchPrompt(errorsToFix, currentJson)
 
       // Use same session (model has context) unless correction model specified
       const patchResult = await runAgent({
@@ -183,14 +209,16 @@ export class Predict<S extends SignatureDef<any, any>> {
 
       // Update errors for next round
       currentErrors = revalidated.errors ?? []
-      
+
       if (currentErrors.length === 0) {
         // No errors but also no data? Shouldn't happen, but handle gracefully
         console.error(`  Unexpected state: no errors but validation failed`)
         break
       }
 
-      console.error(`  Round ${round} complete, ${currentErrors.length} error(s) remaining`)
+      console.error(
+        `  Round ${round} complete, ${currentErrors.length} error(s) remaining`,
+      )
     }
 
     console.error(`  Schema correction failed after ${maxRounds} rounds`)
@@ -216,7 +244,7 @@ export class Predict<S extends SignatureDef<any, any>> {
 
     // Input fields as JSON
     const inputsWithDescriptions: Record<string, unknown> = {}
-    for (const [name, config] of Object.entries(this.sig.inputs) as [
+    for (const [name] of Object.entries(this.sig.inputs) as [
       string,
       FieldConfig,
     ][]) {
@@ -231,7 +259,9 @@ export class Predict<S extends SignatureDef<any, any>> {
     // Output format with JSON Schema
     lines.push('OUTPUT FORMAT:')
     lines.push('Return a JSON object matching this schema EXACTLY.')
-    lines.push('IMPORTANT: For optional fields, OMIT the field entirely - do NOT use null.')
+    lines.push(
+      'IMPORTANT: For optional fields, OMIT the field entirely - do NOT use null.',
+    )
     lines.push('')
     lines.push('```json')
     lines.push(JSON.stringify(this.buildOutputJsonSchema(), null, 2))
@@ -244,7 +274,10 @@ export class Predict<S extends SignatureDef<any, any>> {
   private buildOutputJsonSchema(): Record<string, unknown> {
     // Build a Zod object from the output fields
     const shape: Record<string, z.ZodType> = {}
-    for (const [name, config] of Object.entries(this.sig.outputs) as [string, FieldConfig][]) {
+    for (const [name, config] of Object.entries(this.sig.outputs) as [
+      string,
+      FieldConfig,
+    ][]) {
       shape[name] = config.type
     }
     const outputSchema = z.object(shape)
@@ -254,9 +287,14 @@ export class Predict<S extends SignatureDef<any, any>> {
 
     // Add field descriptions from our config (toJSONSchema uses .describe() metadata)
     // Since our FieldConfig has a separate desc field, merge it in
-    const props = jsonSchema.properties as Record<string, Record<string, unknown>> | undefined
+    const props = jsonSchema.properties as
+      | Record<string, Record<string, unknown>>
+      | undefined
     if (props) {
-      for (const [name, config] of Object.entries(this.sig.outputs) as [string, FieldConfig][]) {
+      for (const [name, config] of Object.entries(this.sig.outputs) as [
+        string,
+        FieldConfig,
+      ][]) {
         if (config.desc && props[name]) {
           // Only add if not already set by .describe()
           if (!props[name].description) {

package/src/signature.ts

@@ -1,5 +1,5 @@
 /**
- * DSTS SDK signature definition.
+ * ocpipe signature definition.
  *
  * Signatures declare input/output contracts for LLM interactions using Zod for validation.
  */

package/src/state.ts

@@ -1,5 +1,5 @@
 /**
- * DSTS SDK state management.
+ * ocpipe state management.
  *
  * Provides base state types and helpers for checkpointable workflow state.
  */

package/src/testing.ts

@@ -1,7 +1,7 @@
 /**
- * DSTS SDK testing utilities.
+ * ocpipe testing utilities.
  *
- * Provides mock backends and test helpers for unit testing DSTS components.
+ * Provides mock backends and test helpers for unit testing ocpipe components.
  */
 
 import type { RunAgentOptions, RunAgentResult, FieldConfig } from './types.js'
@@ -34,7 +34,10 @@ export class MockAgentBackend {
   }
 
   /** addJsonResponse adds a mock JSON response. */
-  addJsonResponse(data: Record<string, unknown>, options?: Partial<MockResponse>): this {
+  addJsonResponse(
+    data: Record<string, unknown>,
+    options?: Partial<MockResponse>,
+  ): this {
     return this.addResponse({
       response: JSON.stringify(data, null, 2),
       ...options,
@@ -76,7 +79,7 @@ export class MockAgentBackend {
     for (let i = 0; i < this.responses.length; i++) {
       const r = this.responses[i]
       if (!r) continue
-      
+
       if (!r.match) {
         response = r
         responseIndex = i
@@ -116,7 +119,8 @@ export class MockAgentBackend {
 
     return {
       text: response.response ?? '',
-      sessionId: response.sessionId ?? options.sessionId ?? this.defaultSessionId,
+      sessionId:
+        response.sessionId ?? options.sessionId ?? this.defaultSessionId,
     }
   }
 
@@ -127,12 +131,14 @@ export class MockAgentBackend {
 }
 
 /** createMockContext creates a test execution context. */
-export function createMockContext(overrides?: Partial<{
-  sessionId: string
-  defaultModel: { providerID: string; modelID: string }
-  defaultAgent: string
-  timeoutSec: number
-}>) {
+export function createMockContext(
+  overrides?: Partial<{
+    sessionId: string
+    defaultModel: { providerID: string; modelID: string }
+    defaultAgent: string
+    timeoutSec: number
+  }>,
+) {
   return {
     sessionId: overrides?.sessionId,
     defaultModel: overrides?.defaultModel ?? {
@@ -145,12 +151,14 @@ export function createMockContext(overrides?: Partial<{
 }
 
 /** generateMockOutputs creates mock output data based on a schema. */
-export function generateMockOutputs(schema: Record<string, FieldConfig>): Record<string, unknown> {
+export function generateMockOutputs(
+  schema: Record<string, FieldConfig>,
+): Record<string, unknown> {
   const result: Record<string, unknown> = {}
   for (const [name, config] of Object.entries(schema)) {
     // Use constructor name for type detection (works across zod versions)
     const typeName = config.type.constructor.name
-    
+
     switch (typeName) {
       case 'ZodString':
         result[name] = `mock_${name}`
@@ -167,11 +175,12 @@ export function generateMockOutputs(schema: Record<string, FieldConfig>): Record
       case 'ZodObject':
         result[name] = {}
         break
-      case 'ZodEnum':
+      case 'ZodEnum': {
         // Get first enum value via options property
         const enumType = config.type as { options?: readonly string[] }
         result[name] = enumType.options?.[0] ?? 'unknown'
         break
+      }
       default:
         result[name] = null
     }

package/src/types.ts

@@ -1,7 +1,5 @@
 /**
- * DSTS SDK shared types.
- *
- * Core type definitions for the Declarative Self-Improving TypeScript SDK.
+ * ocpipe shared types.
  */
 
 import type { z } from 'zod/v4'
@@ -131,16 +129,26 @@ export interface SignatureDef<
 }
 
 /** Infer the input type from a signature definition. */
-export type InferInputs<S extends SignatureDef<any, any>> =
-  S extends SignatureDef<infer I, any>
-    ? { [K in keyof I]: z.infer<I[K]['type']> }
-    : never
+export type InferInputs<
+  S extends SignatureDef<
+    Record<string, FieldConfig>,
+    Record<string, FieldConfig>
+  >,
+> =
+  S extends SignatureDef<infer I, Record<string, FieldConfig>> ?
+    { [K in keyof I]: z.infer<I[K]['type']> }
+  : never
 
 /** Infer the output type from a signature definition. */
-export type InferOutputs<S extends SignatureDef<any, any>> =
-  S extends SignatureDef<any, infer O>
-    ? { [K in keyof O]: z.infer<O[K]['type']> }
-    : never
+export type InferOutputs<
+  S extends SignatureDef<
+    Record<string, FieldConfig>,
+    Record<string, FieldConfig>
+  >,
+> =
+  S extends SignatureDef<Record<string, FieldConfig>, infer O> ?
+    { [K in keyof O]: z.infer<O[K]['type']> }
+  : never
 
 // ============================================================================
 // Retry Configuration