Example

from phi.agent import Agent, RunResponse
from phi.model.nvidia import Nvidia

agent = Agent(model=Nvidia(), markdown=True)

# Get the response in a variable
# run: RunResponse = agent.run("Share a 2 sentence horror story")
# print(run.content)

# Print the response in the terminal
agent.print_response("Share a 2 sentence horror story")

Usage

Get your key from Nvidia here.

1

Create a virtual environment

Open the Terminal and create a python virtual environment.

python3 -m venv ~/.venvs/aienv
source ~/.venvs/aienv/bin/activate
2

Install libraries

pip install -U openai phidata
3

Export `NVIDIA_API_KEY`

export NVIDIA_API_KEY=xxx
4

Run Nvidia Agent

python cookbook/providers/nvidia/basic.py

Information