@snap-agent/core 0.1.2 → 0.1.3

package/dist/index.d.mts

@@ -1,6 +1,6 @@
 import { P as ProviderConfig, a as ProviderType, b as Plugin, S as StoredPluginConfig, A as AgentData, c as StorageAdapter, d as AgentConfig, e as AgentFile, R as RAGDocument, I as IngestOptions, f as IngestResult, B as BulkOperation, g as BulkResult, U as URLSource, h as URLIngestResult, T as ThreadData, i as ThreadConfig, M as MessageRole, j as MessageAttachment, k as MessageData, C as ClientConfig, l as ChatRequest, m as ChatResponse, n as StreamCallbacks, o as RAGPlugin, p as ToolPlugin, q as MiddlewarePlugin, r as AnalyticsPlugin, s as RAGContext } from './index-m2vDW79n.mjs';
 export { L as AgentNotFoundError, K as AgentSDKError, u as BasePlugin, D as DataTransform, H as ErrorTrackingData, w as IngestionSchedule, Q as InvalidConfigError, W as MemoryStorage, V as MongoDBStorage, Y as MongoDBStorageConfig, y as PerformanceTimings, J as PluginRegistryInterface, O as ProviderNotFoundError, t as RAGConfig, z as RAGMetrics, F as RequestTrackingData, G as ResponseTrackingData, N as ThreadNotFoundError, E as TokenMetrics, x as Tool, v as URLSourceAuth, X as UpstashStorage, Z as UpstashStorageConfig } from './index-m2vDW79n.mjs';
-import { LanguageModel, UserModelMessage, AssistantModelMessage } from 'ai';
+import { LanguageModel, UserModelMessage, AssistantModelMessage, Schema } from 'ai';
 
 /**
  * Provider factory for creating language model instances
@@ -215,12 +215,19 @@ declare class Agent {
     /**
      * Generate a text response with optional plugin support
      */
