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.1.tar.gz (17.2 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.1-py3-none-any.whl (23.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ai_service_modules-0.1.1.tar.gz
  • Upload date:
  • Size: 17.2 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.1.tar.gz
Algorithm Hash digest
SHA256 74ce4d770d33eb9510b4d20b848904bcb921db2896c0929883db6366a52b5593
MD5 2acba3eb4c3124dca31d136f55e9e8f4
BLAKE2b-256 24f4b394ddd6f67faafee67027bbdaa15ed80aa437972f0ccd377ef8921cf83f

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ai_service_modules-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 560d4037683cbe152f5bcf16cbc6a155c0c3b2c6e84d55804e1abf71dd5ea849
MD5 9c56309e6289f4e86aae71a2dd73ee1c
BLAKE2b-256 51ae3332bd7e18f129483f11fcbdc7c106abc3b51c06939e1ffdb17dd7011b99

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