hejaz-ai-sdk 1.0.0

Python SDK for the AI Gateway assistant API — chat, streaming, knowledge base, image generation, artifacts, and session management.

Hejaz AI Assistant SDK

Python SDK for the Hejaz AI Gateway — chat, streaming, knowledge base, image generation, and more.

Install

pip install hejaz-ai-sdk

Quick Start

from ai_assistant import AssistantClient

async with AssistantClient(api_key="your-key", base_url="https://api.hejaz.com") as client:
    # Simple chat
    response = await client.chat.send("What is Zakat?")
    print(response.content)

    # Streaming
    async for event in client.chat.stream("Explain fasting in Islam"):
        if event.event_type == "text_delta":
            print(event.data, end="", flush=True)

    # Knowledge base search + chat
    response = await client.knowledge.ask(
        "Compare the four madhabs on marriage rules",
        dataset_ids=["islamic-knowledge"]
    )

    # Image generation
    result = await client.images.generate("a beautiful mosque at sunset")

Features

• Streaming SSE — async for event in client.chat.stream() with 73 typed event types
• Knowledge Base — RAG-powered Q&A over your datasets
• Sub-Agents — Parallel explore/task/plan agents
• Image Generation — Gemini + DashScope with auto-routing
• Artifacts — File upload/download with S3 storage
• Sessions — Multi-turn conversation management
• Tools — 15+ built-in tools + MCP integration

Requirements

• Python 3.11+
• httpx (auto-installed)