Skip to main content

Unified interface for LLM, TTS, and STT providers

Project description

🧠 AI Services Modules

This repository provides a unified interface to interact with multiple Large Language Model (LLM) providers for tasks like:

  • 💬 Text Generation & Chat Completion
  • 🗣️ Text-to-Speech (TTS)
  • 🎧 Speech-to-Text (STT)
  • 📊 Evaluation via prompt templates

📌 Features

  • 🔄 Unified API across major providers:
    • LLM: OpenAI, Gemini, Groq, Anthropic
    • TTS: OpenAI, GTTS, Gemini
    • STT: OpenAI
  • ✅ Easy integration of multiple providers
  • 🎯 Evaluation prompt templates


🔧 Quick Start via GitHub (No Need to Clone)

You can install the package directly from GitHub using pip:

pip install git+https://github.com/guvi-research/guvi-ai-service.git

✅ Steps to Use

  1. Create a new project folder and inside it, add a .env file with your API keys:
OPENAI_API_KEY=your_openai_key_here
GEMINI_API_KEY=your_gemini_key_here
GROQ_API_KEY=your_groq_key_here
ANTHROPIC_API_KEY=your_anthropic_key_here
MONGODB_URI=your_mongo_uri_here
  1. Create a test.py file like this:
from dotenv import load_dotenv
from ai_services.utils.mongo_logger import MongoEventLogger
from ai_services.client import LLMClient
from ai_services.provider_factory import get_provider
import os

load_dotenv()

# Setup logger
mongo_logger = MongoEventLogger(
    db_uri=os.getenv("MONGODB_URI"),
    db_name="AI_logs",
    collection_name="chat_completion_logs"
)

# Messages for the model
messages = [
    {"role": "system", "content": "You are a interview assistant."},
    {"role": "user", "content": "give me a single tip for peace"}
]

# Get provider
provider = get_provider(
    name="openai",
    model="gpt-4.1-nano",
    event_logger=mongo_logger
)

# Initialize client
client = LLMClient(provider)
response = client.chat(messages)
print("Response:", response)
  1. Run the file:
python test.py

🔌 Supported Providers

Task Providers
Text Generation OpenAI, Groq, Gemini, Anthropic
Text-to-Speech (TTS) OpenAI, GTTS, Gemini
Speech-to-Text (STT) OpenAI

Use provider_type as:

  • llm → for text generation
  • tts → for text-to-speech
  • stt → for speech-to-text

🧾 MongoDB Event Logging (Optional)

This project supports logging API events (LLM, TTS, STT) into MongoDB for analytics, debugging, or auditing.

Mongo Log Structure:

Each interaction stores:

{
  "timestamp": "2025-06-19T10:45:12Z",
  "service_type": "TTS",
  "module": "openai_provider",
  "function": "generate_speech",
  "request_data": { "text": "Hello world" },
  "response_data": { "status": "audio_generated" },
  "status": "success",
  "error": null
}

.env Setup:

MONGODB_URI=mongodb+srv://<user>:<pass>@cluster.mongodb.net/

MongoDB Integration:

from utils.mongo_logger import MongoEventLogger

mongo_logger = MongoEventLogger(
    db_uri=os.getenv("MONGODB_URI"),
    db_name="AI_Interview",
    collection_name="tts_logs"
)

tts_provider = get_provider(
    name="openai",
    provider_type="tts",
    model="gpt-4o-mini-tts",
    voice="alloy",
    event_logger=mongo_logger
)

🚀 How to Run by cloning the repo

HOW TO CLONE AI SERVICE REPO:

git clone "https://github.com/guvi-research/guvi-ai-service.git"

🛠 Prerequisites

  • Python 3.8+

  • API Keys (store in .env file):

    OPENAI_API_KEY=your_openai_key_here
    GEMINI_API_KEY=your_gemini_key_here
    GROQ_API_KEY=your_groq_key_here
    ANTHROPIC_API_KEY=your_anthropic_key_here
    MONGODB_URI=your_uri_link_here
    
  • Install requirements:

    pip install -r requirements.txt
    

1️⃣ Text Generation (LLM)

File: examples/llm_usage.py

from llm_module.client import LLMClient
from llm_module.provider_factory import get_provider

uri = os.getenv("MONGODB_URI")
mongo_logger = MongoEventLogger(
    db_uri=uri,
    db_name = "AI_Services",
    collection_name = "chat_completion_logs"
)

messages = [
  {"role": "system", "content": "You are a helpful assistant."},
  {"role": "user", "content": "What is your role?"}
]

provider = get_provider(
    name="gemini",  
    model="gemini-2.0-flash",
    event_logger=mongo_logger
)
client = LLMClient(provider)

response = client.chat(messages)
print("Response:", response)

Run:

python examples/llm_usage.py

2️⃣ Text-to-Speech (TTS)

File: examples/tts_usage.py

from tts_module.client import TTSClient
from tts_module.provider_factory import get_provider

uri = os.getenv("MONGODB_URI")
mongo_logger = MongoEventLogger(
    db_uri=uri,
    db_name = "AI_Services",
    collection_name = "tts_logs"
)

tts_provider = get_provider("openai", provider_type="tts",  model="gpt-4o-mini-tts", voice="alloy", event_logger=mongo_logger)

client = TTSClient(tts_provider)
client.speak("Hello, this is a text-to-speech test.")

Run:

python examples/tts_usage.py

3️⃣ Speech-to-Text (STT)

File: examples/stt_usage.py

from stt_module.client import STTClient
from stt_module.provider_factory import get_provider

uri = os.getenv("MONGODB_URI")
mongo_logger = MongoEventLogger(
    db_uri=uri,
    db_name = "AI_Services",
    collection_name = "stt_logs"
)

audio_path = "path/to/audio.mp3"

stt_provider = get_provider("openai", provider_type="stt", model="gpt-4o-mini-transcribe",event_logger=mongo_logger)
client = STTClient(stt_provider)

result = client.transcribe(audio_path)
print("Transcription:", result)

Run:

python examples/stt_usage.py

✅ Notes

  • Ensure .env is set before running any script.
  • Choose the correct provider and model as required.
  • All logs are automatically stored in MongoDB (if event_logger is passed).

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

ai_service_modules-0.1.2.tar.gz (17.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ai_service_modules-0.1.2-py3-none-any.whl (23.2 kB view details)

Uploaded Python 3

File details

Details for the file ai_service_modules-0.1.2.tar.gz.

File metadata

  • Download URL: ai_service_modules-0.1.2.tar.gz
  • Upload date:
  • Size: 17.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.9

File hashes

Hashes for ai_service_modules-0.1.2.tar.gz
Algorithm Hash digest
SHA256 fd856a495daeaeef859c1e4962cecd248d36a17e64cc62b7b50c55e9a60e3462
MD5 5cf9f92e96939558d6ee35bdb4ee020a
BLAKE2b-256 da5c483787a27a07fb7c19db3cbe5363a642658646496a226f63b01aa18d010b

See more details on using hashes here.

File details

Details for the file ai_service_modules-0.1.2-py3-none-any.whl.

File metadata

File hashes

Hashes for ai_service_modules-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 747458cb27b8a2cb0d40f238799f48ffba4072ec6818c81d1cf848365a87ae21
MD5 800cd61480e85857aa1e3edd75f9bd5c
BLAKE2b-256 05575b9abf8e69e11ca2509d5c1daba5b0c77ad5d632ac9a097d477f846e0bc9

See more details on using hashes here.

Supported by

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