Skip to content

LLM Kit - Python Large Language Model Kit for generating data of your choice

License

Notifications You must be signed in to change notification settings

useentropy/llmkit

Repository files navigation

LLMKit - Language Model Kit

LLMKit is a Python package that simplifies the interaction with Language Models (LLMs) and allows you to generate text data with ease.

Features

  • Simple Interface: LLMKit provides a straightforward interface to interact with various language models.

  • Custom Transformers: You can create custom transformers to format input data into prompts and transform LLM-generated text into the desired data format.

  • Model Abstraction: LLMKit abstracts the details of model selection and provider handling.

Installation

You can install LLMKit using pip:

pip install llm-stack-kit

Usage

from llmkit import LLMKit

llm_kit = LLMKit()

input_data = "Translate the following English text to French: 'Hello, World!'"

# Generate text using the default transformers
generated_text = llm_kit.generate_data(input_data)

print("Generated Text:", generated_text)

Configuration

For more advanced usage and customization, you can create your own transformers and specify model providers. Only two model providers are supported for now: "openai" and "cloudflare". You need to provide a .env file with either OPENAI_API_KEY for OpenAI or CF_API_KEY and CF_ACCOUNT_ID for Cloudflare.

See sample custom transformer for how to write a custom transformer.

Documentation

Comprehensive documentation will be made available soon.

Contributing

We welcome contributions! Our contributions guideline will be added soon.

License

LLMKit is distributed under the MIT License. See LICENSE for details.

Contact

If you have any questions or suggestions, feel free to reach out to us at [email protected].

About

LLM Kit - Python Large Language Model Kit for generating data of your choice

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages