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 Ask 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
  • Full-Text Search: Elasticsearch-based full-text search with Chinese analyzer support
  • 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, hybrid search, and Chinese full-text 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

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.29.tar.gz (49.5 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.29-py3-none-any.whl (67.5 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: workspacex-0.1.29.tar.gz
  • Upload date:
  • Size: 49.5 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.29.tar.gz
Algorithm Hash digest
SHA256 d48fa5964eb86581dfbd650049956be513ebd1c492d52cb8bd7fe0543474bdca
MD5 a6df10cff6449f36b33f4984f827baeb
BLAKE2b-256 74a42c8eed2398afcaae423cbc7ea5f1f123ca76a962bda52d95989b4023a6fa

See more details on using hashes here.

File details

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

File metadata

  • Download URL: workspacex-0.1.29-py3-none-any.whl
  • Upload date:
  • Size: 67.5 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.29-py3-none-any.whl
Algorithm Hash digest
SHA256 75a70030759172ec7859d2dfe51ab170f9543f8b9a5486a7e81498ba4c01e3cc
MD5 9686276d1059d4922c7238fdb024b14e
BLAKE2b-256 41694a65b4cd85502e6064e91bf3ea88573667031af4f3d6682e3afb7e947859

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