langsmith 0.0.1-0

package/dist/cli/main.js

@@ -0,0 +1,252 @@
+import * as fs from "fs";
+import * as path from "path";
+import * as util from "util";
+import { Command } from "commander";
+import * as child_process from "child_process";
+import { setEnvironmentVariable } from "../utils/env.js";
+const currentFileName = __filename;
+const currentDirName = __dirname;
+const exec = util.promisify(child_process.exec);
+const program = new Command();
+async function getDockerComposeCommand() {
+    try {
+        await exec("docker compose --version");
+        return ["docker", "compose"];
+    }
+    catch {
+        try {
+            await exec("docker-compose --version");
+            return ["docker-compose"];
+        }
+        catch {
+            throw new Error("Neither 'docker compose' nor 'docker-compose' commands are available. Please install the Docker server following the instructions for your operating system at https://docs.docker.com/engine/install/");
+        }
+    }
+}
+async function pprintServices(servicesStatus) {
+    const services = [];
+    for (const service of servicesStatus) {
+        const serviceStatus = {
+            Service: String(service["Service"]),
+            Status: String(service["Status"]),
+        };
+        const publishers = service["Publishers"] || [];
+        if (publishers) {
+            serviceStatus["PublishedPorts"] = publishers
+                .map((publisher) => String(publisher["PublishedPort"]))
+                .join(", ");
+        }
+        services.push(serviceStatus);
+    }
+    const maxServiceLen = Math.max(...services.map((service) => service["Service"].length));
+    const maxStateLen = Math.max(...services.map((service) => service["Status"].length));
+    const serviceMessage = [
+        "\n" +
+            "Service".padEnd(maxServiceLen + 2) +
+            "Status".padEnd(maxStateLen + 2) +
+            "Published Ports",
+    ];
+    for (const service of services) {
+        const serviceStr = service["Service"].padEnd(maxServiceLen + 2);
+        const stateStr = service["Status"].padEnd(maxStateLen + 2);
+        const portsStr = service["PublishedPorts"] || "";
+        serviceMessage.push(serviceStr + stateStr + portsStr);
+    }
+    let langchainEndpoint = "http://localhost:1984";
+    const usedNgrok = services.some((service) => service["Service"].includes("ngrok"));
+    if (usedNgrok) {
+        langchainEndpoint = await getNgrokUrl();
+    }
+    serviceMessage.push("\nTo connect, set the following environment variables" +
+        " in your LangChain application:" +
+        "\nLANGCHAIN_TRACING_V2=true" +
+        `\nLANGCHAIN_ENDPOINT=${langchainEndpoint}`);
+    console.info(serviceMessage.join("\n"));
+}
+async function getNgrokUrl() {
+    const ngrokUrl = "http://localhost:4040/api/tunnels";
+    try {
+        // const response = await axios.get(ngrokUrl);
+        const response = await fetch(ngrokUrl);
+        if (response.status !== 200) {
+            throw new Error(`Could not connect to ngrok console. ${response.status}, ${response.statusText}`);
+        }
+        const result = await response.json();
+        const exposedUrl = result["tunnels"][0]["public_url"];
+        return exposedUrl;
+    }
+    catch (error) {
+        throw new Error(`Could not connect to ngrok console. ${error}`);
+    }
+}
+async function createNgrokConfig(authToken) {
+    const configPath = path.join(currentDirName, "ngrok_config.yaml");
+    // Check if is a directory
+    if (fs.existsSync(configPath) && fs.lstatSync(configPath).isDirectory()) {
+        fs.rmdirSync(configPath, { recursive: true });
+    }
+    else if (fs.existsSync(configPath)) {
+        fs.unlinkSync(configPath);
+    }
+    let ngrokConfig = `
+region: us
+tunnels:
+  langchain:
+    addr: langchain-backend:1984
+    proto: http
+version: '2'
+`;
+    if (authToken !== null) {
+        ngrokConfig += `authtoken: ${authToken}`;
+    }
+    fs.writeFileSync(configPath, ngrokConfig);
+    return configPath;
+}
+class PlusCommand {
+    constructor({ dockerComposeCommand }) {
+        Object.defineProperty(this, "dockerComposeCommand", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "dockerComposeFile", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ""
+        });
+        Object.defineProperty(this, "ngrokPath", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ""
+        });
+        this.dockerComposeCommand = dockerComposeCommand;
+        this.dockerComposeFile = path.join(path.dirname(currentFileName), "docker-compose.yaml");
+        this.ngrokPath = path.join(path.dirname(currentFileName), "docker-compose.ngrok.yaml");
+    }
+    static async create() {
+        const dockerComposeCommand = await getDockerComposeCommand();
+        return new PlusCommand({ dockerComposeCommand });
+    }
+    async start(args) {
+        if (args.dev) {
+            setEnvironmentVariable("_LANGCHAINPLUS_IMAGE_PREFIX", "rc-");
+        }
+        if (args.openaiApiKey) {
+            setEnvironmentVariable("OPENAI_API_KEY", args.openaiApiKey);
+        }
+        if (args.expose) {
+            await this.startAndExpose(args.ngrokAuthtoken);
+        }
+        else {
+            await this.startLocal();
+        }
+    }
+    async pull(args) {
+        if (args.dev) {
+            setEnvironmentVariable("_LANGCHAINPLUS_IMAGE_PREFIX", "rc-");
+        }
+        const command = [
+            ...this.dockerComposeCommand,
+            "-f",
+            this.dockerComposeFile,
+            "pull",
+        ];
+        await exec(command.join(" "));
+    }
+    async startLocal() {
+        const command = [
+            ...this.dockerComposeCommand,
+            "-f",
+            this.dockerComposeFile,
+            "up",
+            "--pull=always",
+            "--quiet-pull",
+            "--wait",
+        ];
+        await exec(command.join(" "));
+        console.info("LangSmith server is running at http://localhost:1984.  To connect locally, set the following environment variable when running your LangChain application.");
+        console.info("\tLANGCHAIN_TRACING_V2=true");
+    }
+    async startAndExpose(ngrokAuthToken) {
+        const configPath = await createNgrokConfig(ngrokAuthToken);
+        const command = [
+            ...this.dockerComposeCommand,
+            "-f",
+            this.dockerComposeFile,
+            "-f",
+            this.ngrokPath,
+            "up",
+            "--pull=always",
+            "--quiet-pull",
+            "--wait",
+        ];
+        await exec(command.join(" "));
+        console.info("ngrok is running. You can view the dashboard at http://0.0.0.0:4040");
+        const ngrokUrl = await getNgrokUrl();
+        console.info("LangSmith server is running at http://localhost:1984. To connect remotely, set the following environment variable when running your LangChain application.");
+        console.info("\tLANGCHAIN_TRACING_V2=true");
+        console.info(`\tLANGCHAIN_ENDPOINT=${ngrokUrl}`);
+        fs.unlinkSync(configPath);
+    }
+    async stop() {
+        const command = [
+            ...this.dockerComposeCommand,
+            "-f",
+            this.dockerComposeFile,
+            "-f",
+            this.ngrokPath,
+            "down",
+        ];
+        await exec(command.join(" "));
+    }
+    async status() {
+        const command = [
+            ...this.dockerComposeCommand,
+            "-f",
+            this.dockerComposeFile,
+            "ps",
+            "--format",
+            "json",
+        ];
+        const result = await exec(command.join(" "));
+        const servicesStatus = JSON.parse(result.stdout);
+        if (servicesStatus) {
+            console.info("The LangSmith server is currently running.");
+            await pprintServices(servicesStatus);
+        }
+        else {
+            console.info("The LangSmith server is not running.");
+        }
+    }
+}
+const startCommand = new Command("start")
+    .description("Start the LangSmith server")
+    .option("--expose", "Expose the server to the internet via ngrok (requires ngrok to be installed)")
+    .option("--ngrok-authtoken <ngrokAuthtoken>", "Your ngrok auth token. If this is set, --expose is implied.")
+    .option("--dev", "Run the development version of the LangSmith server")
+    .option("--openai-api-key <openaiApiKey>", "Your OpenAI API key. If this is set, the server will be able to process text and return enhanced plus results.")
+    .action(async (args) => (await PlusCommand.create()).start(args));
+const stopCommand = new Command("stop")
+    .command("stop")
+    .description("Stop the LangSmith server")
+    .action(async () => (await PlusCommand.create()).stop());
+const pullCommand = new Command("pull")
+    .command("pull")
+    .description("Pull the latest version of the LangSmith server")
+    .option("--dev", "Pull the development version of the LangSmith server")
+    .action(async (args) => (await PlusCommand.create()).pull(args));
+const statusCommand = new Command("status")
+    .command("status")
+    .description("Get the status of the LangSmith server")
+    .action(async () => (await PlusCommand.create()).status());
+program
+    .command("plus")
+    .description("Manage the LangSmith server")
+    .addCommand(startCommand)
+    .addCommand(stopCommand)
+    .addCommand(pullCommand)
+    .addCommand(statusCommand);
+program.parse(process.argv);

