hsliu
e8b9425bab
- Move vllm_omni/core/omni_llm.py to vllm_omni/entrypoints/omni_llm.py - Update all import statements across the codebase to reflect new location - Fix relative imports within the moved file - Maintain functionality while improving code organization - All imports and functionality tested and working correctly This change better reflects that OmniLLM and AsyncOmniLLM are the main entry points for vLLM-omni functionality, rather than core implementation details.
Basic vLLM-omni Examples
This directory contains simple examples demonstrating how to use vLLM-omni for text generation.
Examples
1. simple_usage.py - Direct Library Usage
Demonstrates how to use vLLM-omni directly in your Python code:
# Run the example
python simple_usage.py
Features:
- Synchronous text generation
- Asynchronous text generation
- Multi-prompt processing
- Shows both text output and token information
2. api_client.py - API Server Usage
Demonstrates how to interact with the vLLM-omni API server:
# First, start the server
vllm serve Qwen/Qwen3-0.6B --omni --port 8000
# Then run the client (in another terminal)
python api_client.py
Features:
- Health check
- Text generation via HTTP API
- Multiple prompt types
- Full response structure inspection
Quick Start
-
Install vLLM-omni:
pip install -e . -
Run direct usage example:
cd examples/basic python simple_usage.py -
Run API server example:
# Terminal 1: Start server vllm serve Qwen/Qwen3-0.6B --omni --port 8000 # Terminal 2: Run client cd examples/basic python api_client.py
Expected Output
The examples will show:
- ✅ Generated text responses
- ✅ Token counts and completion status
- ✅ Stage processing information
- ✅ Error handling
Requirements
- Python 3.8+
- vLLM-omni installed
- For API examples: HTTP server running