update

phidatahq · Nov 27, 2024 · 4809f57 · 4809f57
1 parent 2261a49
commit 4809f57
Show file tree

Hide file tree

Showing 10 changed files with 449 additions and 211 deletions.
diff --git a/cookbook/providers/vertexai/README.md b/cookbook/providers/vertexai/README.md
@@ -1,4 +1,4 @@
-# Google Gemini Cookbook
+# VertexAI Gemini Cookbook
 
 > Note: Fork and clone this repository if needed
 
@@ -9,16 +9,14 @@ python3 -m venv ~/.venvs/aienv
 source ~/.venvs/aienv/bin/activate
 ```
 
-### 2. Export `GOOGLE_API_KEY`
+### 2. Authenticate with Google Cloud
 
-```shell
-export GOOGLE_API_KEY=***
-```
+[Authenticate with Gcloud](https://cloud.google.com/vertex-ai/generative-ai/docs/start/quickstarts/quickstart-multimodal)
 
 ### 3. Install libraries
 
 ```shell
-pip install -U google-generativeai duckduckgo-search yfinance phidata
+pip install -U google-cloud-aiplatform duckduckgo-search yfinance phidata
 ```
 
 ### 4. Run Agent without Tools

diff --git a/cookbook/providers/vertexai/agent_stream.py b/cookbook/providers/vertexai/agent_stream.py
@@ -2,7 +2,7 @@
 
 from typing import Iterator  # noqa
 from phi.agent import Agent, RunResponse  # noqa
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.tools.yfinance import YFinanceTools
 
 agent = Agent(

diff --git a/cookbook/providers/vertexai/basic_stream.py b/cookbook/providers/vertexai/basic_stream.py
@@ -1,6 +1,6 @@
 from typing import Iterator  # noqa
 from phi.agent import Agent, RunResponse  # noqa
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 
 agent = Agent(model=Gemini(id="gemini-1.5-flash"), markdown=True)
 

diff --git a/cookbook/providers/vertexai/data_analyst.py b/cookbook/providers/vertexai/data_analyst.py
@@ -2,7 +2,7 @@
 
 from textwrap import dedent
 from phi.agent import Agent
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.tools.duckdb import DuckDbTools
 
 duckdb_tools = DuckDbTools(create_tables=False, export_tables=False, summarize_tables=False)

diff --git a/cookbook/providers/vertexai/finance_agent.py b/cookbook/providers/vertexai/finance_agent.py
@@ -1,7 +1,7 @@
 """Run `pip install yfinance` to install dependencies."""
 
 from phi.agent import Agent
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.tools.yfinance import YFinanceTools
 
 agent = Agent(

diff --git a/cookbook/providers/vertexai/knowledge.py b/cookbook/providers/vertexai/knowledge.py
@@ -1,7 +1,7 @@
 """Run `pip install duckduckgo-search sqlalchemy pgvector pypdf openai google.generativeai` to install dependencies."""
 
 from phi.agent import Agent
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.knowledge.pdf import PDFUrlKnowledgeBase
 from phi.vectordb.pgvector import PgVector
 

diff --git a/cookbook/providers/vertexai/storage.py b/cookbook/providers/vertexai/storage.py
@@ -1,7 +1,7 @@
 """Run `pip install duckduckgo-search sqlalchemy google.generativeai` to install dependencies."""
 
 from phi.agent import Agent
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.tools.duckduckgo import DuckDuckGo
 from phi.storage.agent.postgres import PgAgentStorage
 

diff --git a/cookbook/providers/vertexai/structured_output.py b/cookbook/providers/vertexai/structured_output.py
@@ -2,7 +2,7 @@
 from rich.pretty import pprint  # noqa
 from pydantic import BaseModel, Field
 from phi.agent import Agent, RunResponse  # noqa
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 
 
 class MovieScript(BaseModel):

diff --git a/cookbook/providers/vertexai/web_search.py b/cookbook/providers/vertexai/web_search.py
@@ -1,7 +1,7 @@
 """Run `pip install duckduckgo-search` to install dependencies."""
 
 from phi.agent import Agent
-from phi.model.google import Gemini
+from phi.model.vertexai import Gemini
 from phi.tools.duckduckgo import DuckDuckGo
 
 agent = Agent(model=Gemini(id="gemini-1.5-flash"), tools=[DuckDuckGo()], show_tool_calls=True, markdown=True)