cognitive-modules 2.2.1__tar.gz → 2.2.2__tar.gz

{cognitive_modules-2.2.1 → cognitive_modules-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognitive-modules
-Version: 2.2.1
+Version: 2.2.2
 Summary: Structured LLM task runner with schema validation, confidence scoring, and subagent orchestration
 Author: ziel-io
 License: MIT
@@ -59,7 +59,10 @@ Dynamic: license-file
 # Cognitive Modules
 
 [![CI](https://github.com/ziel-io/cognitive-modules/actions/workflows/ci.yml/badge.svg)](https://github.com/ziel-io/cognitive-modules/actions/workflows/ci.yml)
-[![PyPI version](https://badge.fury.io/py/cognitive-modules.svg)](https://pypi.org/project/cognitive-modules/)
+[![npm version](https://img.shields.io/npm/v/cognitive-modules-cli.svg)](https://www.npmjs.com/package/cognitive-modules-cli)
+[![PyPI version](https://img.shields.io/pypi/v/cognitive-modules.svg)](https://pypi.org/project/cognitive-modules/)
+[![npm downloads](https://img.shields.io/npm/dm/cognitive-modules-cli.svg)](https://www.npmjs.com/package/cognitive-modules-cli)
+[![Node.js 18+](https://img.shields.io/badge/node-18+-green.svg)](https://nodejs.org/)
 [![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
@@ -94,13 +97,13 @@ Cognitive Modules is an AI task definition specification designed for generation
 
 | Version | Spec | npm | PyPI | Status |
 |---------|------|-----|------|--------|
-| **v2.2** | v2.2 | `2.2.0` | `2.2.0` | ✅ Stable (recommended) |
+| **v2.2** | v2.2 | `2.2.0` | `2.2.1` | ✅ Stable (recommended) |
 | **v2.5** | v2.5 | `2.5.0-beta.x` | `2.5.0bx` | 🧪 Beta (streaming + multimodal) |
 
 ```bash
 # Install stable v2.2
 npm install cognitive-modules-cli@2.2.0
-pip install cognitive-modules==2.2.0
+pip install cognitive-modules==2.2.1
 
 # Install beta v2.5 (streaming + multimodal)
 npm install cognitive-modules-cli@beta
@@ -122,12 +125,12 @@ npm install -g cognitive-modules-cli@2.2.0
 ### Python (pip)
 
 ```bash
-pip install cognitive-modules==2.2.0
+pip install cognitive-modules==2.2.1
 
 # With LLM support
-pip install "cognitive-modules[openai]==2.2.0"      # OpenAI
-pip install "cognitive-modules[anthropic]==2.2.0"   # Claude
-pip install "cognitive-modules[all]==2.2.0"         # All providers
+pip install "cognitive-modules[openai]==2.2.1"      # OpenAI
+pip install "cognitive-modules[anthropic]==2.2.1"   # Claude
+pip install "cognitive-modules[all]==2.2.1"         # All providers
 ```
 
 | Platform | Package | Command | Features |

{cognitive_modules-2.2.1 → cognitive_modules-2.2.2}/README.md

@@ -1,7 +1,10 @@
 # Cognitive Modules
 
 [![CI](https://github.com/ziel-io/cognitive-modules/actions/workflows/ci.yml/badge.svg)](https://github.com/ziel-io/cognitive-modules/actions/workflows/ci.yml)
-[![PyPI version](https://badge.fury.io/py/cognitive-modules.svg)](https://pypi.org/project/cognitive-modules/)
+[![npm version](https://img.shields.io/npm/v/cognitive-modules-cli.svg)](https://www.npmjs.com/package/cognitive-modules-cli)
+[![PyPI version](https://img.shields.io/pypi/v/cognitive-modules.svg)](https://pypi.org/project/cognitive-modules/)
+[![npm downloads](https://img.shields.io/npm/dm/cognitive-modules-cli.svg)](https://www.npmjs.com/package/cognitive-modules-cli)
+[![Node.js 18+](https://img.shields.io/badge/node-18+-green.svg)](https://nodejs.org/)
 [![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
@@ -36,13 +39,13 @@ Cognitive Modules is an AI task definition specification designed for generation
 
 | Version | Spec | npm | PyPI | Status |
 |---------|------|-----|------|--------|
-| **v2.2** | v2.2 | `2.2.0` | `2.2.0` | ✅ Stable (recommended) |
+| **v2.2** | v2.2 | `2.2.0` | `2.2.1` | ✅ Stable (recommended) |
 | **v2.5** | v2.5 | `2.5.0-beta.x` | `2.5.0bx` | 🧪 Beta (streaming + multimodal) |
 
 ```bash
 # Install stable v2.2
 npm install cognitive-modules-cli@2.2.0
-pip install cognitive-modules==2.2.0
+pip install cognitive-modules==2.2.1
 
 # Install beta v2.5 (streaming + multimodal)
 npm install cognitive-modules-cli@beta
@@ -64,12 +67,12 @@ npm install -g cognitive-modules-cli@2.2.0
 ### Python (pip)
 
 ```bash
-pip install cognitive-modules==2.2.0
+pip install cognitive-modules==2.2.1
 
 # With LLM support
-pip install "cognitive-modules[openai]==2.2.0"      # OpenAI
-pip install "cognitive-modules[anthropic]==2.2.0"   # Claude
-pip install "cognitive-modules[all]==2.2.0"         # All providers
+pip install "cognitive-modules[openai]==2.2.1"      # OpenAI
+pip install "cognitive-modules[anthropic]==2.2.1"   # Claude
+pip install "cognitive-modules[all]==2.2.1"         # All providers
 ```
 
 | Platform | Package | Command | Features |

{cognitive_modules-2.2.1 → cognitive_modules-2.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cognitive-modules"
-version = "2.2.1"
+version = "2.2.2"
 description = "Structured LLM task runner with schema validation, confidence scoring, and subagent orchestration"
 readme = "README.md"
 license = {text = "MIT"}

{cognitive_modules-2.2.1 → cognitive_modules-2.2.2}/src/cognitive/loader.py

@@ -184,16 +184,20 @@ def load_v2_format(module_path: Path) -> dict:
         "require_suggested_mapping": overflow_raw.get("require_suggested_mapping", True)
     }
     
-    enums = manifest.get("enums", {
+    # Merge enums with defaults (ensure defaults are applied even if partial config exists)
+    enums_defaults = {
         "strategy": "extensible" if tier in ("decision", "exploration") else "strict",
         "unknown_tag": "custom"
-    })
+    }
+    enums = {**enums_defaults, **manifest.get("enums", {})}
     
-    compat = manifest.get("compat", {
+    # Merge compat with defaults (ensure defaults are applied even if partial config exists)
+    compat_defaults = {
         "accepts_v21_payload": True,
         "runtime_auto_wrap": True,
         "schema_output_alias": "data"
-    })
+    }
+    compat = {**compat_defaults, **manifest.get("compat", {})}
     
     io_config = manifest.get("io", {})
     tests = manifest.get("tests", [])

{cognitive_modules-2.2.1 → cognitive_modules-2.2.2}/src/cognitive/providers/__init__.py

@@ -1,11 +1,15 @@
 """
 LLM Providers - Unified interface for calling different LLM backends.
+
+Supports both synchronous and streaming modes:
+- call_llm(): Synchronous, returns complete response
+- call_llm_stream(): Streaming, yields response chunks
 """
 
 import json
 import os
 from pathlib import Path
-from typing import Optional
+from typing import Optional, Iterator
 
 
 def call_llm(prompt: str, model: Optional[str] = None) -> str:
@@ -38,6 +42,153 @@ def call_llm(prompt: str, model: Optional[str] = None) -> str:
         return _call_stub(prompt)
 
 
+def call_llm_stream(prompt: str, model: Optional[str] = None) -> Iterator[str]:
+    """
+    Call the configured LLM with streaming output.
+    
+    Yields chunks of the response as they are generated.
+    
+    Args:
+        prompt: The prompt to send
+        model: Optional model override
+    
+    Yields:
+        String chunks of the LLM's response
+    """
+    provider = os.environ.get("LLM_PROVIDER", "stub").lower()
+    
+    if provider == "openai":
+        yield from _stream_openai(prompt, model)
+    elif provider == "anthropic":
+        yield from _stream_anthropic(prompt, model)
+    elif provider == "ollama":
+        yield from _stream_ollama(prompt, model)
+    elif provider == "minimax":
+        yield from _stream_minimax(prompt, model)
+    else:
+        # Stub doesn't support streaming, return full response
+        yield _call_stub(prompt)
+
+
+def _stream_openai(prompt: str, model: Optional[str] = None) -> Iterator[str]:
+    """Stream from OpenAI API."""
+    try:
+        from openai import OpenAI
+    except ImportError:
+        raise ImportError("OpenAI not installed. Run: pip install cognitive[openai]")
+    
+    api_key = os.environ.get("OPENAI_API_KEY")
+    if not api_key:
+        raise ValueError("OPENAI_API_KEY environment variable not set")
+    
+    client = OpenAI(api_key=api_key)
+    model = model or os.environ.get("LLM_MODEL", "gpt-4o")
+    
+    stream = client.chat.completions.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": "You output only valid JSON matching the required schema."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.2,
+        stream=True
+    )
+    
+    for chunk in stream:
+        if chunk.choices and chunk.choices[0].delta.content:
+            yield chunk.choices[0].delta.content
+
+
+def _stream_anthropic(prompt: str, model: Optional[str] = None) -> Iterator[str]:
+    """Stream from Anthropic Claude API."""
+    try:
+        import anthropic
+    except ImportError:
+        raise ImportError("Anthropic not installed. Run: pip install cognitive[anthropic]")
+    
+    api_key = os.environ.get("ANTHROPIC_API_KEY")
+    if not api_key:
+        raise ValueError("ANTHROPIC_API_KEY environment variable not set")
+    
+    client = anthropic.Anthropic(api_key=api_key)
+    model = model or os.environ.get("LLM_MODEL", "claude-sonnet-4-20250514")
+    
+    with client.messages.stream(
+        model=model,
+        max_tokens=8192,
+        system="You output only valid JSON matching the required schema.",
+        messages=[{"role": "user", "content": prompt}]
+    ) as stream:
+        for text in stream.text_stream:
+            yield text
+
+
+def _stream_minimax(prompt: str, model: Optional[str] = None) -> Iterator[str]:
+    """Stream from MiniMax API (OpenAI-compatible)."""
+    try:
+        from openai import OpenAI
+    except ImportError:
+        raise ImportError("OpenAI SDK not installed. Run: pip install openai")
+    
+    api_key = os.environ.get("MINIMAX_API_KEY")
+    if not api_key:
+        raise ValueError("MINIMAX_API_KEY environment variable not set")
+    
+    client = OpenAI(
+        api_key=api_key,
+        base_url="https://api.minimax.chat/v1"
+    )
+    model = model or os.environ.get("LLM_MODEL", "MiniMax-Text-01")
+    
+    stream = client.chat.completions.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": "You output only valid JSON matching the required schema."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.2,
+        stream=True
+    )
+    
+    for chunk in stream:
+        if chunk.choices and chunk.choices[0].delta.content:
+            yield chunk.choices[0].delta.content
+
+
+def _stream_ollama(prompt: str, model: Optional[str] = None) -> Iterator[str]:
+    """Stream from local Ollama instance."""
+    try:
+        import requests
+    except ImportError:
+        raise ImportError("Requests not installed. Run: pip install cognitive[ollama]")
+    
+    host = os.environ.get("OLLAMA_HOST", "http://localhost:11434")
+    model = model or os.environ.get("LLM_MODEL", "llama3.1")
+    
+    response = requests.post(
+        f"{host}/api/generate",
+        json={
+            "model": model,
+            "prompt": prompt,
+            "stream": True,
+            "format": "json",
+            "options": {"temperature": 0.2}
+        },
+        stream=True
+    )
+    response.raise_for_status()
+    
+    for line in response.iter_lines():
+        if line:
+            try:
+                data = json.loads(line)
+                if "response" in data:
+                    yield data["response"]
+            except json.JSONDecodeError:
+                # Skip malformed JSON lines
+                continue
+
+
 def _call_openai(prompt: str, model: Optional[str] = None) -> str:
     """Call OpenAI API."""
     try: