Gemini AI module (using Gemini AI API through requests lib)
Project description
GeminiRequests
GeminiRequests is a Python library (and code example) designed for interacting with Google’s Gemini AI text generation model using the requests library.
This code provides the Gen class, which allows you to store dialogue history, send requests to the API, load/save conversation history, and more.
Note: The last message in the dialogue history must always be from the user when generating text.
Features
-
Easy integration with Gemini AI:
Uses the standardrequestslibrary to send HTTP requests to Google’s Gemini AI API. -
Dialogue history management:
TheGenclass keeps track of conversation history (a list of messages with roles and text), making it highly useful for context-dependent tasks.
Important: Ensure the last message in the history is from the user before callinggenerate(). -
Convenient export and import methods:
export_history(filename)/import_history(filename)import_history_anyway(filename)(attempts to load history and creates a file if it doesn’t exist)
-
API key management:
On the first run, it automatically creates agemini_api_keyfile (in theGeminiRequestsfolder) where you can save your API key.
Installation and Setup
Installing from PyPI
pip install GeminiRequests
GitHub
-
Clone the repository (or copy this code into your project):
git clone https://github.com/Python-shik/GeminiRequests.git
-
Install dependencies:
pip install -r requirements.txt
If you’re only using this module, you can simply install:
pip install requests
Set up the API key:
- On the first run, the script will ask you to enter your Gemini AI API key.
- The key will be saved in the
GeminiRequests/gemini_api_keyfile.
Usage
Assume the file with the Gen class is stored in the GeminiRequests folder. Example:
from GeminiRequests import Gen # Assuming the file is named gen.py
# Create an instance of the class.
# On the first run, you’ll be prompted to enter your API key if the gemini_api_key file is missing.
ai = Gen()
# Add some messages to the dialogue history
ai.history_add(role="user", content="Hello, Gemini! Tell me a joke.")
ai.history_add(role="system", content="For context generation.")
# Ensure the last message in the history is from the user
ai.history_add(role="user", content="What else can you do?")
# Generate a response
response = ai.generate()
print("Response from Gemini AI:", response)
# Save the history
ai.export_history("dialog_history.pkl")
# Clear local history (from memory)
ai.history = []
# Load history again if needed
ai.import_history("dialog_history.pkl")
The Gen Class and Its Methods
-
Gen(history=[], system_instructions=None)
Constructor. On the first run, it creates agemini_api_keyfile if it doesn’t exist.history: a list of dictionaries like{"role": ..., "parts": [{"text": ...}]}, describing the current conversation history.system_instructions: a list of instructions sent as part of the request (essentially "prompts" or "context" from the system).
-
history_add(role, content)
Adds a new message (role and text) to the dialogue history. -
generate()
Sends the dialogue history (and system instructions, if any) to the API and retrieves the generated text.
Important: Ensure the last message in the history is from the user before calling this method. -
export_history(filename)/import_history(filename)
Save/load the dialogue history to/from a file usingpickle. -
import_history_anyway(filename)
Attempts to load the history from a file. If the file doesn’t exist, it creates one and retries. -
clear_history(filename)
Deletes the history file and clears the history from memory.
License
This project is distributed under the MIT License.
You are free to modify and use the code in your projects, including commercial ones.
Author
Developed by Illya Lazarev
Email: sbdt.israel@gmail.com
If you have ideas or suggestions for improvements, feel free to submit pull requests or open an Issue.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file geminirequests-0.1.1.tar.gz.
File metadata
- Download URL: geminirequests-0.1.1.tar.gz
- Upload date:
- Size: 4.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.12.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
63eeed83ccd00d98f37bab5291886c6617a4b0ce998b78ab1b012a1926af60b2
|
|
| MD5 |
9ae292cedbc514c9272dcd3d3476bf5f
|
|
| BLAKE2b-256 |
73311ddceac1d1ce9b304491198794ccea186ec7f625f17add04fd5ab1778658
|
File details
Details for the file GeminiRequests-0.1.1-py3-none-any.whl.
File metadata
- Download URL: GeminiRequests-0.1.1-py3-none-any.whl
- Upload date:
- Size: 6.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.12.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
e98bb4ac17a9399c0c822ffec49ec7f2a3adb3c9a8cbbbbfe7f579d93317e850
|
|
| MD5 |
748d550cf0230c2481cc3365fb134e59
|
|
| BLAKE2b-256 |
5e02faa8df2475247706bdce3ecdba32cbe8d99def91a921f8158d9022af560b
|