Skip to main content

workspacex is a Python library for managing AIGC (AI-Generated Content) artifacts. It provides a collaborative workspace environment for handling multiple artifacts with features like version control, update notifications, artifact management, and pluggable storage and embedding backends.

Project description

workspacex

License: MIT DeepWiki

workspacex is a Python library for managing AIGC (AI-Generated Content) artifacts. It provides a collaborative workspace environment for handling multiple artifacts with features like version control, update notifications, artifact management, and pluggable storage and embedding backends.

workspace{width=800px height=400px}

Features

  • Artifact Management: Create, update, and manage different types of artifacts (text, code, novels, etc.)
  • Workspace Organization: Group related artifacts in collaborative workspaces
  • Parallel Processing: 🚀 Subartifacts are processed in parallel for improved performance
  • Storage Backends: Local file system and S3-compatible storage (via s3fs)
  • Embedding Backends: OpenAI-compatible and Ollama embedding support
  • Vector Search: Hybrid search combining semantic and keyword-based search
  • Reranking: Local reranking using Qwen3-Reranker models
  • HTTP Service: FastAPI-based reranking service

Process

img.png{width=400px height=800px}


Installation

Basic Installation

pip install workspacex

With Reranker Support

pip install "workspacex[reranker]"  # For using reranker in your code
pip install "workspacex[reranker-server]"  # For running the reranker HTTP service

Using Poetry:

poetry install --extras "reranker-server"  # Installs all features

Usage

Basic Example

import asyncio
from workspacex.utils.logger import logger

from workspacex import WorkSpace, ArtifactType

if __name__ == '__main__':
    workspace = WorkSpace.from_local_storages(workspace_id="demo")
    asyncio.run(workspace.create_artifact(ArtifactType.TEXT, "artifact_001"))

Parallel Processing Demo

WorkspaceX now supports high-performance parallel processing of artifacts and subartifacts, providing significant performance improvements:

Key Features:

  • 🚀 Full Parallel Processing: Main artifacts and subartifacts processed simultaneously
  • Thread Pool Optimization: CPU-intensive operations moved to thread pool
  • 🎯 Configurable Concurrency: Control concurrent operations with max_concurrent_embeddings
  • 🛡️ Error Handling: Robust error handling with detailed logging
  • 📊 Performance Monitoring: Real-time performance metrics and logging
import asyncio
from workspacex import WorkSpace, ArtifactType

async def demo_enhanced_parallel_processing():
    workspace = WorkSpace(workspace_id="parallel_demo", clear_existing=True)
    
    # Configure concurrency limits (optional)
    workspace.workspace_config.max_concurrent_embeddings = 10
    
    # Create an artifact with multiple subartifacts
    # All artifacts and subartifacts will be processed in parallel for maximum performance
    await workspace.create_artifact(
        artifact_type=ArtifactType.NOVEL,
        novel_file_path="path/to/novel.txt",
        embedding_flag=True  # Enables parallel embedding processing
    )

# Run the demo
asyncio.run(demo_enhanced_parallel_processing())

Performance Improvements:

  • Sequential Processing: ~1.0x baseline
  • Parallel Subartifacts Only: ~2-3x faster
  • Full Parallel Processing: ~5-10x faster
  • Batch Processing: ~10-20x faster

For a complete performance comparison demo, see src/examples/parallel_processing_example.py.

More Examples

For more detailed examples on features like reranking, storage/embedding backends, and hybrid search, please refer to the scripts in the src/examples/ directory.

To run an example:

export PYTHONPATH=src
python src/examples/embeddings/openai_example.py

Running the Reranker Server[Optional]

  1. Install server dependencies:
pip install "workspacex[reranker-server]"
  1. Start the server:
python -m workspacex.reranker.server.reranker_server

Default model: Qwen/Qwen3-Reranker-0.6B

To download the model first:

# Install huggingface_hub
pip install -U huggingface_hub

# Set mirror for faster download in China
export HF_ENDPOINT=https://hf-mirror.com

# Download the model
huggingface-cli download --resume-download Qwen/Qwen3-Reranker-0.6B --local-dir Qwen/Qwen3-Reranker-0.6B

The server can be configured with these environment variables:

RERANKER_MODEL=Qwen/Qwen3-Reranker-0.6B  # or Qwen/Qwen3-Reranker-8B
RERANKER_PORT=8000
RERANKER_RELOAD=False

The server will start on http://localhost:8000. Interactive API docs are available at /docs and /redoc. It provides endpoints like /rerank and a Dify-compatible /dify/rerank.


Changelog

S3 Storage Improvements

  • All files uploaded to S3 now automatically set the correct MIME TYPE (Content-Type), including txt, json, images, etc.
  • Uses automatic type inference based on file extension, no manual specification needed.
  • This ensures files are properly recognized and handled in S3.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

workspacex-0.1.13.tar.gz (36.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

workspacex-0.1.13-py3-none-any.whl (49.9 kB view details)

Uploaded Python 3

File details

Details for the file workspacex-0.1.13.tar.gz.

File metadata

  • Download URL: workspacex-0.1.13.tar.gz
  • Upload date:
  • Size: 36.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.2 CPython/3.12.9 Darwin/24.3.0

File hashes

Hashes for workspacex-0.1.13.tar.gz
Algorithm Hash digest
SHA256 58371c9d45d769deb22c5576e0d9bcf4de89df18e036e850218dacb6e0bdfee0
MD5 1d899456b964de2366352ecb8e11c8f7
BLAKE2b-256 36f81eb610730e626e0e622a628d666c628b45d917dca5ac401d737318f82136

See more details on using hashes here.

File details

Details for the file workspacex-0.1.13-py3-none-any.whl.

File metadata

  • Download URL: workspacex-0.1.13-py3-none-any.whl
  • Upload date:
  • Size: 49.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.2 CPython/3.12.9 Darwin/24.3.0

File hashes

Hashes for workspacex-0.1.13-py3-none-any.whl
Algorithm Hash digest
SHA256 b9186a965f2939f8c48f09685ef9fa52a00565393b252f2504487e80f032b484
MD5 2b827b293f51face643df453ec1c525b
BLAKE2b-256 3b0247b4b8e7473658836ccf8a5d3cb4e8bbcd3d5dcf3584a10203b102a1d56a

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page