Native async WebSocket server for the smooth-operator protocol — parity with the Rust and C# reference servers, consuming the in-process smooai-smooth-operator-core engine.
Project description
smooai-smooth-operator-server
Wiring a chat loop is a weekend project. A production agent server is not.
Sessions that survive a reconnect. A wire protocol your clients can actually speak. Streaming turns you can watch token by token. Tools the model can call — and hard limits on the ones it must never call. Human-in-the-loop when a tool wants to write.
smooai-smooth-operator-server is that server, async and native to Python. It speaks the smooth-operator wire protocol (spec/) and consumes the in-process smooai-smooth-operator-core engine — each turn runs a SmoothAgent and maps its stream onto stream_token / stream_chunk / eventual_response. It's the Python sibling of the Rust, Go, TypeScript, and C# servers, all speaking the one protocol.
The client lives in
python/src(smooai-smooth-operator). This is the server half.
Spin up a real agent server
python -m smooth_operator_server
# → smooth-operator-server (local flavor, python) listening on ws://127.0.0.1:8787/ws
That's a full agent backend — sessions, streaming turns, tool-calling, citations — on one WebSocket, in-memory, auth off, zero config. Env knobs: SMOOTH_OPERATOR_BIND (default 127.0.0.1:8787), SMOOTH_OPERATOR_SEED_KB=1 for the demo knowledge docs. The gateway is read from SMOOAI_GATEWAY_URL / SMOOAI_GATEWAY_KEY — with no key, send_message returns a clean LLM_UNAVAILABLE error and the rest of the protocol still works.
Set SMOOTH_AGENT_PREAMBLE_MODEL to a fast model id (e.g. groq-gpt-oss-20b) and every streaming turn also runs that small model in parallel, emitting one ephemeral stream_preamble sentence ("what I'm about to do") to cover the reasoning model's time-to-first-token. Same gateway/key as the main turn, capped at 64 output tokens. It is suppressed the moment the real answer starts streaming, is never folded into the reply or persisted, and any failure is swallowed. Unset (the default) ⇒ no extra call, behavior unchanged.
Or embed it in your own async app:
import asyncio
from smooth_operator_server import ServerState, serve
from smooth_operator_server.session_store import InMemorySessionStore
async def main():
state = ServerState(store=InMemorySessionStore(), chat_client=my_openai_client)
server = await serve(state, "127.0.0.1", 0) # port 0 → ephemeral
print(server.ws_url())
# ... drive real streaming turns ...
await server.shutdown() # graceful drain + clean exit
asyncio.run(main())
Extensible — and safe by construction
An agent is only useful when it can do things, and only trustworthy when you can say what it may never do. This server gives you both seams.
Give it your tools. Hand engine Tools to ServerState and they merge with the built-ins for every turn:
from smooth_operator_core import Tool # the engine's tool base
class OpenTicket(Tool):
name = "open_ticket"
description = "Open a support ticket for the current customer."
parameters = {"type": "object", "properties": {"subject": {"type": "string"}}}
async def execute(self, arguments):
return f"ticket opened: {arguments}"
state = ServerState(
store=InMemorySessionStore(),
chat_client=my_client,
tools=[OpenTicket()],
)
Or let it gain tools with no redeploy. The server hosts SEP extensions — out-of-process tool providers discovered at runtime, their ui/confirm prompts bridged into the protocol's confirmation frames for HITL. Gated: an extension contributes tools only if you name it in SMOOTH_EXTENSIONS_ALLOW. Nothing loads by default.
Now declare the lines it can't cross. Install an AgentConfigResolver, and every tool — built-in, yours, or from an extension — flows through the same gates:
- Per-agent allow-list — an agent's
tool_config.enabledToolsrestricts its turn to exactly those tools. Off the list, off the table. - The authLevel gate — a tool declaring
supports_auth_requirement = Trueis blocked at call time on a public agent when taggedadmin, or when taggedend_userand the session isn't identity-verified (its OTP bit, or yourSessionAuthenticatorseam). Fail-closed — an absent authenticator means "not authenticated". - End-user OTP flow — a refused
end_usertool can offer a one-time-code identity flow via theOtpServiceseam; the server never generates, delivers, or validates a code (the host owns generation, delivery, expiry, attempt counting).
You decide what the agent can touch; the runner enforces it.
What it does
| Piece | Module | Mirrors |
|---|---|---|
| WS transport + per-connection loop + single writer | server.py |
Rust server.rs, C# SmoothOperatorWebSocketExtensions |
Frame dispatch (ping / create / get / send_message / cancel) |
dispatcher.py |
C# FrameDispatcher, Rust handler.rs |
| Session + message store | session_store.py |
C# SessionStore, Rust storage adapter |
| Streaming turn (engine → protocol events) | turn_runner.py |
C# TurnRunner, Rust runner.rs |
| Per-agent config (instructions / workflow / persona / tools) | agent_config.py |
monorepo agents schema |
| Conversation-workflow steps + post-turn judge | workflow.py |
monorepo general-agent/workflow.ts |
| SEP extension hosting | extensions.py |
Rust extensions.rs |
| Auth verifier seam (permissive + local HS256 JWT) | auth.py |
C# Auth.cs, Rust verifier seam |
Turn cancellation (the "Stop button"). {"action": "cancel", "requestId": "<the send_message requestId>"} cancels the in-flight turn's asyncio.Task — CancelledError fires at its next await, abandoning the LLM/tool call — and emits the terminal cancelled event (status: 499, echoing the turn's requestId) in place of the eventual_response. One active turn per connection: a second send_message mid-turn is rejected with TURN_IN_PROGRESS, never run concurrently. A cancel with no active turn is a silent no-op. Partial output is discarded — the user's message is persisted at the start of the turn so it stays, the assistant reply only at the end, which the cancellation skips. A client disconnect mid-turn aborts the turn too.
Graceful SIGTERM drain. A shared asyncio.Event cancel switch is the single source of truth for "stop". Each connection loop races "cancel set" vs "next inbound frame" — with the turn dispatch awaited inside the frame branch, so an in-flight turn finishes before the loop exits (a drain lets a turn complete; only a client cancel/disconnect aborts one), then a backplane detach always runs.
Per-agent config + conversation workflows. create_conversation_session carries only an agent UUID, so config is resolved server-side per turn: an agent's instructions become its system prompt; personality / greeting are appended (greeting only on the first turn); a conversation_workflow (goal + ordered steps) renders the current step into the prompt and a cheap post-turn judge advances the pointer when the criteria are met. Parsing is tolerant (malformed → server default, never crashes a session) and the judge is failure-tolerant (any error → stay on the current step). With no resolver installed, behavior is unchanged.
Five languages, one protocol
The same server — same wire protocol, same conformance corpus — exists in five languages. Run it where your stack already lives.
| Language | Server package | Registry |
|---|---|---|
| Python | smooai-smooth-operator-server |
in-repo (this package) |
| Rust | smooai-smooth-operator-server |
crates.io |
| C# / .NET | SmooAI.SmoothOperator.Server |
in-repo |
| TypeScript | @smooai/smooth-operator-server |
in-repo |
| Go | github.com/SmooAI/smooth-operator/go/server |
in-repo |
Every native client — TypeScript, Go, .NET, Python, Rust — connects to any of them unmodified.
Develop
cd python/server
uv sync
uv run --quiet ruff format .
uv run --quiet ruff check .
uv run --quiet pytest -q
Part of the smooth-operator service — Smoo AI's polyglot AI agent service. Don't want to run it yourself? lom.smoo.ai hosts it for you.
License
MIT © 2026 Smoo AI. See LICENSE.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file smooai_smooth_operator_server-1.32.0.tar.gz.
File metadata
- Download URL: smooai_smooth_operator_server-1.32.0.tar.gz
- Upload date:
- Size: 114.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4534b61ef31c4640a428510f5675a87907a515220b72e93fd1c21ebb822bbcf7
|
|
| MD5 |
0cf0dec693eef370c40bf7d66f4bf418
|
|
| BLAKE2b-256 |
5d6f67a8dd10c97ce172d545502bc800b7570346addc735dd3c2889ec83fd6d5
|
File details
Details for the file smooai_smooth_operator_server-1.32.0-py3-none-any.whl.
File metadata
- Download URL: smooai_smooth_operator_server-1.32.0-py3-none-any.whl
- Upload date:
- Size: 59.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b06c0a2360df732e0dfa7504be4df6c5b4ac7231605d977d0bad92f00a0e9d67
|
|
| MD5 |
4de0063c51a9cf16d69e5f4d41f90a22
|
|
| BLAKE2b-256 |
758abda9dcf908a97e59f6b666195e6eae2dae73c73c60aee11c15e41bae9a57
|