Preview Flyte 2 for productionHosted on Union.ai
Google Gemini

Google Gemini

flyteplugins-gemini

Flyte SDK (v2)Developer Toolsgemini

This plugin provides integration between Flyte and Google's Gemini API, enabling you to use Flyte tasks as tools for Gemini agents.

Install
pip install flyteplugins-gemini

Quick Start(example, may need adjustment)

pip install flyteplugins-gemini

from flytekit import task, workflow
from flyteplugins.gemini import Agent, function_tool, run_agent

@task
def my_task() -> None:
    function_tool(...)

@workflow
def my_workflow() -> None:
    my_task()

Available Imports (3)

configAgent

A Gemini agent configuration.

extends dataclass — configuration or data structure for plugin setup

from flyteplugins.gemini import Agent

taskfunction_tool

Task for Google Gemini.

from flyteplugins.gemini import function_tool

agentrun_agent

Backend connector for Google Gemini.

from flyteplugins.gemini import run_agent

Dependencies

google-genai

Related Plugins

Package Info

Min Flyte SDK
Modules3

Downloads

Last day56
Last week237
Last month343