tasks-prompts-chain 0.0.4__tar.gz → 0.1.0__tar.gz

tasks_prompts_chain-0.1.0/PKG-INFO

@@ -0,0 +1,407 @@
+Metadata-Version: 2.4
+Name: tasks_prompts_chain
+Version: 0.1.0
+Summary: A Python library for creating and executing chains of prompts using multiple LLM providers with streaming support and template formatting.
+Project-URL: Homepage, https://github.com/smirfolio/tasks_prompts_chain
+Project-URL: Issues, https://github.com/smirfolio/tasks_prompts_chain/issues
+Author-email: Samir Ben Sghaier <ben.sghaier.samir@gmail.com>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+
+# TasksPromptsChain
+
+A Mini Python library for creating and executing chains of prompts using multiple LLM providers with streaming support and output template formatting.
+
+## Features
+
+- Sequential prompt chain execution
+- Streaming responses
+- Template-based output formatting
+- System prompt support
+- Placeholder replacement between prompts
+- Multiple output formats (JSON, Markdown, CSV, Text)
+- Async/await support
+- Support for multiple LLM providers (OpenAI, Anthropic, Cerebras, etc.)
+- Multi-model support - use different models for different prompts in the chain
+
+## Dependencies
+
+Please install typing-extensions and the SDK for your preferred LLM providers:
+
+For OpenAI:
+```bash
+pip install typing-extensions
+pip install openai
+```
+
+For Anthropic:
+```bash
+pip install typing-extensions
+pip install anthropic
+```
+
+For Cerebras:
+```bash
+pip install typing-extensions
+pip install cerebras
+```
+
+To Install the library: 
+```
+pip install tasks_prompts_chain
+```
+
+## Installation from source code
+
+### For Users required from source gitHub repo
+```bash
+pip install -r requirements/requirements.txt
+```
+
+### For Developers required from source gitHub repo
+```bash
+pip install -r requirements/requirements.txt
+pip install -r requirements/requirements-dev.txt
+```
+
+## Quick Start
+
+```python
+from tasks_prompts_chain import TasksPromptsChain
+from openai import AsyncOpenAI
+from anthropic import AsyncAnthropic
+from cerebras import AsyncCerebras
+
+async def main():
+    # Initialize the chain with multiple LLM configurations
+    llm_configs = [
+        {
+            "llm_id": "gpt",  # Unique identifier for this LLM
+            "llm_class": AsyncOpenAI,  # LLM SDK class
+            "model_options": {
+                "model": "gpt-4o",
+                "api_key": "your-openai-api-key",
+                "temperature": 0.1,
+                "max_tokens": 4120,
+            }
+        },
+        {
+            "llm_id": "claude",  # Unique identifier for this LLM
+            "llm_class": AsyncAnthropic,  # LLM SDK class
+            "model_options": {
+                "model": "claude-3-sonnet-20240229",
+                "api_key": "your-anthropic-api-key",
+                "temperature": 0.1,
+                "max_tokens": 8192,
+            }
+        },
+        {
+            "llm_id": "llama",  # Unique identifier for this LLM
+            "llm_class": AsyncCerebras,  # LLM SDK class
+            "model_options": {
+                "model": "llama-3.3-70b",
+                "api_key": "your-cerebras-api-key",
+                "base_url": "https://api.cerebras.ai/v1",
+                "temperature": 0.1,
+                "max_tokens": 4120,
+            }
+        }
+    ]
+
+    chain = TasksPromptsChain(
+        llm_configs,
+        final_result_placeholder="design_result"
+    )
+
+    # Define your prompts - specify which LLM to use for each prompt
+    prompts = [
+        {
+            "prompt": "Create a design concept for a luxury chocolate bar",
+            "output_format": "TEXT",
+            "output_placeholder": "design_concept",
+            "llm_id": "gpt"  # Use the GPT model for this prompt
+        },
+        {
+            "prompt": "Based on this concept: {{design_concept}}, suggest a color palette",
+            "output_format": "JSON",
+            "output_placeholder": "color_palette",
+            "llm_id": "claude"  # Use the Claude model for this prompt
+        },
+        {
+            "prompt": "Based on the design and colors: {{design_concept}} and {{color_palette}}, suggest packaging materials",
+            "output_format": "MARKDOWN",
+            "output_placeholder": "packaging",
+            "llm_id": "llama"  # Use the Cerebras model for this prompt
+        }
+    ]
+
+    # Stream the responses
+    async for chunk in chain.execute_chain(prompts):
+        print(chunk, end="", flush=True)
+
+    # Get specific results
+    design = chain.get_result("design_concept")
+    colors = chain.get_result("color_palette")
+    packaging = chain.get_result("packaging")
+```
+
+## Advanced Usage
+
+### Using System Prompts
+
+```python
+chain = TasksPromptsChain(
+    llm_configs=[
+        {
+            "llm_id": "default_model",
+            "llm_class": AsyncOpenAI,
+            "model_options": {
+                "model": "gpt-4o",
+                "api_key": "your-openai-api-key",
+                "temperature": 0.1,
+                "max_tokens": 4120,
+            }
+        }
+    ],
+    final_result_placeholder="result",
+    system_prompt="You are a professional design expert specialized in luxury products",
+    system_apply_to_all_prompts=True
+)
+```
+
+### Using Cerebras Models
+
+```python
+from cerebras import AsyncCerebras
+
+llm_configs = [
+    {
+        "llm_id": "cerebras",
+        "llm_class": AsyncCerebras,
+        "model_options": {
+            "model": "llama-3.3-70b",
+            "api_key": "your-cerebras-api-key",
+            "base_url": "https://api.cerebras.ai/v1",
+            "temperature": 0.1,
+            "max_tokens": 4120,
+        }
+    }
+]
+
+chain = TasksPromptsChain(
+    llm_configs,
+    final_result_placeholder="result",
+)
+```
+
+### Custom API Endpoint
+
+```python
+llm_configs = [
+    {
+        "llm_id": "custom_endpoint",
+        "llm_class": AsyncOpenAI,
+        "model_options": {
+            "model": "your-custom-model",
+            "api_key": "your-api-key",
+            "base_url": "https://your-custom-endpoint.com/v1",
+            "temperature": 0.1,
+            "max_tokens": 4120,
+        }
+    }
+]
+
+chain = TasksPromptsChain(
+    llm_configs,
+    final_result_placeholder="result",
+)
+```
+
+### Using Templates
+
+You must call this set method before the execution of the prompting query (chain.execute_chain(prompts))
+
+```python
+# Set output template before execution
+chain.template_output("""
+<result>
+    <design>
+    ### Design Concept:
+    {{design_concept}}
+    </design>
+    
+    <colors>
+    ### Color Palette:
+    {{color_palette}}
+    </colors>
+</result>
+""")
+```
+then retrieves the final result within the template : 
+
+```python
+# print out the final result in the well formated template
+print(chain.get_final_result_within_template())
+```
+
+
+## API Reference
+
+### TasksPromptsChain Class
+
+#### Constructor Parameters
+
+- `llm_configs` (List[Dict]): List of LLM configurations, each containing:
+  - `llm_id` (str): Unique identifier for this LLM configuration
+  - `llm_class`: The LLM class to use (e.g., `AsyncOpenAI`, `AsyncAnthropic`, `AsyncCerebras`)
+  - `model_options` (Dict): Configuration for the LLM:
+    - `model` (str): The model identifier
+    - `api_key` (str): Your API key for the LLM provider
+    - `temperature` (float): Temperature setting for response generation
+    - `max_tokens` (int): Maximum tokens in generated responses
+    - `base_url` (Optional[str]): Custom API endpoint URL
+- `system_prompt` (Optional[str]): System prompt for context
+- `final_result_placeholder` (str): Name for the final result placeholder
+- `system_apply_to_all_prompts` (Optional[bool]): Apply system prompt to all prompts
+
+#### Methods
+
+- `execute_chain(prompts: List[Dict], streamout: bool = True) -> AsyncGenerator[str, None]`
+  - Executes the prompt chain and streams responses
+  
+- `template_output(template: str) -> None`
+  - Sets the output template format
+  
+- `get_final_result_within_template(self) -> Optional[str]`
+  - Retrieves the final query result with the defined template in template_output();
+
+- `get_result(placeholder: str) -> Optional[str]`
+  - Retrieves a specific result by placeholder
+
+### Prompt Format
+
+Each prompt in the chain can be defined as a dictionary:
+```python
+{
+    "prompt": str,              # The actual prompt text
+    "output_format": str,       # "JSON", "MARKDOWN", "CSV", or "TEXT"
+    "output_placeholder": str,  # Identifier for accessing this result
+    "llm_id": str               # Optional: ID of the LLM to use for this prompt
+}
+```
+
+## Supported LLM Providers
+
+TasksPromptsChain currently supports the following LLM providers:
+
+1. **OpenAI** - via `AsyncOpenAI` from the `openai` package
+2. **Anthropic** - via `AsyncAnthropic` from the `anthropic` package
+3. **Cerebras** - via `AsyncCerebras` from the `cerebras` package
+
+Each provider has different capabilities and models. The library adapts the API calls to work with each provider's specific requirements.
+
+## Error Handling
+
+The library includes comprehensive error handling:
+- Template validation
+- API error handling
+- Placeholder validation
+- LLM validation (checks if specified LLM ID exists)
+
+Errors are raised with descriptive messages indicating the specific issue and prompt number where the error occurred.
+
+## Best Practices
+
+1. Always set templates before executing the chain
+2. Use meaningful placeholder names
+3. Handle streaming responses appropriately
+4. Choose appropriate models for different types of tasks
+5. Use system prompts for consistent context
+6. Select the best provider for specific tasks:
+   - OpenAI is great for general purpose applications
+   - Anthropic (Claude) excels at longer contexts and complex reasoning
+   - Cerebras is excellent for high-performance AI tasks
+
+## How You Can Get Involved
+✅ Try out tasks_prompts_chain: Give our software a try in your own setup and let us know how it goes - your experience helps us improve!
+
+✅ Find a bug: Found something that doesn't work quite right? We'd appreciate your help in documenting it so we can fix it together.
+
+✅ Fixing Bugs: Even small code contributions make a big difference! Pick an issue that interests you and share your solution with us.
+
+✅ Share your thoughts: Have an idea that would make this project more useful? We're excited to hear your thoughts and explore new possibilities together!
+
+Your contributions, big or small, truly matter to us. We're grateful for any help you can provide and look forward to welcoming you to our community!
+
+### Developer Contribution Workflow
+1. Fork the Repository: Create your own copy of the project by clicking the "Fork" button on our GitHub repository.
+2. Clone Your Fork: 
+``` bash
+git clone git@github.com:<your-username>/tasks_prompts_chain.git
+cd tasks_prompts_chain/
+```
+3. Set Up Development Environment
+``` bash
+# Create and activate a virtual environment
+python3 -m venv .venv
+source .venv/bin/activate  # On Windows: .venv\Scripts\activate
+
+# Install development dependencies
+pip install -r requirements/requirements-dev.txt
+```
+4. Stay Updated
+```bash
+# Add the upstream repository
+git remote add upstream https://github.com/original-owner/tasks_prompts_chain.git
+
+# Fetch latest changes from upstream
+git fetch upstream
+git merge upstream/main
+```
+#### Making Changes
+1. Create a Feature Branch
+```bash
+git checkout -b feature/your-feature-name
+# or
+git checkout -b bugfix/issue-you-are-fixing
+```
+2. Implement Your Changes
+    - Write tests for your changes when applicable
+    - Ensure existing tests pass with pytest
+    - Follow our code style guidelines
+
+3. Commit Your Changes
+```bash
+git add .
+git commit -m "Your descriptive commit message"
+```
+4. Push to Your Fork
+```bash
+git push origin feature/your-feature-name
+```
+5. Create a Pull Request
+6. Code Review Process
+    - Maintainers will review your PR
+    - Address any requested changes
+    - Once approved, your contribution will be merged!
+
+## Release Notes
+
+### 0.1.0 - Breaking Changes
+
+- **Complete API redesign**: The constructor now requires a list of LLM configurations instead of a single LLM class
+- **Multi-model support**: Use different models for different prompts in the chain
+- **Constructor changes**: Replaced `AsyncLLmAi` and `model_options` with `llm_configs`
+- **New provider support**: Added official support for Cerebras models
+- **Removed dependencies**: No longer directly depends on OpenAI SDK
+- **Prompt configuration**: Added `llm_id` field to prompt dictionaries to specify which LLM to use
+
+Users upgrading from version 0.0.x will need to modify their code to use the new API structure.
+
+## License
+
+MIT License