gsdk 🚀

gsdk (Gemini SDK) is a lightweight, high-performance Python wrapper for the Google Gemini API (built on the modern google-genai). It is designed for production use, offering automatic key rotation, session persistence, and real-time streaming.

✨ Key Features

• 🔑 Smart Key Rotation: Automatically switch between multiple API keys when you hit rate limits (429/403).
• 🌊 Streaming Support: Real-time response generation with automatic history updates.
• 🔄 Configurable Retries: Set custom retry counts and delays for maximum stability.
• 💾 Session Persistence: Built-in support for File and Redis storage.
• ⚙️ Full Flexibility: Pass any generation parameter (temperature, top_p, max_tokens) globally or per request.
• 🔍 Google Search Grounding: Integrated real-time web search capabilities.
• 📁 Media Support: Simplified async file uploads for multimodal tasks.

📦 Installation

pip install gsdk

🚀 Quick Start

Basic Chat

import asyncio
from gsdk import GeminiSDK

async def main():
    sdk = GeminiSDK(api_keys=["YOUR_API_KEY"], model_name="gemini-flash-latest")

    response = await sdk.ask("session_1", "Hello! Who are you?")
    print(f"AI: {response.text}")

asyncio.run(main())

🌊 Real-time Streaming

Perfect for chat interfaces where you want to show text as it is generated.

async def stream_example():
    sdk = GeminiSDK(api_keys=["KEY_1", "KEY_2"])
    
    print("AI: ", end="", flush=True)
    async for chunk in sdk.ask_stream("session_1", "Write a long poem about coding."):
        print(chunk, end="", flush=True)

asyncio.run(stream_example())

🛠 Advanced Usage

1. Production Storage (Redis)

Share session history across multiple workers or servers.

from gsdk.storage import RedisStorage

storage = RedisStorage(host='localhost', port=6379)
sdk = GeminiSDK(api_keys=["..."], storage=storage)

2. Handling Images and Files

# Upload image or document
media = await sdk.media.upload_file("chart.png")

# Multimodal request
response = await sdk.ask("session_2", [media, "Analyze this chart."])
print(response.text)

3. Google Search Grounding

sdk = GeminiSDK(api_keys=["..."], use_search=True)

response = await sdk.ask("news", "What's happening in AI today?")
print(f"Sources used: {response.sources}")

📖 API Reference

GeminiSDK Methods

Configuration Parameters

🤝 Contributing

1. Fork the Project.
2. Create your Feature Branch.
3. Commit your Changes.
4. Push to the Branch.
5. Open a Pull Request.

📜 License

Distributed under the MIT License.

gsdk — Powering the next generation of Gemini applications. 🚀