diff --git a/optillm/__init__.py b/optillm/__init__.py
index c21e426..c17bd9b 100644
--- a/optillm/__init__.py
+++ b/optillm/__init__.py
@@ -1,5 +1,5 @@
 # Version information
-__version__ = "0.3.7"
+__version__ = "0.3.8"
 
 # Import from server module
 from .server import (
diff --git a/pyproject.toml b/pyproject.toml
index 7e8ded1..ece589c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "optillm"
-version = "0.3.7"
+version = "0.3.8"
 description = "An optimizing inference proxy for LLMs."
 readme = "README.md"
 license = "Apache-2.0"
@@ -29,6 +29,7 @@ dependencies = [
     "tiktoken",
     "scikit-learn",
     "litellm",
+    "google-cloud-aiplatform",
     "requests",
     "beautifulsoup4",
     "lxml",
diff --git a/requirements.txt b/requirements.txt
index a849f1c..ee9d7e2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,6 +10,7 @@ azure.identity
 tiktoken
 scikit-learn
 litellm
+google-cloud-aiplatform
 requests
 beautifulsoup4
 lxml