types

daydreamsai · Jan 29, 2025 · 1ab2450 · 1ab2450
1 parent 841f0d0
commit 1ab2450
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/packages/core/src/core/processors/research-processor.ts b/packages/core/src/core/processors/research-processor.ts
@@ -1,5 +1,5 @@
 import { LLMClient } from "../llm-client";
-import type { Character } from "../types";
+import type { ActionIOHandler, Character, OutputIOHandler } from "../types";
 import { LogLevel } from "../types";
 import { getTimeContext, validateLLMResponseSchema } from "../utils";
 import { z } from "zod";
@@ -199,7 +199,9 @@ export class ResearchQuantProcessor extends BaseProcessor {
         }
     }
 
-    private buildHandlerSchemaPart(handlers?: IOHandler[]): string {
+    private buildHandlerSchemaPart(
+        handlers?: OutputIOHandler[] | ActionIOHandler[]
+    ): string {
         if (!handlers || handlers.length === 0) return "None";
         return handlers
             .filter((handler) => handler.outputSchema)
@@ -217,8 +219,8 @@ export class ResearchQuantProcessor extends BaseProcessor {
     private async combineChunkResults(
         results: any[],
         ioContext?: {
-            availableOutputs: IOHandler[];
-            availableActions: IOHandler[];
+            availableOutputs: OutputIOHandler[];
+            availableActions: ActionIOHandler[];
         }
     ): Promise<any> {
         const prompt = `
@@ -321,8 +323,8 @@ export class ResearchQuantProcessor extends BaseProcessor {
         content: any,
         otherContext: string,
         ioContext?: {
-            availableOutputs: IOHandler[];
-            availableActions: IOHandler[];
+            availableOutputs: OutputIOHandler[];
+            availableActions: ActionIOHandler[];
         }
     ): Promise<any> {
         const contentStr =