Skip to content

DSAgent v0.6.2 - API Server & Provider Support

Choose a tag to compare

View all tags
@nmlemus nmlemus released this 11 Jan 00:12
· 43 commits to main since this release
v0.6.2
f2fd83c

What's New

API Server (Phase 8)

DSAgent now includes a full WebSocket and REST API server for integration with web applications, mobile apps, and other systems.

# Start the server
dsagent serve --host 0.0.0.0 --port 8000

# With API key authentication
export DSAGENT_API_KEY=your-secret-key
dsagent serve

New Endpoints

Category Endpoints
Sessions Create, list, get, update, delete, export
Chat Sync and streaming (SSE) chat
Kernel State, variables, execute, reset
Files Upload, download, list, delete
Artifacts List, download, delete generated outputs
WebSocket Real-time bidirectional chat at /ws/chat/{session_id}
Health /health and /health/ready

Full API documentation available at /docs when server is running.

Google/Gemini Support

dsagent init now supports Google as a provider:

dsagent init
# Select "google"
# Default model: gemini/gemini-2.5-flash

Fixes

  • LiteLLM compatibility: Pinned to >=1.70.0,<1.80.0 to avoid Pydantic serialization warnings in newer versions
  • OpenAI API base: Explicitly set default OPENAI_API_BASE for better compatibility across LiteLLM versions
  • Documentation: Clarified automatic model selection in dsagent init

Documentation

  • Added docs/SCALING_PROPOSAL.md with production deployment strategies (PostgreSQL, S3, Redis, Kubernetes)
  • Updated CLI and README documentation

Installation

pip install datascience-agent

# With API server support
pip install "datascience-agent[api]"

Full Changelog

New Features:

  • WebSocket server for real-time chat
  • REST API for sessions, chat, kernel, files, artifacts
  • Server-Sent Events (SSE) for streaming responses
  • Optional API key authentication
  • Google/Gemini provider in setup wizard

Improvements:

  • Better LiteLLM version compatibility
  • Clearer documentation for model selection
Assets 2
Loading