diff --git a/trustgraph-flow/pyproject.toml b/trustgraph-flow/pyproject.toml
index 6984c478..499aa1c5 100644
--- a/trustgraph-flow/pyproject.toml
+++ b/trustgraph-flow/pyproject.toml
@@ -19,8 +19,6 @@ dependencies = [
     "faiss-cpu",
     "falkordb",
     "fastembed",
-    "google-genai",
-    "google-api-core",
     "ibis",
     "jsonschema",
     "langchain",
@@ -105,7 +103,6 @@ text-completion-azure = "trustgraph.model.text_completion.azure:run"
 text-completion-azure-openai = "trustgraph.model.text_completion.azure_openai:run"
 text-completion-claude = "trustgraph.model.text_completion.claude:run"
 text-completion-cohere = "trustgraph.model.text_completion.cohere:run"
-text-completion-googleaistudio = "trustgraph.model.text_completion.googleaistudio:run"
 text-completion-llamafile = "trustgraph.model.text_completion.llamafile:run"
 text-completion-lmstudio = "trustgraph.model.text_completion.lmstudio:run"
 text-completion-mistral = "trustgraph.model.text_completion.mistral:run"
diff --git a/trustgraph-vertexai/pyproject.toml b/trustgraph-vertexai/pyproject.toml
index e8054b56..48f92207 100644
--- a/trustgraph-vertexai/pyproject.toml
+++ b/trustgraph-vertexai/pyproject.toml
@@ -26,6 +26,7 @@ classifiers = [
 Homepage = "https://github.com/trustgraph-ai/trustgraph"
 
 [project.scripts]
+text-completion-googleaistudio = "trustgraph.model.text_completion.googleaistudio:run"
 text-completion-vertexai = "trustgraph.model.text_completion.vertexai:run"
 
 [tool.setuptools.packages.find]
diff --git a/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/__init__.py b/trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/__init__.py
similarity index 100%
rename from trustgraph-flow/trustgraph/model/text_completion/googleaistudio/__init__.py
rename to trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/__init__.py
diff --git a/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/__main__.py b/trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/__main__.py
similarity index 100%
rename from trustgraph-flow/trustgraph/model/text_completion/googleaistudio/__main__.py
rename to trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/__main__.py
diff --git a/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py b/trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/llm.py
similarity index 100%
rename from trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py
rename to trustgraph-vertexai/trustgraph/model/text_completion/googleaistudio/llm.py