From 97b32d815781c39368e8fa2b4d903c9f66aa2c26 Mon Sep 17 00:00:00 2001
From: Chester Curme <chester.curme@gmail.com>
Date: Mon, 9 Sep 2024 16:19:04 -0400
Subject: [PATCH] fix mistral dep and lock

---
 libs/vertexai/poetry.lock    | 9 ++++-----
 libs/vertexai/pyproject.toml | 2 +-
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/libs/vertexai/poetry.lock b/libs/vertexai/poetry.lock
index e6a840eb..677b9e55 100644
--- a/libs/vertexai/poetry.lock
+++ b/libs/vertexai/poetry.lock
@@ -1367,7 +1367,7 @@ subdirectory = "libs/core"
 
 [[package]]
 name = "langchain-mistralai"
-version = "0.1.13"
+version = "0.2.0.dev1"
 description = "An integration package connecting Mistral and LangChain"
 optional = false
 python-versions = ">=3.9,<4.0"
@@ -1377,15 +1377,14 @@ develop = false
 [package.dependencies]
 httpx = ">=0.25.2,<1"
 httpx-sse = ">=0.3.1,<1"
-langchain-core = "^0.3.0.dev"
-pydantic = ">2,<3"
+langchain-core = "^0.3.0.dev4"
 tokenizers = ">=0.15.1,<1"
 
 [package.source]
 type = "git"
 url = "https://github.com/langchain-ai/langchain.git"
 reference = "v0.3rc"
-resolved_reference = "6c8d626d701b9a9e365270d06a80e3a82de19963"
+resolved_reference = "2070d659a06d1919093df56b6a63180f56ef8afa"
 subdirectory = "libs/partners/mistralai"
 
 [[package]]
@@ -2757,4 +2756,4 @@ mistral = ["langchain-mistralai"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<4.0"
-content-hash = "fd6089e49c9de43f254f7643f0a5e6bb631f21f2e1e8558d4d7e82dbdb2de2f9"
+content-hash = "00bdba3e2d330934ca0bd2384a01fdaaa42f3f4eefe33eafede493f946f5f604"
diff --git a/libs/vertexai/pyproject.toml b/libs/vertexai/pyproject.toml
index d0dd0c2b..380a97d1 100644
--- a/libs/vertexai/pyproject.toml
+++ b/libs/vertexai/pyproject.toml
@@ -17,7 +17,7 @@ google-cloud-aiplatform = "^1.56.0"
 google-cloud-storage = "^2.17.0"
 # optional dependencies
 anthropic = { extras = ["vertexai"], version = ">=0.30.0,<1", optional = true }
-langchain-mistralai = { git = "https://github.com/langchain-ai/langchain.git", subdirectory = "libs/partners/mistralai", optional = true, branch = "v0.3rc"}
+langchain-mistralai = { version = "^0.2.0.dev1", allow-prereleases = true }
 httpx = "^0.27.0"
 httpx-sse = "^0.4.0"
 pydantic = ">=2,<3"