langtrain 0.1.6 → 0.1.9

package/dist/index.d.mts

@@ -4,3 +4,44 @@ export { Langvision } from 'langvision';
 import * as langtune from 'langtune';
 export { langtune as Text };
 export { Langtune } from 'langtune';
+
+interface Agent {
+    id: string;
+    workspace_id: string;
+    name: string;
+    description?: string;
+    model_id?: string;
+    config: any;
+    is_active: boolean;
+    created_at: string;
+    updated_at: string;
+}
+interface AgentRun {
+    id: string;
+    conversation_id: string;
+    success: boolean;
+    output?: any;
+    error?: string;
+    latency_ms: number;
+    tokens_used: number;
+}
+declare class AgentClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Agent[]>;
+    execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
+}
+
+type agent_Agent = Agent;
+type agent_AgentClient = AgentClient;
+declare const agent_AgentClient: typeof AgentClient;
+type agent_AgentRun = AgentRun;
+declare namespace agent {
+  export { type agent_Agent as Agent, agent_AgentClient as AgentClient, type agent_AgentRun as AgentRun };
+}
+
+export { type Agent, AgentClient, type AgentRun, agent as AgentTypes };

package/dist/index.d.ts

@@ -4,3 +4,44 @@ export { Langvision } from 'langvision';
 import * as langtune from 'langtune';
 export { langtune as Text };
 export { Langtune } from 'langtune';
+
+interface Agent {
+    id: string;
+    workspace_id: string;
+    name: string;
+    description?: string;
+    model_id?: string;
+    config: any;
+    is_active: boolean;
+    created_at: string;
+    updated_at: string;
+}
+interface AgentRun {
+    id: string;
+    conversation_id: string;
+    success: boolean;
+    output?: any;
+    error?: string;
+    latency_ms: number;
+    tokens_used: number;
+}
+declare class AgentClient {
+    private config;
+    private client;
+    constructor(config: {
+        apiKey: string;
+        baseUrl?: string;
+    });
+    list(workspaceId?: string): Promise<Agent[]>;
+    execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
+}
+
+type agent_Agent = Agent;
+type agent_AgentClient = AgentClient;
+declare const agent_AgentClient: typeof AgentClient;
+type agent_AgentRun = AgentRun;
+declare namespace agent {
+  export { type agent_Agent as Agent, agent_AgentClient as AgentClient, type agent_AgentRun as AgentRun };
+}
+
+export { type Agent, AgentClient, type AgentRun, agent as AgentTypes };

package/dist/index.js

@@ -1125,6 +1125,8 @@ var require_merge_stream2 = __commonJS({
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AgentClient: () => AgentClient,
+  AgentTypes: () => agent_exports,
   Langtune: () => Langtune,
   Langvision: () => Langvision,
   Text: () => dist_exports2,
@@ -6791,6 +6793,13 @@ var FinetuneConfigSchema = external_exports.object({
 var Langvision = class {
   constructor(options = {}) {
     this.cliPath = options.cliPath || "langvision";
+    this.apiKey = options.apiKey;
+  }
+  getEnv() {
+    return {
+      ...process.env,
+      LANGVISION_API_KEY: this.apiKey || process.env.LANGVISION_API_KEY
+    };
   }
   /**
    * Run a fine-tuning job using the local CLI
@@ -6815,7 +6824,10 @@ var Langvision = class {
       validated.outputDir
     ];
     try {
-      await execa(this.cliPath, args, { stdio: "inherit" });
+      await execa(this.cliPath, args, {
+        stdio: "inherit",
+        env: this.getEnv()
+      });
     } catch (error) {
       throw new Error(`Langvision fine-tuning failed: ${error}`);
     }
@@ -6838,7 +6850,9 @@ var Langvision = class {
       args.push("--temperature", options.temperature.toString());
     }
     try {
-      const { stdout } = await execa(this.cliPath, args);
+      const { stdout } = await execa(this.cliPath, args, {
+        env: this.getEnv()
+      });
       return stdout;
     } catch (error) {
       throw new Error(`Langvision generation failed: ${error}`);
@@ -12508,6 +12522,13 @@ var FinetuneConfigSchema2 = external_exports2.object({
 var Langtune = class {
   constructor(options = {}) {
     this.cliPath = options.cliPath || "langtune";
+    this.apiKey = options.apiKey;
+  }
+  getEnv() {
+    return {
+      ...process.env,
+      LANGTUNE_API_KEY: this.apiKey || process.env.LANGTUNE_API_KEY
+    };
   }
   /**
    * Run a fine-tuning job using the local CLI
@@ -12540,7 +12561,10 @@ var Langtune = class {
       args.push("--use-lisa");
     }
     try {
-      await execa2(this.cliPath, args, { stdio: "inherit" });
+      await execa2(this.cliPath, args, {
+        stdio: "inherit",
+        env: this.getEnv()
+      });
     } catch (error) {
       throw new Error(`Langtune fine-tuning failed: ${error}`);
     }
@@ -12563,15 +12587,52 @@ var Langtune = class {
       args.push("--temperature", options.temperature.toString());
     }
     try {
-      const { stdout } = await execa2(this.cliPath, args);
+      const { stdout } = await execa2(this.cliPath, args, {
+        env: this.getEnv()
+      });
       return stdout;
     } catch (error) {
       throw new Error(`Langtune generation failed: ${error}`);
     }
   }
 };
+
+// src/agent.ts
+var agent_exports = {};
+__export(agent_exports, {
+  AgentClient: () => AgentClient
+});
+var import_axios = __toESM(require("axios"));
+var AgentClient = class {
+  constructor(config) {
+    this.config = config;
+    this.client = import_axios.default.create({
+      baseURL: config.baseUrl || "https://api.langtrain.ai/api/v1",
+      headers: {
+        "X-API-Key": config.apiKey,
+        "Content-Type": "application/json"
+      }
+    });
+  }
+  async list(workspaceId) {
+    const params = {};
+    if (workspaceId) params.workspace_id = workspaceId;
+    const response = await this.client.get("/agents", { params });
+    return response.data.agents;
+  }
+  async execute(agentId, input, messages = [], conversationId) {
+    const response = await this.client.post(`/agents/${agentId}/execute`, {
+      input,
+      messages,
+      conversation_id: conversationId
+    });
+    return response.data;
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AgentClient,
+  AgentTypes,
   Langtune,
   Langvision,
   Text,