package/dist/cli/main.ts

@@ -0,0 +1,292 @@
+import * as fs from "fs";
+import * as path from "path";
+import * as util from "util";
+import { Command } from "commander";
+import * as child_process from "child_process";
+import { setEnvironmentVariable } from "../utils/env.js";
+
+const currentFileName = __filename;
+const currentDirName = __dirname;
+
+const exec = util.promisify(child_process.exec);
+
+const program = new Command();
+
+async function getDockerComposeCommand(): Promise<string[]> {
+  try {
+    await exec("docker compose --version");
+    return ["docker", "compose"];
+  } catch {
+    try {
+      await exec("docker-compose --version");
+      return ["docker-compose"];
+    } catch {
+      throw new Error(
+        "Neither 'docker compose' nor 'docker-compose' commands are available. Please install the Docker server following the instructions for your operating system at https://docs.docker.com/engine/install/"
+      );
+    }
+  }
+}
+
+async function pprintServices(servicesStatus: any[]) {
+  const services = [];
+  for (const service of servicesStatus) {
+    const serviceStatus: Record<string, string> = {
+      Service: String(service["Service"]),
+      Status: String(service["Status"]),
+    };
+    const publishers = service["Publishers"] || [];
+    if (publishers) {
+      serviceStatus["PublishedPorts"] = publishers
+        .map((publisher: any) => String(publisher["PublishedPort"]))
+        .join(", ");
+    }
+    services.push(serviceStatus);
+  }
+
+  const maxServiceLen = Math.max(
+    ...services.map((service) => service["Service"].length)
+  );
+  const maxStateLen = Math.max(
+    ...services.map((service) => service["Status"].length)
+  );
+  const serviceMessage = [
+    "\n" +
+      "Service".padEnd(maxServiceLen + 2) +
+      "Status".padEnd(maxStateLen + 2) +
+      "Published Ports",
+  ];
+  for (const service of services) {
+    const serviceStr = service["Service"].padEnd(maxServiceLen + 2);
+    const stateStr = service["Status"].padEnd(maxStateLen + 2);
+    const portsStr = service["PublishedPorts"] || "";
+    serviceMessage.push(serviceStr + stateStr + portsStr);
+  }
+
+  let langchainEndpoint = "http://localhost:1984";
+  const usedNgrok = services.some((service) =>
+    service["Service"].includes("ngrok")
+  );
+  if (usedNgrok) {
+    langchainEndpoint = await getNgrokUrl();
+  }
+
+  serviceMessage.push(
+    "\nTo connect, set the following environment variables" +
+      " in your LangChain application:" +
+      "\nLANGCHAIN_TRACING_V2=true" +
+      `\nLANGCHAIN_ENDPOINT=${langchainEndpoint}`
+  );
+  console.info(serviceMessage.join("\n"));
+}
+
+async function getNgrokUrl(): Promise<string> {
+  const ngrokUrl = "http://localhost:4040/api/tunnels";
+  try {
+    // const response = await axios.get(ngrokUrl);
+    const response = await fetch(ngrokUrl);
+    if (response.status !== 200) {
+      throw new Error(
+        `Could not connect to ngrok console. ${response.status}, ${response.statusText}`
+      );
+    }
+    const result = await response.json();
+    const exposedUrl = result["tunnels"][0]["public_url"];
+    return exposedUrl;
+  } catch (error) {
+    throw new Error(`Could not connect to ngrok console. ${error}`);
+  }
+}
+
+async function createNgrokConfig(authToken: string | null): Promise<string> {
+  const configPath = path.join(currentDirName, "ngrok_config.yaml");
+  // Check if is a directory
+  if (fs.existsSync(configPath) && fs.lstatSync(configPath).isDirectory()) {
+    fs.rmdirSync(configPath, { recursive: true });
+  } else if (fs.existsSync(configPath)) {
+    fs.unlinkSync(configPath);
+  }
+  let ngrokConfig = `
+region: us
+tunnels:
+  langchain:
+    addr: langchain-backend:1984
+    proto: http
+version: '2'
+`;
+
+  if (authToken !== null) {
+    ngrokConfig += `authtoken: ${authToken}`;
+  }
+  fs.writeFileSync(configPath, ngrokConfig);
+  return configPath;
+}
+
+class PlusCommand {
+  dockerComposeCommand: string[] = [];
+  dockerComposeFile = "";
+  ngrokPath = "";
+
+  constructor({ dockerComposeCommand }: { dockerComposeCommand: string[] }) {
+    this.dockerComposeCommand = dockerComposeCommand;
+    this.dockerComposeFile = path.join(
+      path.dirname(currentFileName),
+      "docker-compose.yaml"
+    );
+    this.ngrokPath = path.join(
+      path.dirname(currentFileName),
+      "docker-compose.ngrok.yaml"
+    );
+  }
+
+  public static async create() {
+    const dockerComposeCommand = await getDockerComposeCommand();
+    return new PlusCommand({ dockerComposeCommand });
+  }
+
+  async start(args: any) {
+    if (args.dev) {
+      setEnvironmentVariable("_LANGCHAINPLUS_IMAGE_PREFIX", "rc-");
+    }
+    if (args.openaiApiKey) {
+      setEnvironmentVariable("OPENAI_API_KEY", args.openaiApiKey);
+    }
+
+    if (args.expose) {
+      await this.startAndExpose(args.ngrokAuthtoken);
+    } else {
+      await this.startLocal();
+    }
+  }
+
+  async pull(args: any) {
+    if (args.dev) {
+      setEnvironmentVariable("_LANGCHAINPLUS_IMAGE_PREFIX", "rc-");
+    }
+
+    const command = [
+      ...this.dockerComposeCommand,
+      "-f",
+      this.dockerComposeFile,
+      "pull",
+    ];
+    await exec(command.join(" "));
+  }
+
+  async startLocal() {
+    const command = [
+      ...this.dockerComposeCommand,
+      "-f",
+      this.dockerComposeFile,
+      "up",
+      "--pull=always",
+      "--quiet-pull",
+      "--wait",
+    ];
+    await exec(command.join(" "));
+    console.info(
+      "LangSmith server is running at http://localhost:1984.  To connect locally, set the following environment variable when running your LangChain application."
+    );
+    console.info("\tLANGCHAIN_TRACING_V2=true");
+  }
+
+  async startAndExpose(ngrokAuthToken: string | null) {
+    const configPath = await createNgrokConfig(ngrokAuthToken);
+    const command = [
+      ...this.dockerComposeCommand,
+      "-f",
+      this.dockerComposeFile,
+      "-f",
+      this.ngrokPath,
+      "up",
+      "--pull=always",
+      "--quiet-pull",
+      "--wait",
+    ];
+    await exec(command.join(" "));
+    console.info(
+      "ngrok is running. You can view the dashboard at http://0.0.0.0:4040"
+    );
+    const ngrokUrl = await getNgrokUrl();
+    console.info(
+      "LangSmith server is running at http://localhost:1984. To connect remotely, set the following environment variable when running your LangChain application."
+    );
+    console.info("\tLANGCHAIN_TRACING_V2=true");
+    console.info(`\tLANGCHAIN_ENDPOINT=${ngrokUrl}`);
+
+    fs.unlinkSync(configPath);
+  }
+
+  async stop() {
+    const command = [
+      ...this.dockerComposeCommand,
+      "-f",
+      this.dockerComposeFile,
+      "-f",
+      this.ngrokPath,
+      "down",
+    ];
+    await exec(command.join(" "));
+  }
+  async status() {
+    const command = [
+      ...this.dockerComposeCommand,
+      "-f",
+      this.dockerComposeFile,
+      "ps",
+      "--format",
+      "json",
+    ];
+    const result = await exec(command.join(" "));
+    const servicesStatus = JSON.parse(result.stdout);
+    if (servicesStatus) {
+      console.info("The LangSmith server is currently running.");
+      await pprintServices(servicesStatus);
+    } else {
+      console.info("The LangSmith server is not running.");
+    }
+  }
+}
+
+const startCommand = new Command("start")
+  .description("Start the LangSmith server")
+  .option(
+    "--expose",
+    "Expose the server to the internet via ngrok (requires ngrok to be installed)"
+  )
+  .option(
+    "--ngrok-authtoken <ngrokAuthtoken>",
+    "Your ngrok auth token. If this is set, --expose is implied."
+  )
+  .option("--dev", "Run the development version of the LangSmith server")
+  .option(
+    "--openai-api-key <openaiApiKey>",
+    "Your OpenAI API key. If this is set, the server will be able to process text and return enhanced plus results."
+  )
+  .action(async (args: string[]) => (await PlusCommand.create()).start(args));
+
+const stopCommand = new Command("stop")
+  .command("stop")
+  .description("Stop the LangSmith server")
+  .action(async () => (await PlusCommand.create()).stop());
+
+const pullCommand = new Command("pull")
+  .command("pull")
+  .description("Pull the latest version of the LangSmith server")
+  .option("--dev", "Pull the development version of the LangSmith server")
+  .action(async (args: string[]) => (await PlusCommand.create()).pull(args));
+
+const statusCommand = new Command("status")
+  .command("status")
+  .description("Get the status of the LangSmith server")
+  .action(async () => (await PlusCommand.create()).status());
+
+program
+  .command("plus")
+  .description("Manage the LangSmith server")
+  .addCommand(startCommand)
+  .addCommand(stopCommand)
+  .addCommand(pullCommand)
+  .addCommand(statusCommand);
+
+program.parse(process.argv);