Python SDK

git clone https://github.com/soketlabs/realtime-sdk-python.git
cd realtime-sdk-python
pip install .

import asyncio
import pyaudio
import base64
from loguru import logger
from realtime.client import RealtimeClient
import os
from connect import ConversationHandler, AudioManager, audio_recorder
from dotenv import load_dotenv
load_dotenv()

# Initialize the OpenAI client
client = RealtimeClient(
        url="wss://api.soket.ai/v1/realtime",
        api_key=os.getenv("TENSOR_STUDIO_API_KEY"),  # Ensure your API key is set
        debug=True  # Set to True to enable debugging logs
    )
# Initialize the conversation handler and audio manager
audio_manager = AudioManager()
conversation_handler = ConversationHandler(audio_manager)

# Register the event handlers
client.on("conversation.updated", conversation_handler.on_conversation_updated)
client.on("conversation.item.completed", conversation_handler.on_conversation_item_completed)
client.on("error", conversation_handler.on_error)
client.on("conversation.interrupted", conversation_handler.on_conversation_interrupted)

async def main():
    recorder_task = None
    try:
        await client.connect()
        await client.update_session(
            instructions = '''
        You are Soket bot, a helpful assistant. Please respond clearly and concisely.
                ''',
                turn_detection = {
                    "type": "server_vad",
                    "threshold": 0.2,
                    "prefix_padding_ms": 300,
                    "silence_duration_ms": 1000,
                }
        )
        logger.info("Connected to Realtime API.")
        message_content = [
                {
                    "type": "input_text",
                    "text": "तुम्हारा नाम क्या है?"
                }
            ]
        await client.send_user_message_content(content=message_content)
        logger.info("Message sent.")

        # Start audio recorder as a coroutine
        recorder_task = asyncio.create_task(audio_recorder(client))

        # Keep the connection alive to process events
        await asyncio.Event().wait()

    except Exception as e:
        logger.error(f"An error occurred: {e}")
        raise e

    finally:
        # Disconnect from the API
        await client.disconnect()
        logger.info("Disconnected from Realtime API.")

        # Stop and clean up AudioManager
        await audio_manager.shutdown()

        # Cancel recorder_task if it's still running
        if recorder_task and not recorder_task.done():
            recorder_task.cancel()
            try:
                await recorder_task
            except asyncio.CancelledError:
                logger.info("Recorder task cancelled.")

if __name__ == "__main__":
    asyncio.run(main())