langsmith 0.0.2 → 0.0.3

package/dist/client.cjs

@@ -214,7 +214,7 @@ class Client {
         });
         await raiseForStatus(response, "delete run");
     }
-    async createProject({ projectName, projectExtra, mode, upsert, }) {
+    async createProject({ projectName, projectExtra, upsert, }) {
         const upsert_ = upsert ? `?upsert=true` : "";
         const endpoint = `${this.apiUrl}/sessions${upsert_}`;
         const body = {
@@ -223,9 +223,6 @@ class Client {
         if (projectExtra !== undefined) {
             body["extra"] = projectExtra;
         }
-        if (mode !== undefined) {
-            body["mode"] = mode;
-        }
         const response = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: { ...this.headers, "Content-Type": "application/json" },
@@ -290,15 +287,25 @@ class Client {
         });
         await raiseForStatus(response, `delete session ${projectId_} (${projectName})`);
     }
-    async uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, }) {
+    async uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, dataType, name, }) {
         const url = `${this.apiUrl}/datasets/upload`;
         const formData = new FormData();
         formData.append("file", csvFile, fileName);
-        formData.append("input_keys", inputKeys.join(","));
-        formData.append("output_keys", outputKeys.join(","));
+        inputKeys.forEach((key) => {
+            formData.append("input_keys", key);
+        });
+        outputKeys.forEach((key) => {
+            formData.append("output_keys", key);
+        });
         if (description) {
             formData.append("description", description);
         }
+        if (dataType) {
+            formData.append("data_type", dataType);
+        }
+        if (name) {
+            formData.append("name", name);
+        }
         const response = await this.caller.call(fetch, url, {
             method: "POST",
             headers: this.headers,
@@ -315,14 +322,18 @@ class Client {
         const result = await response.json();
         return result;
     }
-    async createDataset(name, { description } = {}) {
+    async createDataset(name, { description, dataType, } = {}) {
+        const body = {
+            name,
+            description,
+        };
+        if (dataType) {
+            body.data_type = dataType;
+        }
         const response = await this.caller.call(fetch, `${this.apiUrl}/datasets`, {
             method: "POST",
             headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify({
-                name,
-                description,
-            }),
+            body: JSON.stringify(body),
             signal: AbortSignal.timeout(this.timeout_ms),
         });
         if (!response.ok) {

package/dist/client.d.ts

@@ -1,5 +1,5 @@
 import { AsyncCallerParams } from "./utils/async_caller.js";
-import { Dataset, Example, ExampleUpdate, Feedback, KVMap, Run, RunCreate, RunType, RunUpdate, ScoreType, TracerSession, TracerSessionResult, ValueType } from "./schemas.js";
+import { Dataset, Example, ExampleUpdate, Feedback, KVMap, Run, RunCreate, RunType, RunUpdate, ScoreType, TracerSession, TracerSessionResult, ValueType, DataType } from "./schemas.js";
 import { RunEvaluator } from "./evaluation/evaluator.js";
 interface ClientConfig {
     apiUrl?: string;
@@ -23,6 +23,8 @@ interface UploadCSVParams {
     inputKeys: string[];
     outputKeys: string[];
     description?: string;
+    dataType?: DataType;
+    name?: string;
 }
 interface CreateRunParams {
     name: string;
@@ -62,10 +64,9 @@ export declare class Client {
     private _loadChildRuns;
     listRuns({ projectId, projectName, executionOrder, runType, error, id, limit, offset, }: ListRunsParams): Promise<Run[]>;
     deleteRun(runId: string): Promise<void>;
-    createProject({ projectName, projectExtra, mode, upsert, }: {
+    createProject({ projectName, projectExtra, upsert, }: {
         projectName: string;
         projectExtra?: object;
-        mode?: string;
         upsert?: boolean;
     }): Promise<TracerSession>;
     readProject({ projectId, projectName, }: {
@@ -77,9 +78,10 @@ export declare class Client {
         projectId?: string;
         projectName?: string;
     }): Promise<void>;
-    uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, }: UploadCSVParams): Promise<Dataset>;
-    createDataset(name: string, { description }?: {
+    uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, dataType, name, }: UploadCSVParams): Promise<Dataset>;
+    createDataset(name: string, { description, dataType, }?: {
         description?: string;
+        dataType?: DataType;
     }): Promise<Dataset>;
     readDataset({ datasetId, datasetName, }: {
         datasetId?: string;

package/dist/client.js

@@ -188,7 +188,7 @@ export class Client {
         });
         await raiseForStatus(response, "delete run");
     }
-    async createProject({ projectName, projectExtra, mode, upsert, }) {
+    async createProject({ projectName, projectExtra, upsert, }) {
         const upsert_ = upsert ? `?upsert=true` : "";
         const endpoint = `${this.apiUrl}/sessions${upsert_}`;
         const body = {
@@ -197,9 +197,6 @@ export class Client {
         if (projectExtra !== undefined) {
             body["extra"] = projectExtra;
         }
-        if (mode !== undefined) {
-            body["mode"] = mode;
-        }
         const response = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: { ...this.headers, "Content-Type": "application/json" },
@@ -264,15 +261,25 @@ export class Client {
         });
         await raiseForStatus(response, `delete session ${projectId_} (${projectName})`);
     }
-    async uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, }) {
+    async uploadCsv({ csvFile, fileName, inputKeys, outputKeys, description, dataType, name, }) {
         const url = `${this.apiUrl}/datasets/upload`;
         const formData = new FormData();
         formData.append("file", csvFile, fileName);
-        formData.append("input_keys", inputKeys.join(","));
-        formData.append("output_keys", outputKeys.join(","));
+        inputKeys.forEach((key) => {
+            formData.append("input_keys", key);
+        });
+        outputKeys.forEach((key) => {
+            formData.append("output_keys", key);
+        });
         if (description) {
             formData.append("description", description);
         }
+        if (dataType) {
+            formData.append("data_type", dataType);
+        }
+        if (name) {
+            formData.append("name", name);
+        }
         const response = await this.caller.call(fetch, url, {
             method: "POST",
             headers: this.headers,
@@ -289,14 +296,18 @@ export class Client {
         const result = await response.json();
         return result;
     }
-    async createDataset(name, { description } = {}) {
+    async createDataset(name, { description, dataType, } = {}) {
+        const body = {
+            name,
+            description,
+        };
+        if (dataType) {
+            body.data_type = dataType;
+        }
         const response = await this.caller.call(fetch, `${this.apiUrl}/datasets`, {
             method: "POST",
             headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify({
-                name,
-                description,
-            }),
+            body: JSON.stringify(body),
             signal: AbortSignal.timeout(this.timeout_ms),
         });
         if (!response.ok) {

package/dist/schemas.d.ts

@@ -3,7 +3,6 @@ export interface TracerSession {
     id: string;
     start_time: number;
     name?: string;
-    mode?: string;
 }
 export interface TracerSessionResult extends TracerSession {
     run_count?: number;
@@ -21,6 +20,7 @@ export type KVMap = Record<string, any>;
 export type RunType = "llm" | "chain" | "tool" | "retriever" | "embedding";
 export type ScoreType = number | boolean | null;
 export type ValueType = number | boolean | string | object | null;
+export type DataType = "kv" | "llm" | "chat";
 export interface BaseExample {
     dataset_id: string;
     inputs: KVMap;
@@ -85,6 +85,7 @@ export interface BaseDataset {
     name: string;
     description: string;
     tenant_id: string;
+    data_type?: DataType;
 }
 export interface Dataset extends BaseDataset {
     id: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "files": [
     "dist/",