This project is a lightweight framework for creating context servers compatible with Zed's Assistant. It allows you to easily define and implement custom slash commands that can be used within Zed's AI-powered assistant.
- Simple decorator-based API for defining slash commands
- Automatic handling of JSON-RPC communication
- Easy integration with Zed's Assistant
- Support for command arguments with type hints
To use Python Context Server, clone this repository and install it:
git clone https://github.com/zed-industries/python-context-server.git
cd python-context-server
pip install .import asyncio
from context_server import ContextServer
ctx = ContextServer()
@ctx.slash_command(name="rot13", description="Perform a rot13 transformation")
@ctx.argument(name="input", type=str, help="String to rot13")
async def rot13(input: str) -> str:
return "".join(
chr((ord(c) - 97 + 13) % 26 + 97) if c.isalpha() else c for c in input.lower()
)
if __name__ == "__main__":
asyncio.run(ctx.run())To use your Python Context Server with Zed, add the following configuration to your Zed settings:
{
"experimental.context_servers": {
"servers": [
{
"id": "python_context_server",
"executable": "python",
"args": ["-m", "your_context_server_module"]
}
]
}
}To quickly try the Rot13 example in this repo, install this project and add the following configuration to your Zed settings:
{
"experimental.context_servers": {
"servers": [
{
"id": "rot13",
"executable": "python",
"args": ["/path/to/repo/examples/rot13.py"]
}
]
}
}Contributions to Python Context Server are welcome! Please feel free to submit issues, fork the repository and send pull requests.
For more details on Context Servers and how they integrate with Zed's Assistant, please refer to the Zed documentation on Context Servers.