GitLocker: The Coding Marketplace

Description:

gemnine 0.1.13

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()))