Simple Wrapper For The Gemini AI API

This module is designed for asynchronous usage and provides a simple interface to the Google Gemini API.

It also provides a command-line interface for interactive usage.

NOTICE

Images are not tested and may not work as expected.

Installation

pip install -U gemnine

Usage

Command Line

Configuration

gemnine requires a model and an API key to be set.

You can use config file, command parameters or interactive input to set these values.

By default, gemnine will look for files named config.json and session.json in the user config and cache directory.

gemnine -c /path/to/config.json  # Use a config file
gemnine -m "models/gemini-pro"  # Set model, input API key interactively
gemnine -l  # list available models

Parameters

• -c, --config: Path to a JSON file
• -m, --model: Model name
• -s, --session: Session history file
• -V, --version: Show version

Precedence: Interactive input > Command parameters > Config file

Interactive Mode

This mode mimics a chat interface. You can type your message and get a response.

Commands are started with a / and are case-sensitive.

• /exit: Exit the program
• /save <path>: Save the session history to a file
• /load: Load a session history from a file
• /rollback <step>: Rollback to the previous state
• /clear: Clear the session history
• /role: Switch role
• /model: Switch model
• /help: Show help

To embed an image, insert #image(<url>) in your message. Use double # to escape embedding.

await bot.send("""
What are these?
#image(https://example.com/image.png)
#image(file:///path/to/image.png)
#image(base64://<base64-encoded-image>)
""")

All URLs will be downloaded and embedded as base64-encoded images.

API

import asyncio
import gemnine

bot = gemnine.Bot(model="models/gemini-pro", api_key="your-api-key")

async def main():
    response = await bot.send("Hello, how are you?")
    print(response)

    async for r in bot.stream("I'm fine, thank you."):
        print(r, end='')
    
    # `Bot` instance it self doesn't track the session history
    # To keep track of the session history, use `Session` instance
    sess = bot.new_session()

    print(await sess.send("Hello, how are you?"))

    async for r in sess.stream("What was my last question?"):
        print(r, end='')

asyncio.run(main())

Save and load session history

Session and Bot use pydantic models under the hood. You can save and load the session history using model_dump_json and model_validate_json or pass arguments to new_session methods.

import json
from pathlib import Path
sess_path = Path("session.json")

sess = bot.new_session()

data = sess.model_dump_json()

sess_path.write_text(data)

sess = bot.new_session(**json.loads(sess_path.read_text()))