A Python SDK for interacting with the YandexGPT API.
Project description
YandexGPT Python SDK
Introduction
The YandexGPT Python SDK provides an easy-to-use interface for interacting with the Yandex GPT API. It includes asynchronous methods for sending requests to the Yandex GPT API, handling authentication, and processing responses. This SDK is designed to simplify the integration of Yandex GPT functionalities into Python applications.
Table of Contents
- Introduction
- Features
- Installation using pip
- Local installation using pip
- Usage
- Configuration
- Examples
- Troubleshooting
- Contributors
- License
Features
- Asynchronous API calls to Yandex GPT.
- Easy configuration of API credentials and model parameters.
- Supports multiple GPT models.
- Includes utility for managing API request headers and payload.
The SDK depends on several Python libraries for its operation. These dependencies are specified in the requirements.txt file.
Installation using pip
To install the YandexGPT Python SDK as a package using pip, run the following command:
pip install yandexgpt-python
Local installation using pip
If you want to install the SDK locally (for example, if you want to modify SDK code), you can do so by cloning the repository and running the setup script:
git clone https://github.com/allseeteam/yandexgpt-python.git
cd yandexgpt-python
pip install -e .
Usage
The YandexGPT SDK is designed for asynchronous operation. To use it, instantiate the YandexGPT class with a configuration manager that includes your Yandex Cloud catalog ID, API key/IAM token, and the desired GPT model type. Currently, the supported model types are:
- yandexgpt
- yandexgpt-lite
- summarization
Configuration
Configuration for the YandexGPT SDK involves setting up the model type, catalog ID, and API key/IAM token. This can be done through the YandexGPTConfigManagerBase class or its subclasses, which allows for easy management of these settings either directly or via environment variables (see examples in env directory).
Examples
Using API key for authentication
from yandex_gpt import YandexGPT, YandexGPTConfigManagerForAPIKey
# Setup configuration (input fields may be empty if they are set in environment variables)
config = YandexGPTConfigManagerForAPIKey(model_type="yandexgpt", catalog_id="your_catalog_id", api_key="your_api_key")
# Instantiate YandexGPT
yandex_gpt = YandexGPT(config_manager=config)
# Async function to get completion
async def get_completion():
messages = [{"role": "user", "text": "Hello, world!"}]
completion = await yandex_gpt.get_async_completion(messages=messages)
print(completion)
# Run the async function
import asyncio
asyncio.run(get_completion())
Using IAM token for authentication
from dotenv import load_dotenv
from yandex_gpt import YandexGPT, YandexGPTConfigManagerForAPIKey
# Load environment variables
load_dotenv('../env/.env.iam_token_generation')
# Setup configuration (input fields are empty, because iam_token will be generated from environment variables)
config = YandexGPTConfigManagerForIAMToken()
# Instantiate YandexGPT
yandex_gpt = YandexGPT(config_manager=config)
# Async function to get completion
async def get_completion():
messages = [{"role": "user", "text": "Hello, world!"}]
completion = await yandex_gpt.get_async_completion(messages=messages)
print(completion)
# Run the async function
import asyncio
asyncio.run(get_completion())
Troubleshooting
For any issues related to the configuration or usage of the SDK, ensure that the catalog ID, API key, and model type are correctly set and that the environment variables (if used) are properly configured.
Contributors
This project is developed and maintained by Gregory Matsnev. Contributions are welcome.
License
This project is licensed under the MIT License. For more information, see the LICENSE file in the project's GitHub repository.
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
Hashes for yandexgpt_python-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d1bac876670fac807fb6808ceafff45c89f9013ca5653dba2817f9ef86003bc5 |
|
MD5 | 9458ca96b2141e6211a6814a2f6526b5 |
|
BLAKE2b-256 | b61d7c54d86565e83df3ee67333a08f5b7dce59ffa4c09429ea4cd1608111e26 |