Fix migrations (#459)

langchain-ai · Feb 21, 2024 · e56ea50 · e56ea50
2 parents bedc1c9 + fef4281
commit e56ea50
Show file tree

Hide file tree

Showing 5 changed files with 61 additions and 5 deletions.
diff --git a/js/package.json b/js/package.json
@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "packageManager": "[email protected]",
   "files": [

diff --git a/js/src/cli/docker-compose.yaml b/js/src/cli/docker-compose.yaml
@@ -29,6 +29,8 @@ services:
         condition: service_healthy
       clickhouse-setup:
         condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
     restart: always
   langchain-queue:
     image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainplus-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
@@ -44,6 +46,8 @@ services:
         condition: service_healthy
       clickhouse-setup:
         condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
     restart: always
   langchain-hub:
     image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainhub-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
@@ -55,7 +59,15 @@ services:
     ports:
       - 1985:1985
     depends_on:
-      - langchain-db
+      langchain-db:
+        condition: service_healthy
+      langchain-redis:
+        condition: service_healthy
+      clickhouse-setup:
+        condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
+    restart: always
   langchain-db:
     image: postgres:14.7
     command:
@@ -121,6 +133,22 @@ services:
         "-c",
         "migrate -source file://clickhouse/migrations -database 'clickhouse://langchain-clickhouse:9000?username=default&password=password&database=default&x-multi-statement=true&x-migrations-table-engine=MergeTree' up",
       ]
+  postgres-setup:
+    image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainplus-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
+    depends_on:
+      langchain-db:
+        condition: service_healthy
+    environment:
+      - LANGCHAIN_ENV=local_docker
+      - LOG_LEVEL=warning
+      - LANGSMITH_LICENSE_KEY=${LANGSMITH_LICENSE_KEY}
+    restart: "on-failure:10"
+    entrypoint:
+      [
+        "bash",
+        "-c",
+        "alembic upgrade head",
+      ]
 volumes:
   langchain-db-data:
   langchain-redis-data:

diff --git a/js/src/index.ts b/js/src/index.ts
@@ -11,4 +11,4 @@ export type {
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 
 // Update using yarn bump-version
-export const __version__ = "0.1.3";
+export const __version__ = "0.1.4";
diff --git a/python/langsmith/cli/docker-compose.yaml b/python/langsmith/cli/docker-compose.yaml
@@ -29,6 +29,8 @@ services:
         condition: service_healthy
       clickhouse-setup:
         condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
     restart: always
   langchain-queue:
     image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainplus-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
@@ -44,6 +46,8 @@ services:
         condition: service_healthy
       clickhouse-setup:
         condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
     restart: always
   langchain-hub:
     image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainhub-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
@@ -55,7 +59,15 @@ services:
     ports:
       - 1985:1985
     depends_on:
-      - langchain-db
+      langchain-db:
+        condition: service_healthy
+      langchain-redis:
+        condition: service_healthy
+      clickhouse-setup:
+        condition: service_completed_successfully
+      postgres-setup:
+        condition: service_completed_successfully
+    restart: always
   langchain-db:
     image: postgres:14.7
     command:
@@ -121,6 +133,22 @@ services:
         "-c",
         "migrate -source file://clickhouse/migrations -database 'clickhouse://langchain-clickhouse:9000?username=default&password=password&database=default&x-multi-statement=true&x-migrations-table-engine=MergeTree' up",
       ]
+  postgres-setup:
+    image: langchain/${_LANGSMITH_IMAGE_PREFIX-}langchainplus-backend:${_LANGSMITH_IMAGE_VERSION:-latest}
+    depends_on:
+      langchain-db:
+        condition: service_healthy
+    environment:
+      - LANGCHAIN_ENV=local_docker
+      - LOG_LEVEL=warning
+      - LANGSMITH_LICENSE_KEY=${LANGSMITH_LICENSE_KEY}
+    restart: "on-failure:10"
+    entrypoint:
+      [
+        "bash",
+        "-c",
+        "alembic upgrade head",
+      ]
 volumes:
   langchain-db-data:
   langchain-redis-data:

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langsmith"
-version = "0.1.4"
+version = "0.1.5"
 description = "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform."
 authors = ["LangChain <[email protected]>"]
 license = "MIT"