From a7fa827be7d0e64664e4b86cb2d11741b76f808e Mon Sep 17 00:00:00 2001
From: Henry <hzj94@hotmail.com>
Date: Wed, 2 Aug 2023 19:00:06 +0100
Subject: [PATCH] langsmith env variables

---
 docker/.env.example                   |  5 +++++
 packages/server/.env.example          |  5 +++++
 packages/server/src/commands/start.ts | 12 +++++++++++-
 3 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/docker/.env.example b/docker/.env.example
index dd3b00fb..ec13bc0a 100644
--- a/docker/.env.example
+++ b/docker/.env.example
@@ -20,3 +20,8 @@ LOG_PATH=/root/.flowise/logs
 # EXECUTION_MODE=main (child | main)
 # TOOL_FUNCTION_BUILTIN_DEP=crypto,fs
 # TOOL_FUNCTION_EXTERNAL_DEP=moment,lodash
+
+# LANGCHAIN_TRACING_V2=true
+# ANGCHAIN_ENDPOINT=https://api.smith.langchain.com
+# LANGCHAIN_API_KEY=your_api_key
+# LANGCHAIN_PROJECT=your_project
\ No newline at end of file
diff --git a/packages/server/.env.example b/packages/server/.env.example
index 1e10452c..7ab05518 100644
--- a/packages/server/.env.example
+++ b/packages/server/.env.example
@@ -20,3 +20,8 @@ PASSPHRASE=MYPASSPHRASE # Passphrase used to create encryption key
 # EXECUTION_MODE=main (child | main)
 # TOOL_FUNCTION_BUILTIN_DEP=crypto,fs
 # TOOL_FUNCTION_EXTERNAL_DEP=moment,lodash
+
+# LANGCHAIN_TRACING_V2=true
+# LANGCHAIN_ENDPOINT=https://api.smith.langchain.com
+# LANGCHAIN_API_KEY=your_api_key
+# LANGCHAIN_PROJECT=your_project
diff --git a/packages/server/src/commands/start.ts b/packages/server/src/commands/start.ts
index 6c6260b6..71459d17 100644
--- a/packages/server/src/commands/start.ts
+++ b/packages/server/src/commands/start.ts
@@ -35,7 +35,11 @@ export default class Start extends Command {
         DATABASE_HOST: Flags.string(),
         DATABASE_NAME: Flags.string(),
         DATABASE_USER: Flags.string(),
-        DATABASE_PASSWORD: Flags.string()
+        DATABASE_PASSWORD: Flags.string(),
+        LANGCHAIN_TRACING_V2: Flags.string(),
+        LANGCHAIN_ENDPOINT: Flags.string(),
+        LANGCHAIN_API_KEY: Flags.string(),
+        LANGCHAIN_PROJECT: Flags.string()
     }
 
     async stopProcess() {
@@ -99,6 +103,12 @@ export default class Start extends Command {
         if (flags.DATABASE_USER) process.env.DATABASE_USER = flags.DATABASE_USER
         if (flags.DATABASE_PASSWORD) process.env.DATABASE_PASSWORD = flags.DATABASE_PASSWORD
 
+        // Langsmith tracing
+        if (flags.LANGCHAIN_TRACING_V2) process.env.LANGCHAIN_TRACING_V2 = flags.LANGCHAIN_TRACING_V2
+        if (flags.LANGCHAIN_ENDPOINT) process.env.LANGCHAIN_ENDPOINT = flags.LANGCHAIN_ENDPOINT
+        if (flags.LANGCHAIN_API_KEY) process.env.LANGCHAIN_API_KEY = flags.LANGCHAIN_API_KEY
+        if (flags.LANGCHAIN_PROJECT) process.env.LANGCHAIN_PROJECT = flags.LANGCHAIN_PROJECT
+
         await (async () => {
             try {
                 logger.info('Starting Flowise...')