Skip to main content

A toolkit for working with OpenAI and Azure OpenAI API

Project description

AiManagerToolkit 🤖

Python License Azure OpenAI

AiManagerToolkit es una librería Python diseñada para simplificar la interacción con la API de Azure OpenAI. Esta herramienta proporciona una forma flexible y eficiente de gestionar conversaciones con modelos de lenguaje, integrar herramientas personalizadas, generar respuestas estructuradas, y manejar funcionalidades de voz y texto, ideal para desarrolladores que buscan aprovechar la potencia de la inteligencia artificial en sus aplicaciones.

Características ✨

  • Soporte para Azure OpenAI: Fácil integración con la plataforma de Azure.
  • Herramientas Personalizadas: Define y registra herramientas para mejorar las interacciones con el modelo.
  • Salidas Estructuradas: Genera respuestas en formato JSON basadas en esquemas definidos.
  • Chat Sincrónico y Streaming: Manejo de conversaciones tanto en modo sincrónico como en streaming.
  • Embeddings: Generación y manejo de embeddings de texto.
  • Speech to Text (STT): Transcripción y traducción de audio a texto.
  • Text to Speech (TTS): Generación de audio a partir de texto.
  • Logging Configurable: Sistema de logging integrado para monitorear y depurar las interacciones.
  • Manejo Avanzado de Mensajes: Nueva clase Message para una gestión eficiente de conversaciones y tipos de contenido.

Instalación 🚀

Puedes instalar AiManagerToolkit desde PyPI utilizando pip:

pip install AiManagerToolkit

Uso Básico 💻

1. Configuración Inicial 🛠️

Configura la conexión a la API de Azure OpenAI utilizando variables de entorno o parámetros en el código.

Configuración utilizando .env 🌐

Crea un archivo .env en el directorio raíz de tu proyecto con las credenciales necesarias:

AZURE_OPENAI_DEPLOYMENT=gpt-4o
AZURE_OPENAI_ENDPOINT=https://tu-endpoint.azure.com/
AZURE_OPENAI_API_KEY=tu-clave-api
AZURE_OPENAI_API_VERSION=2024-06-01
AZURE_OPENAI_EMBEDDINGS_MODEL=text-embedding-3-small

Configuración en el Código 🔧

Puedes pasar la configuración directamente en tu código:

from AiManagerToolkit import AzureAiToolkit

azure_ai = AzureAiToolkit(
    model="gpt-4o",
    azure_endpoint="https://tu-endpoint.azure.com/",
    api_key="tu-clave-api",
    temperature=0.7
)

2. Ejemplo de Uso de Chat 🔄

from AiManagerToolkit import AzureAiToolkit

azure_ai = AzureAiToolkit()

messages = [{"role": "user", "content": "¿Cuál es el estado de mi pedido?"}]
response = azure_ai.chat(messages)
print(response.choices[0].message.content)

3. Generación de Embeddings 📊

embedding = azure_ai.embeddings("Texto para generar embedding")
print(embedding.data[0].embedding)

4. Transcripción de Audio 🎙️

transcription = azure_ai.transcribe("ruta/al/archivo/audio.mp3")
print(transcription)

5. Generación de Voz 🔊

azure_ai.speech("Texto para convertir en voz", output_file_path="salida.mp3")

6. Cálculo de Similitud Coseno 📐

vector1 = [1, 2, 3]
vector2 = [4, 5, 6]
similarity = azure_ai.cosine_similarity(vector1, vector2)
print(f"Similitud coseno: {similarity}")

7. Manejo Avanzado de Mensajes 💬

from AiManagerToolkit import Message, SystemMessage, UserMessage

# Iniciar una conversación
conversation = Message(SystemMessage("Eres un asistente útil."))

# Añadir mensajes a la conversación
conversation.add_message(UserMessage("Hola, ¿cómo estás?"))
conversation.add_message("¿Puedes ayudarme con una tarea?")

# Obtener la conversación completa
full_conversation = conversation.get_full_conversation()

Contribuciones 👥

¡Las contribuciones son bienvenidas! Si deseas contribuir al proyecto, sigue estos pasos:

  1. Realiza un fork del repositorio.
  2. Crea una nueva rama (git checkout -b feature/mi-nueva-funcionalidad).
  3. Realiza tus cambios y haz commit (git commit -am 'Añadir nueva funcionalidad').
  4. Haz push a la rama (git push origin feature/mi-nueva-funcionalidad).
  5. Crea un nuevo Pull Request.

Roadmap 🛤️

  • Mejoras en la documentación con ejemplos avanzados.
  • Añadir más tests unitarios y de integración.
  • Soporte para operaciones avanzadas con Azure OpenAI.
  • Integración con más servicios de Azure AI.

Licencia 📄

Este proyecto está licenciado bajo la Licencia MIT - consulta el archivo LICENSE para más detalles.


¡Gracias por usar AiManagerToolkit! Si tienes alguna pregunta o sugerencia, no dudes en abrir un issue en el repositorio. 😊

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aimanagertoolkit-0.1.3.tar.gz (13.6 kB view details)

Uploaded Source

Built Distribution

AiManagerToolkit-0.1.3-py3-none-any.whl (23.9 kB view details)

Uploaded Python 3

File details

Details for the file aimanagertoolkit-0.1.3.tar.gz.

File metadata

  • Download URL: aimanagertoolkit-0.1.3.tar.gz
  • Upload date:
  • Size: 13.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.11.9

File hashes

Hashes for aimanagertoolkit-0.1.3.tar.gz
Algorithm Hash digest
SHA256 410831b80cc39d6f4cf40eee9557e9ce254ce6ded3c82ac5983deea6b0803a68
MD5 4e13a1a66247e7105bd8cd14a37098f1
BLAKE2b-256 a7082496c8be334a9943ec5ff43b70baf625d962d913b80cdd6dee000d704c12

See more details on using hashes here.

File details

Details for the file AiManagerToolkit-0.1.3-py3-none-any.whl.

File metadata

File hashes

Hashes for AiManagerToolkit-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 ae245f74d9137adfa46a814b357be6f458846bbba956cc8f897b0c890b7f36d2
MD5 9876fccec624a562ade100c34c888897
BLAKE2b-256 f4b5fe1ac64484be39f885b8724830c39bf169395d7f4aefee94969dfba14795

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page