khazai 0.1.0

Simple AI chat client with memory and multi-model support

khazai

Unofficial Python client for AI chat — simple, lightweight, with memory.

from khazai import Client

ai = Client(system="Kamu adalah asisten yang membantu.")
print(ai(prompt="Halo!", memory=True))

Features

• Simple API — just Client() + ai(prompt="...")
• System prompt — set persona once or override per-call
• Memory — conversation history with automatic summarization & session rotation
• Multiple models — GPT, Claude, Gemini, DeepSeek, Grok, Qwen
• Streaming — real-time token generation
• Auto session — fresh cookies & CSRF on limit (2000 tokens) or rate-limit (429)

Install

pip install khazai

Usage

Quick start

from khazai import Client

ai = Client()
reply = ai(prompt="hello")
print(reply)

With system prompt

ai = Client(system="You are a helpful assistant. Be concise.")

reply = ai(prompt="explain quantum computing")

Memory mode

ai = Client(system="You are a friendly chatbot.")

ai(prompt="my name is Rizky", memory=True)
ai(prompt="what is my name?", memory=True)
# "Your name is Rizky."

Change model

# Short names
ai = Client(model="gpt")       # default
ai = Client(model="claude")
ai = Client(model="gemini")
ai = Client(model="deepseek")
ai = Client(model="grok")

# Override per-call
ai(prompt="hello", model="claude")

# Full model ID
ai = Client(model="openai/gpt-4o-mini")

Available: gpt, claude, gemini, deepseek, grok, qwen.

Streaming

for chunk in ai(prompt="tell me a story", stream=True):
    print(chunk, end="", flush=True)

Per-call system prompt

ai(prompt="who are you?", system="You are a French assistant. Answer in French.")
# "Je suis votre assistant virtuel..."

Stateless (no memory)

reply = ai(prompt="hello")             # memory=False by default
reply = ai(prompt="hello", memory=False)

Each stateless call rotates the session and discards history.

CLI

python -m khazai "your prompt here"

python -m khazai
> hello
Hello! How can I help you?
> /quit

Interactive example

from khazai import Client

ai = Client(system="Kamu adalah customer service yang ramah.")

while True:
    msg = input("> ")
    if msg == "quit":
        break
    print(ai(prompt=msg, memory=True))

Models

Short name	Full ID
gpt	openai/gpt-4o-mini
claude	anthropic/claude-haiku-4-5
gemini	google/gemini-2.0-flash-001
deepseek	deepseek/deepseek-chat-v3-0324
grok	x-ai/grok-3-mini-beta
qwen	qwen/qwen-2.5-72b-instruct

License

MIT