agent0-js 0.0.12 → 0.0.13

package/README.md

@@ -73,6 +73,7 @@ const client = new Agent0({
 |--------|------|----------|---------|-------------|
 | `apiKey` | `string` | Yes | - | Your Agent0 API key |
 | `baseUrl` | `string` | No | `https://app.agent0.com` | The base URL for the Agent0 API |
+| `environment` | `'staging' \| 'production'` | No | `'production'` | Default environment for all runs (can be overridden per-run) |
 
 ## Methods
 
@@ -85,6 +86,7 @@ Execute an agent and get the complete response.
 ```typescript
 interface RunOptions {
   agentId: string;                    // The ID of the agent to run
+  environment?: 'staging' | 'production'; // Environment to run (default: 'production')
   variables?: Record<string, string>; // Variables to pass to the agent
   overrides?: ModelOverrides;         // Runtime model configuration overrides
   extraMessages?: Message[];          // Extra messages to append to the prompt
@@ -454,6 +456,54 @@ const response = await client.generate({
 // Prompt becomes: "Hello Sarah, let's talk about machine learning"
 ```
 
+### Staging and Production Environments
+
+Agent0 supports deploying different versions of your agent to staging and production environments. This allows you to test changes before rolling them out to production.
+
+The environment can be set at two levels with the following priority:
+1. **Run-level** (highest priority): Set `environment` in `generate()` or `stream()` options
+2. **Constructor-level**: Set `environment` when creating the `Agent0` client
+3. **Default**: Falls back to `'production'`
+
+```typescript
+// Set default environment at constructor level
+const stagingClient = new Agent0({
+  apiKey: process.env.AGENT0_API_KEY!,
+  environment: 'staging' // All runs will use staging by default
+});
+
+// This uses 'staging' from constructor
+const response1 = await stagingClient.generate({
+  agentId: 'agent_123',
+  variables: { name: 'Test User' }
+});
+
+// Override constructor setting at run level
+const response2 = await stagingClient.generate({
+  agentId: 'agent_123',
+  environment: 'production', // Overrides the constructor's 'staging'
+  variables: { name: 'Real User' }
+});
+
+// Default client (no constructor environment) uses 'production'
+const defaultClient = new Agent0({
+  apiKey: process.env.AGENT0_API_KEY!
+});
+
+// This uses 'production' (the default)
+const response3 = await defaultClient.generate({
+  agentId: 'agent_123',
+  variables: { name: 'User' }
+});
+
+// Run-level environment takes precedence
+const response4 = await defaultClient.generate({
+  agentId: 'agent_123',
+  environment: 'staging',
+  variables: { name: 'Test User' }
+});
+```
+
 ### Model Overrides
 
 The `overrides` option allows you to dynamically configure the model at runtime. This is useful for:

package/dist/index.d.ts

@@ -24,4 +24,4 @@ export declare class Agent0 {
      */
     embedMany(options: EmbedManyOptions): Promise<EmbedManyResponse>;
 }
-export type { Agent0Config, CustomTool, EmbedManyOptions, EmbedManyResponse, EmbedModel, EmbedOptions, EmbedResponse, GenerateResponse, ModelOverrides, ProviderOptions, RunOptions, } from "./types";
+export type { Agent0Config, CustomTool, EmbedManyOptions, EmbedManyResponse, EmbedModel, EmbedOptions, EmbedResponse, Environment, GenerateResponse, ModelOverrides, ProviderOptions, RunOptions, } from "./types";

package/dist/index.js

@@ -27,6 +27,7 @@ class Agent0 {
     async generate(options) {
         const response = await this.fetchApi("/api/v1/run", {
             agent_id: options.agentId,
+            environment: options.environment,
             variables: options.variables,
             overrides: options.overrides,
             extra_messages: options.extraMessages,
@@ -38,6 +39,7 @@ class Agent0 {
     async *stream(options) {
         const response = await this.fetchApi("/api/v1/run", {
             agent_id: options.agentId,
+            environment: options.environment,
             variables: options.variables,
             overrides: options.overrides,
             extra_messages: options.extraMessages,

package/dist/types.d.ts

@@ -50,8 +50,16 @@ export interface CustomTool {
     /** JSON Schema defining the parameters this tool accepts */
     inputSchema?: Record<string, unknown>;
 }
+/**
+ * Environment to run the agent in.
+ * - 'staging': Run the staging-deployed version of the agent
+ * - 'production': Run the production-deployed version of the agent (default)
+ */
+export type Environment = "staging" | "production";
 export interface RunOptions {
     agentId: string;
+    /** Environment to run ('staging' or 'production'). Defaults to 'production'. */
+    environment?: Environment;
     variables?: Record<string, string>;
     /** Runtime model overrides for load balancing, fallbacks, etc. */
     overrides?: ModelOverrides;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent0-js",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "description": "TypeScript SDK for Agent0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -5,6 +5,7 @@ import type {
 	EmbedManyResponse,
 	EmbedOptions,
 	EmbedResponse,
+	Environment,
 	GenerateResponse,
 	RunOptions,
 } from "./types";
@@ -12,10 +13,19 @@ import type {
 export class Agent0 {
 	private apiKey: string;
 	private baseUrl: string;
+	private environment?: Environment;
 
 	constructor(config: Agent0Config) {
 		this.apiKey = config.apiKey;
 		this.baseUrl = config.baseUrl || "https://app.agent0.com"; // Default URL, can be overridden
+		this.environment = config.environment;
+	}
+
+	/**
+	 * Resolve the environment to use: run-level > constructor-level > default 'production'
+	 */
+	private resolveEnvironment(runEnvironment?: Environment): Environment {
+		return runEnvironment ?? this.environment ?? "production";
 	}
 
 	private async fetchApi(endpoint: string, body: unknown): Promise<Response> {
@@ -46,6 +56,7 @@ export class Agent0 {
 	async generate(options: RunOptions): Promise<GenerateResponse> {
 		const response = await this.fetchApi("/api/v1/run", {
 			agent_id: options.agentId,
+			environment: this.resolveEnvironment(options.environment),
 			variables: options.variables,
 			overrides: options.overrides,
 			extra_messages: options.extraMessages,
@@ -61,6 +72,7 @@ export class Agent0 {
 	): AsyncGenerator<TextStreamPart<ToolSet>, void, unknown> {
 		const response = await this.fetchApi("/api/v1/run", {
 			agent_id: options.agentId,
+			environment: this.resolveEnvironment(options.environment),
 			variables: options.variables,
 			overrides: options.overrides,
 			extra_messages: options.extraMessages,
@@ -142,6 +154,7 @@ export type {
 	EmbedModel,
 	EmbedOptions,
 	EmbedResponse,
+	Environment,
 	GenerateResponse,
 	ModelOverrides,
 	ProviderOptions,

package/src/types.ts

@@ -6,6 +6,8 @@ import type { embed, embedMany, ModelMessage } from "ai";
 export interface Agent0Config {
 	apiKey: string;
 	baseUrl?: string;
+	/** Default environment to use for all runs. Can be overridden per-run. Defaults to 'production'. */
+	environment?: Environment;
 }
 
 /**
@@ -55,8 +57,17 @@ export interface CustomTool {
 	inputSchema?: Record<string, unknown>;
 }
 
+/**
+ * Environment to run the agent in.
+ * - 'staging': Run the staging-deployed version of the agent
+ * - 'production': Run the production-deployed version of the agent (default)
+ */
+export type Environment = "staging" | "production";
+
 export interface RunOptions {
 	agentId: string;
+	/** Environment to run ('staging' or 'production'). Defaults to 'production'. */
+	environment?: Environment;
 	variables?: Record<string, string>;
 	/** Runtime model overrides for load balancing, fallbacks, etc. */
 	overrides?: ModelOverrides;