-    generateResponse(messages: AIMessage$1[], options?: {
+    generateResponse<T = unknown>(messages: AIMessage$1[], options?: {
         useRAG?: boolean;
         ragFilters?: Record<string, any>;
         threadId?: string;
+        output?: {
+            mode: 'json';
+        } | {
+            mode: 'object';
+            schema: Schema<T>;
+        };
     }): Promise<{
         text: string;
+        parsed?: T;
         metadata?: Record<string, any>;
     }>;
     /**

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { P as ProviderConfig, a as ProviderType, b as Plugin, S as StoredPluginConfig, A as AgentData, c as StorageAdapter, d as AgentConfig, e as AgentFile, R as RAGDocument, I as IngestOptions, f as IngestResult, B as BulkOperation, g as BulkResult, U as URLSource, h as URLIngestResult, T as ThreadData, i as ThreadConfig, M as MessageRole, j as MessageAttachment, k as MessageData, C as ClientConfig, l as ChatRequest, m as ChatResponse, n as StreamCallbacks, o as RAGPlugin, p as ToolPlugin, q as MiddlewarePlugin, r as AnalyticsPlugin, s as RAGContext } from './index-m2vDW79n.js';
 export { L as AgentNotFoundError, K as AgentSDKError, u as BasePlugin, D as DataTransform, H as ErrorTrackingData, w as IngestionSchedule, Q as InvalidConfigError, W as MemoryStorage, V as MongoDBStorage, Y as MongoDBStorageConfig, y as PerformanceTimings, J as PluginRegistryInterface, O as ProviderNotFoundError, t as RAGConfig, z as RAGMetrics, F as RequestTrackingData, G as ResponseTrackingData, N as ThreadNotFoundError, E as TokenMetrics, x as Tool, v as URLSourceAuth, X as UpstashStorage, Z as UpstashStorageConfig } from './index-m2vDW79n.js';
-import { LanguageModel, UserModelMessage, AssistantModelMessage } from 'ai';
+import { LanguageModel, UserModelMessage, AssistantModelMessage, Schema } from 'ai';
 
 /**
  * Provider factory for creating language model instances
@@ -215,12 +215,19 @@ declare class Agent {
     /**
      * Generate a text response with optional plugin support
      */
-    generateResponse(messages: AIMessage$1[], options?: {
+    generateResponse<T = unknown>(messages: AIMessage$1[], options?: {
         useRAG?: boolean;
         ragFilters?: Record<string, any>;
         threadId?: string;
+        output?: {
+            mode: 'json';
+        } | {
+            mode: 'object';
+            schema: Schema<T>;
+        };
     }): Promise<{
         text: string;
+        parsed?: T;
         metadata?: Record<string, any>;
     }>;
     /**

package/dist/index.js

@@ -7688,11 +7688,37 @@ var Agent = class _Agent {
       ragMetadata = allMetadata;
     }
     const model = await this.providerFactory.getModel(this.data.provider, this.data.model);
-    const { text } = await (0, import_ai.generateText)({
-      model,
-      messages: beforeResult.messages,
-      system: systemPrompt
-    });
+    let text;
+    let parsed;
+    if (options?.output?.mode === "object") {
+      const result = await (0, import_ai.generateText)({
+        model,
+        messages: beforeResult.messages,
+        system: systemPrompt,
+        experimental_output: import_ai.Output.object({ schema: options.output.schema })
+      });
+      text = JSON.stringify(result.experimental_output);
+      parsed = result.experimental_output;
+    } else if (options?.output?.mode === "json") {
+      const jsonSystemPrompt = systemPrompt + "\n\n---\nOUTPUT FORMAT: You MUST respond with valid JSON only. No markdown code blocks, no explanations, no additional text - just raw JSON that can be parsed directly.";
+      const result = await (0, import_ai.generateText)({
+        model,
+        messages: beforeResult.messages,
+        system: jsonSystemPrompt
+      });
+      text = result.text;
+      try {
+        parsed = JSON.parse(text);
+      } catch {
+      }
+    } else {
+      const result = await (0, import_ai.generateText)({
+        model,
+        messages: beforeResult.messages,
+        system: systemPrompt
+      });
+      text = result.text;
+    }
     const afterResult = await this.pluginManager.executeAfterResponse(text, {
       agentId: this.data.id,
       threadId: options?.threadId,
@@ -7708,6 +7734,7 @@ var Agent = class _Agent {
     });
     return {
       text: afterResult.response,
+      ...parsed !== void 0 && { parsed },
       metadata: {
         ...afterResult.metadata,
         ragMetadata,

package/dist/index.mjs

@@ -5,7 +5,7 @@ import {
 } from "./chunk-FS7G3ID4.mjs";
 
 // src/core/Agent.ts
-import { generateText, streamText } from "ai";
+import { generateText, streamText, Output } from "ai";
 
 // src/core/PluginManager.ts
 var PluginManager = class {
@@ -352,11 +352,37 @@ var Agent = class _Agent {
       ragMetadata = allMetadata;
     }
     const model = await this.providerFactory.getModel(this.data.provider, this.data.model);
-    const { text } = await generateText({
-      model,
-      messages: beforeResult.messages,
-      system: systemPrompt
-    });
+    let text;
+    let parsed;
+    if (options?.output?.mode === "object") {
+      const result = await generateText({
+        model,
+        messages: beforeResult.messages,
+        system: systemPrompt,
+        experimental_output: Output.object({ schema: options.output.schema })
+      });
+      text = JSON.stringify(result.experimental_output);
+      parsed = result.experimental_output;
+    } else if (options?.output?.mode === "json") {
+      const jsonSystemPrompt = systemPrompt + "\n\n---\nOUTPUT FORMAT: You MUST respond with valid JSON only. No markdown code blocks, no explanations, no additional text - just raw JSON that can be parsed directly.";
+      const result = await generateText({
+        model,
+        messages: beforeResult.messages,
+        system: jsonSystemPrompt
+      });
+      text = result.text;
+      try {
+        parsed = JSON.parse(text);
+      } catch {
+      }
+    } else {
+      const result = await generateText({
+        model,
+        messages: beforeResult.messages,
+        system: systemPrompt
+      });
+      text = result.text;
+    }
     const afterResult = await this.pluginManager.executeAfterResponse(text, {
       agentId: this.data.id,
       threadId: options?.threadId,
@@ -372,6 +398,7 @@ var Agent = class _Agent {
     });
     return {
       text: afterResult.response,
+      ...parsed !== void 0 && { parsed },
       metadata: {
         ...afterResult.metadata,
         ragMetadata,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@snap-agent/core",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "SnapAgent - The lightweight, snap-in AI agent SDK. Multi-provider support. Edge-runtime compatible.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -89,6 +89,7 @@
     "eslint": "^8.0.0",
     "mongodb": "^7.0.0",
     "tsup": "^8.0.0",
+    "tsx": "^4.21.0",
     "typescript": "^5.8.0",
     "vitest": "^1.0.0"
   },