@rws-framework/ai-tools 3.0.0 → 3.1.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@rws-framework/ai-tools",
   "private": false,
-  "version": "3.0.0",
+  "version": "3.1.1",
   "description": "",
   "main": "src/index.ts",
   "scripts": {},
@@ -15,7 +15,8 @@
     "@rws-framework/db": "*",
     "@rws-framework/console": "*",
     "uuid": "^9.0.0",
-    "xml2js": "^0.6.2"
+    "xml2js": "^0.6.2",
+    "zod": "^3.25.76"
   },
   "devDependencies": {
     "@types/xml2js": "^0.4.14"

package/src/index.ts

@@ -1,5 +1,5 @@
 
-import RWSPrompt, { IChainCallOutput } from '@rws-framework/ai-tools/src/models/prompts/_prompt';
+import RWSPrompt, { IChainCallOutput } from './models/prompts/_prompt';
 import { ILLMChunk, IRWSPromptRequestExecutor, IRWSSinglePromptRequestExecutor, IRWSPromptStreamExecutor, IRWSPromptJSON, ChainStreamType, IAIRequestOptions, IAITool, IAIToolSchema, IAIToolParameter, IToolCall, ToolHandler } from './types/IPrompt';
 import { EmbedLoader as RWSEmbed, IConvoDebugXMLData, IEmbeddingsHandler, ISplitterParams } from './models/convo/EmbedLoader';
 import RWSVectorStore from './models/convo/VectorStore';
@@ -9,8 +9,10 @@ import { LangChainRAGService, ILangChainRAGConfig, IRAGIndexRequest, IRAGSearchR
 import { IContextToken } from './types/IContextToken';
 import { IEmbeddingConfig, IChunkConfig } from './types';
 import type { IAiCfg } from './types/IAiCfg';
+import { z as ZOD } from 'zod/v4';
 
 export {    
+    ZOD,
     IAiCfg,
     RWSVectorStore,
     RWSEmbed,

package/src/models/prompts/_prompt.ts

@@ -69,7 +69,7 @@ class RWSPrompt implements IPromptInstance {
     }
 
     // Delegation methods for input/output management
-    listen(source: string, stream: boolean = true): RWSPrompt {
+    listen(source: string | object, stream: boolean = true): RWSPrompt {        
         this.ioManager.listen(source, stream);
         return this;
     }
@@ -108,7 +108,7 @@ class RWSPrompt implements IPromptInstance {
         return this;
     }
 
-    readOutput(): string {
+    readOutput(): string | object {
         return this.ioManager.readOutput();
     }
 

package/src/models/prompts/inc/execution-methods-handler.ts

@@ -7,11 +7,11 @@ import {
 } from './types';
 
 export interface IPromptInstance {
-    readOutput(): string;
+    readOutput(): string | object;
     readInput(): any[];
     getInput(): any[];
     setSentInput(input: any[]): void;
-    injestOutput(content: string): void;
+    injestOutput(content: string | object): void;
 }
 
 export class ExecutionMethodsHandler {

package/src/models/prompts/inc/input-output-manager.ts

@@ -5,7 +5,7 @@ export class InputOutputManager {
     private enhancedInput: IPromptEnchantment[] = [];
     private sentInput: CompoundInput[] = [];
     private originalInput: CompoundInput[] = [];
-    private output: string = '';
+    private output: string | object = null;
     private onStream: (chunk: string) => void = () => {};
 
     constructor(input: CompoundInput[]) {
@@ -54,18 +54,20 @@ export class InputOutputManager {
         this.output = content;
     }
 
-    readOutput(): string {
+    readOutput(): string | object{
         return this.output;
     }
 
-    listen(source: string, stream: boolean = true): void {
-        this.output = '';
+    listen(source: string | object, stream: boolean = true): void {
+        if(stream){
+            this.output = '';
+        }        
 
         if (!stream) {
             this.output = source;
         } else {
-            this.output += source;
-            this.onStream(source);
+            this.output += source as string;
+            this.onStream(source as string);
         }
     }
 
@@ -114,7 +116,7 @@ export class InputOutputManager {
         return this.originalInput;
     }
 
-    getOutput(): string {
+    getOutput(): string | object {
         return this.output;
     }
 }

package/src/types/IPrompt.ts

@@ -3,6 +3,7 @@ import { IterableReadableStream } from '@langchain/core/utils/stream';
 import { ChainValues } from '@langchain/core/utils/types';
 import { IContextToken } from './IContextToken';
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import z4, { Schema } from 'zod/v4';
 
 // General tool interfaces for AI models
 interface IAIToolParameterBase {
@@ -92,6 +93,7 @@ interface IAIRequestOptions {
     intruderPrompt?: string | null, 
     ensureJson?: boolean, 
     debugVars?: any,
+    schema?: z4.core.$ZodType | Schema,
     tools?: IAITool[]
 }
 
@@ -114,7 +116,7 @@ interface IRWSPromptJSON {
     enhancedInput: IPromptEnchantment[];
     sentInput: CompoundInput[];
     originalInput: CompoundInput[];
-    output: string;
+    output: string | object;
     modelId: string;
     modelType: string;
     multiTemplate: PromptTemplate;