Weather Forecast Agent with Pydantic AI & Nebius

This notebook demonstrates how to build a simple weather forecast agent using Pydantic AI and Nebius Token Factory. The agent fetches the current weather forecast for a specified city by searching DuckDuckGo with the meta-llama/Meta-Llama-3.1-70B-Instruct model.

Nebius Token Factory provides access to many state-of-the-art LLM models. Check out the full list of models here.

Visit http://tokenfactory.nebius.com/ and sign up to get an API key.

Step 1: Install Dependencies

Step 2: Enter Your API Key

For security, we’ll prompt you to enter your Nebius API key instead of hardcoding it, since Colab doesn’t natively support .env files.

Step 3: Import Required Modules

We import essential modules from pydantic_ai to work with Nebius Token Factory and DuckDuckGo search.

Step 4: Initializing the Weather Agent

Set up the Nebius AI model and configure the agent to fetch weather forecasts.

Step 5: Running a Weather Forecast Query

Ask the agent for the weather forecast of a city (e.g., Kolkata). You can change the city to any location you’d like!