epi-recorder 2.1.2__tar.gz → 2.1.3__tar.gz

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/CHANGELOG.md

@@ -5,6 +5,32 @@ All notable changes to EPI Recorder will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.1.3] - 2026-01-24
+
+### Added
+- **Google Gemini Support**: Automatic interception of Gemini API calls via `patch_gemini()`
+- **`epi chat` command**: Interactive AI-powered querying of evidence files using natural language
+- **google-generativeai dependency**: Gemini AI features work out of the box
+
+### Changed
+- Updated `patch_all()` to include Gemini alongside OpenAI
+- Added 'gemini' to package keywords
+
+### Fixed
+- **Windows Compatibility**: Replaced Unicode emojis in CLI output with ASCII to prevent crashes on legacy terminals
+- **Error Handling**: Improved API error reporting (e.g., Quota Exceeded) with user-friendly UI panels
+- **Deprecation Warnings**: Suppressed `FutureWarning` spam from google-generativeai SDK
+
+## [2.1.2] - 2025-01-17
+
+### Security
+- **Client-Side Verification**: Embedded HTML viewer now verifies Ed25519 signatures offline using JS
+- **Manifest V1.1**: Canonical JSON hashing and public key inclusion
+
+### Changed
+- Updated trust badges in Viewer UI
+- Spec version bump to 1.1-json
+
 ## [2.1.1] - 2025-12-16
 
 ### Added

{epi_recorder-2.1.2/epi_recorder.egg-info → epi_recorder-2.1.3}/PKG-INFO

@@ -1,15 +1,15 @@
 Metadata-Version: 2.4
 Name: epi-recorder
-Version: 2.1.2
-Summary: Evidence Packaged Infrastructure (.epi) for AI Evidence - Record, Verify, and Replay AI runs with automatic PATH setup
+Version: 2.1.3
+Summary: The PDF for AI Evidence - Cryptographic proof of what Autonomous AI Systems actually did.
 Author-email: Mohd Ibrahim Afridi <epitechforworld@outlook.com>
 Maintainer-email: Mohd Ibrahim Afridi <epitechforworld@outlook.com>
 License: Apache-2.0
-Project-URL: Homepage, https://github.com/mohdibrahimaiml/EPI-V2.1.2
-Project-URL: Documentation, https://github.com/mohdibrahimaiml/EPI-V2.1.2#readme
-Project-URL: Repository, https://github.com/mohdibrahimaiml/EPI-V2.1.2
-Project-URL: Issues, https://github.com/mohdibrahimaiml/EPI-V2.1.2/issues
-Keywords: ai,reproducibility,verification,llm,evidence,openai,cryptography,workflow,audit
+Project-URL: Homepage, https://github.com/mohdibrahimaiml/EPI-V2.1.3
+Project-URL: Documentation, https://github.com/mohdibrahimaiml/EPI-V2.1.3#readme
+Project-URL: Repository, https://github.com/mohdibrahimaiml/EPI-V2.1.3
+Project-URL: Issues, https://github.com/mohdibrahimaiml/EPI-V2.1.3/issues
+Keywords: ai,reproducibility,verification,llm,evidence,openai,gemini,cryptography,workflow,audit
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -29,6 +29,7 @@ Requires-Dist: cryptography>=41.0.0
 Requires-Dist: cbor2>=5.6.0
 Requires-Dist: typer[all]>=0.12.0
 Requires-Dist: rich>=13.0.0
+Requires-Dist: google-generativeai>=0.4.0
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
@@ -43,18 +44,18 @@ Dynamic: license-file
 
 # EPI Recorder
 
-### The PDF for AI Evidence™
+### The PDF for AI Evidence
 
-**Record. Verify. Trust.**
+**Cryptographic proof of what Autonomous AI Systems actually did.**
 
-[![PyPI](https://img.shields.io/pypi/v/epi-recorder?color=blue&label=PyPI&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
-[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?logo=python&logoColor=white)](https://www.python.org)
-[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)](LICENSE)
-[![Status](https://img.shields.io/badge/Status-MVP-orange.svg)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
-[![Downloads](https://img.shields.io/pypi/dm/epi-recorder?color=blue&label=Downloads)](https://pypi.org/project/epi-recorder/)
-[![Stars](https://img.shields.io/github/stars/mohdibrahimaiml/EPI-V2.0.0?style=social)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
+[![PyPI](https://img.shields.io/badge/PyPI-v2.1.3-blue?style=flat&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
+[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?style=flat&logo=python&logoColor=white)](https://www.python.org)
+[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg?style=flat)](LICENSE)
+[![Status](https://img.shields.io/badge/Status-v2.1.3-blue?style=flat)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
+[![Downloads](https://img.shields.io/badge/Downloads-3.8k-blue?style=flat)](https://pypi.org/project/epi-recorder/)
+[![Stars](https://img.shields.io/badge/Stars-16-ea4aaa?style=social&logo=github)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
 
-[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.0.0/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.2/blob/main/colab_demo.ipynb)
+[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.1.3/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.3/blob/main/colab_demo.ipynb)
 
 </div>
 
@@ -73,12 +74,12 @@ Dynamic: license-file
 
 **Unix/Mac:**
 ```bash
-curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.sh | sh
+curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.sh | sh
 ```
 
 **Windows:**
 ```powershell
-iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.ps1 -useb | iex
+iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.ps1 -useb | iex
 ```
 
 **Manual (pip):**
@@ -476,21 +477,23 @@ Use `python -m epi_cli` instead (always works). Or run `python -m epi_cli doctor
 
 ---
 
-## 🆕 What's New in v2.1.2
+## 🆕 What's New in v2.1.3
 
 > [!IMPORTANT]
-> **Critical Security Update:** Implemented client-side Ed25519 signature verification in the embedded viewer. The viewer now cryptographically verifies that the standard signature matches the content and the embedded public key, preventing potential tamper attacks.
+> **Gemini Native:** EPI now natively supports Google Gemini. Just run your script, and Gemini calls are captured automatically. Plus, talk to your evidence with `epi chat`.
 
-### 🛡️ Security & Integrity
+### 🤖 Gemini Integration
+- **Automatic Patcher:** Intercepts `google.generativeai` calls without code changes. Logs prompts, responses, and token usage.
+- **Evidence Chat:** New command `epi chat` allows you to query your `.epi` files using Gemini AI. "What happened in this run?"
+- **Error Capture:** Automatically records API errors like Quota Exceeded (429) or Blocked Content.
+
+### 🛡️ Security & Integrity (v2.1.2)
 - **Client-Side Verification:** The HTML viewer now includes a bundled crypto library to verify signatures offline.
-- **Spec Update (v1.1-json):** Manifests now use Canonical JSON hashing and include the `public_key` for self-contained proof.
 - **Trust Badges:** UI now explicitly shows "Verified" (Green), "Unsigned" (Yellow), or "Tampered" (Red).
 
-### ✨ Enhanced CLI Reliability (v2.1.1)
-
-**1. Python Module Fallback**
-- `python -m epi_cli` now works as a 100% reliable alternative
-- Perfect for restricted environments
+### ✨ Enhanced CLI Reliability
+- **Windows Compatibility:** Fixed Unicode issues in CLI for legacy terminals.
+- **Python Module Fallback:** `python -m epi_cli` works reliably everywhere.
 
 **2. Automatic PATH Configuration**
 - Post-install script auto-fixes PATH on Windows

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/README.md

@@ -4,18 +4,18 @@
 
 # EPI Recorder
 
-### The PDF for AI Evidence™
+### The PDF for AI Evidence
 
-**Record. Verify. Trust.**
+**Cryptographic proof of what Autonomous AI Systems actually did.**
 
-[![PyPI](https://img.shields.io/pypi/v/epi-recorder?color=blue&label=PyPI&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
-[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?logo=python&logoColor=white)](https://www.python.org)
-[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)](LICENSE)
-[![Status](https://img.shields.io/badge/Status-MVP-orange.svg)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
-[![Downloads](https://img.shields.io/pypi/dm/epi-recorder?color=blue&label=Downloads)](https://pypi.org/project/epi-recorder/)
-[![Stars](https://img.shields.io/github/stars/mohdibrahimaiml/EPI-V2.0.0?style=social)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
+[![PyPI](https://img.shields.io/badge/PyPI-v2.1.3-blue?style=flat&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
+[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?style=flat&logo=python&logoColor=white)](https://www.python.org)
+[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg?style=flat)](LICENSE)
+[![Status](https://img.shields.io/badge/Status-v2.1.3-blue?style=flat)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
+[![Downloads](https://img.shields.io/badge/Downloads-3.8k-blue?style=flat)](https://pypi.org/project/epi-recorder/)
+[![Stars](https://img.shields.io/badge/Stars-16-ea4aaa?style=social&logo=github)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
 
-[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.0.0/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.2/blob/main/colab_demo.ipynb)
+[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.1.3/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.3/blob/main/colab_demo.ipynb)
 
 </div>
 
@@ -34,12 +34,12 @@
 
 **Unix/Mac:**
 ```bash
-curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.sh | sh
+curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.sh | sh
 ```
 
 **Windows:**
 ```powershell
-iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.ps1 -useb | iex
+iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.ps1 -useb | iex
 ```
 
 **Manual (pip):**
@@ -437,21 +437,23 @@ Use `python -m epi_cli` instead (always works). Or run `python -m epi_cli doctor
 
 ---
 
-## 🆕 What's New in v2.1.2
+## 🆕 What's New in v2.1.3
 
 > [!IMPORTANT]
-> **Critical Security Update:** Implemented client-side Ed25519 signature verification in the embedded viewer. The viewer now cryptographically verifies that the standard signature matches the content and the embedded public key, preventing potential tamper attacks.
+> **Gemini Native:** EPI now natively supports Google Gemini. Just run your script, and Gemini calls are captured automatically. Plus, talk to your evidence with `epi chat`.
 
-### 🛡️ Security & Integrity
+### 🤖 Gemini Integration
+- **Automatic Patcher:** Intercepts `google.generativeai` calls without code changes. Logs prompts, responses, and token usage.
+- **Evidence Chat:** New command `epi chat` allows you to query your `.epi` files using Gemini AI. "What happened in this run?"
+- **Error Capture:** Automatically records API errors like Quota Exceeded (429) or Blocked Content.
+
+### 🛡️ Security & Integrity (v2.1.2)
 - **Client-Side Verification:** The HTML viewer now includes a bundled crypto library to verify signatures offline.
-- **Spec Update (v1.1-json):** Manifests now use Canonical JSON hashing and include the `public_key` for self-contained proof.
 - **Trust Badges:** UI now explicitly shows "Verified" (Green), "Unsigned" (Yellow), or "Tampered" (Red).
 
-### ✨ Enhanced CLI Reliability (v2.1.1)
-
-**1. Python Module Fallback**
-- `python -m epi_cli` now works as a 100% reliable alternative
-- Perfect for restricted environments
+### ✨ Enhanced CLI Reliability
+- **Windows Compatibility:** Fixed Unicode issues in CLI for legacy terminals.
+- **Python Module Fallback:** `python -m epi_cli` works reliably everywhere.
 
 **2. Automatic PATH Configuration**
 - Post-install script auto-fixes PATH on Windows

epi_recorder-2.1.3/epi_cli/chat.py

@@ -0,0 +1,193 @@
+"""
+EPI CLI Chat - Interactive evidence querying with AI.
+
+Allows users to ask natural language questions about their .epi evidence files.
+"""
+
+import json
+import os
+import warnings
+from pathlib import Path
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.prompt import Prompt
+import google.api_core.exceptions
+
+from epi_core.container import EPIContainer
+
+
+console = Console()
+
+
+def load_steps_from_epi(epi_path: Path) -> list:
+    """Load steps from an .epi file."""
+    import tempfile
+    
+    temp_dir = Path(tempfile.mkdtemp())
+    extracted = EPIContainer.unpack(epi_path, temp_dir)
+    
+    steps_file = extracted / "steps.jsonl"
+    if not steps_file.exists():
+        return []
+    
+    steps = []
+    with open(steps_file, 'r', encoding='utf-8') as f:
+        for line in f:
+            if line.strip():
+                steps.append(json.loads(line))
+    
+    return steps
+
+
+def chat(
+    epi_file: Path = typer.Argument(..., help="Path to .epi file to chat with"),
+    model: str = typer.Option("gemini-2.0-flash", "--model", "-m", help="Gemini model to use")
+):
+    """
+    Chat with your evidence file using AI.
+    
+    Ask natural language questions about what happened in your recording.
+    
+    Example:
+        epi chat my_recording.epi
+    """
+    # Resolve path
+    if not epi_file.exists():
+        # Try epi-recordings directory
+        recordings_dir = Path("./epi-recordings")
+        potential_path = recordings_dir / f"{epi_file.stem}.epi"
+        if potential_path.exists():
+            epi_file = potential_path
+        else:
+            console.print(f"[red]Error:[/red] File not found: {epi_file}")
+            raise typer.Exit(1)
+    
+    # Check for API key
+    api_key = os.environ.get("GOOGLE_API_KEY") or os.environ.get("GEMINI_API_KEY")
+    if not api_key:
+        console.print(Panel(
+            "[yellow]No API key found![/yellow]\n\n"
+            "Set your Google AI API key:\n"
+            "  [cyan]set GOOGLE_API_KEY=your-key-here[/cyan]  (Windows)\n"
+            "  [cyan]export GOOGLE_API_KEY=your-key-here[/cyan]  (Mac/Linux)\n\n"
+            "Get a free key at: [link]https://makersuite.google.com/app/apikey[/link]",
+            title="[!] API Key Required",
+            border_style="yellow"
+        ))
+        raise typer.Exit(1)
+    
+    # Load the .epi file
+    console.print(f"\n[dim]Loading evidence from:[/dim] {epi_file}")
+    
+    try:
+        manifest = EPIContainer.read_manifest(epi_file)
+        steps = load_steps_from_epi(epi_file)
+    except Exception as e:
+        console.print(f"[red]Error loading .epi file:[/red] {e}")
+        raise typer.Exit(1)
+    
+    # Initialize Gemini
+    try:
+        import warnings
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            import google.generativeai as genai
+            
+        genai.configure(api_key=api_key)
+        ai_model = genai.GenerativeModel(model)
+    except ImportError:
+        console.print(Panel(
+            "[red]Google Generative AI package not installed![/red]\n\n"
+            "Install it with:\n"
+            "  [cyan]pip install google-generativeai[/cyan]",
+            title="[X] Missing Dependency",
+            border_style="red"
+        ))
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error initializing Gemini:[/red] {e}")
+        raise typer.Exit(1)
+    
+    # Build context
+    context = f"""You are an expert assistant analyzing an EPI evidence recording file.
+    
+The recording contains cryptographically signed, tamper-proof evidence of an AI workflow execution.
+
+Recording metadata:
+- Created: {manifest.created_at}
+- Goal: {manifest.goal or 'Not specified'}
+- Command: {manifest.cli_command or 'Not specified'}
+- Workflow ID: {manifest.workflow_id}
+- Total steps: {len(steps)}
+
+Here are the recorded steps (this is the timeline of events):
+{json.dumps(steps[:50], indent=2, default=str)[:8000]}
+
+When answering questions:
+1. Be specific and cite step indices when relevant
+2. Distinguish between LLM requests, responses, and other events
+3. If asked about security, note that API keys are automatically redacted
+4. Keep answers concise but informative
+"""
+
+    # Start chat session
+    chat_session = ai_model.start_chat(history=[])
+    
+    # Display header
+    console.print()
+    console.print(Panel(
+        f"[bold cyan]EPI Evidence Chat[/bold cyan]\n\n"
+        f"[dim]File:[/dim] {epi_file.name}\n"
+        f"[dim]Steps:[/dim] {len(steps)}\n"
+        f"[dim]Model:[/dim] {model}\n\n"
+        f"Ask questions about this evidence recording.\n"
+        f"Type [yellow]exit[/yellow] or [yellow]quit[/yellow] to end the session.",
+        border_style="cyan"
+    ))
+    console.print()
+    
+    # Chat loop
+    while True:
+        try:
+            question = Prompt.ask("[bold cyan]You[/bold cyan]")
+        except (KeyboardInterrupt, EOFError):
+            console.print("\n[dim]Goodbye![/dim]")
+            break
+        
+        if question.lower() in ('exit', 'quit', 'q'):
+            console.print("[dim]Goodbye![/dim]")
+            break
+        
+        if not question.strip():
+            continue
+        
+        # Send to Gemini with context
+        try:
+            full_prompt = f"{context}\n\nUser question: {question}"
+            response = chat_session.send_message(full_prompt)
+            
+            console.print()
+            console.print("[bold green]AI:[/bold green]")
+            console.print(Markdown(response.text))
+            console.print()
+            
+        except google.api_core.exceptions.ResourceExhausted:
+            console.print(Panel(
+                "[yellow]API Quota Exceeded[/yellow]\n\n"
+                "You have hit the rate limit for the Gemini API (free tier).\n"
+                "Please wait a minute before trying again.",
+                title="[!] Rate Limit",
+                border_style="yellow"
+            ))
+        except google.api_core.exceptions.NotFound:
+            console.print(f"[red]Error:[/red] The model '{model}' was not found. Try using a different model with --model.")
+        except google.api_core.exceptions.InvalidArgument as e:
+            console.print(f"[red]Error:[/red] Invalid argument: {e}")
+        except Exception as e:
+            console.print(f"[red]Error:[/red] {e}")
+            console.print("[dim]Try asking a different question.[/dim]")
+            console.print()

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_cli/main.py

@@ -12,7 +12,9 @@ from epi_cli.keys import generate_default_keypair_if_missing
 # Create Typer app
 app = typer.Typer(
     name="epi",
-    help="""EPI - Evidence Packaged Infrastructure for AI workflows
+    help="""EPI - The PDF for AI Evidence.
+    
+Cryptographic proof of what Autonomous AI Systems actually did.
 
 Commands:
   run    <script.py>        Record, auto-verify and open viewer. (Zero-config)
@@ -120,6 +122,10 @@ app.add_typer(view_app, name="view", help="Open recording in browser (name resol
 from epi_cli.ls import ls as ls_command
 app.command(name="ls", help="List local recordings (./epi-recordings/)")(ls_command)
 
+# NEW: chat command (v2.1.3 - AI-powered evidence querying)
+from epi_cli.chat import chat as chat_command
+app.command(name="chat", help="Chat with your evidence file using AI")(chat_command)
+
 # Phase 1: keys command (for manual key management)
 @app.command()
 def keys(

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_core/__init__.py

@@ -2,7 +2,7 @@
 EPI Core - Core data structures, serialization, and container management.
 """
 
-__version__ = "2.1.1"
+__version__ = "2.1.3"
 
 from epi_core.schemas import ManifestModel, StepModel
 from epi_core.serialize import get_canonical_hash

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_recorder/__init__.py

@@ -4,7 +4,7 @@ EPI Recorder - Runtime interception and workflow capture.
 Python API for recording AI workflows with cryptographic verification.
 """
 
-__version__ = "2.1.1"
+__version__ = "2.1.3"
 
 # Export Python API
 from epi_recorder.api import (

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_recorder/patcher.py

@@ -325,7 +325,113 @@ def _patch_openai_legacy() -> bool:
         return False
 
 
-    return results
+# ==================== Google Gemini Patcher ====================
+
+def patch_gemini() -> bool:
+    """
+    Patch Google Generative AI library to intercept Gemini API calls.
+    
+    Returns:
+        bool: True if patching succeeded, False otherwise
+    """
+    try:
+        import warnings
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            import google.generativeai as genai
+            from google.generativeai.types import GenerateContentResponse
+        
+        # Get the GenerativeModel class
+        GenerativeModel = genai.GenerativeModel
+        
+        # Store original method
+        original_generate_content = GenerativeModel.generate_content
+        
+        @wraps(original_generate_content)
+        def wrapped_generate_content(self, *args, **kwargs):
+            """Wrapped Gemini generate_content with recording."""
+            
+            # Only record if context is active
+            if not is_recording():
+                return original_generate_content(self, *args, **kwargs)
+            
+            context = get_recording_context()
+            start_time = time.time()
+            
+            # Extract prompt from args/kwargs
+            contents = args[0] if args else kwargs.get("contents", "")
+            
+            # Capture request
+            request_data = {
+                "provider": "google",
+                "method": "GenerativeModel.generate_content",
+                "model": getattr(self, '_model_name', getattr(self, 'model_name', 'gemini')),
+                "contents": str(contents)[:2000],  # Truncate long prompts
+                "generation_config": str(kwargs.get("generation_config", {})),
+            }
+            
+            # Log request step
+            context.add_step("llm.request", request_data)
+            
+            # Execute original call
+            try:
+                response = original_generate_content(self, *args, **kwargs)
+                elapsed = time.time() - start_time
+                
+                # Capture response
+                response_text = ""
+                try:
+                    if hasattr(response, 'text'):
+                        response_text = response.text[:2000]  # Truncate long responses
+                    elif hasattr(response, 'parts'):
+                        response_text = str(response.parts)[:2000]
+                except Exception:
+                    response_text = "[Response text extraction failed]"
+                
+                response_data = {
+                    "provider": "google",
+                    "model": getattr(self, '_model_name', getattr(self, 'model_name', 'gemini')),
+                    "response": response_text,
+                    "latency_seconds": round(elapsed, 3)
+                }
+                
+                # Try to get usage info if available
+                try:
+                    if hasattr(response, 'usage_metadata'):
+                        usage = response.usage_metadata
+                        response_data["usage"] = {
+                            "prompt_tokens": getattr(usage, 'prompt_token_count', None),
+                            "completion_tokens": getattr(usage, 'candidates_token_count', None),
+                            "total_tokens": getattr(usage, 'total_token_count', None)
+                        }
+                except Exception:
+                    pass
+                
+                # Log response step
+                context.add_step("llm.response", response_data)
+                
+                return response
+            
+            except Exception as e:
+                # Log error step
+                context.add_step("llm.error", {
+                    "provider": "google",
+                    "error": str(e),
+                    "error_type": type(e).__name__
+                })
+                raise
+        
+        # Apply patch
+        GenerativeModel.generate_content = wrapped_generate_content
+        
+        return True
+    
+    except ImportError:
+        # google-generativeai not installed
+        return False
+    except Exception as e:
+        print(f"Warning: Failed to patch Gemini: {e}")
+        return False
 
 
 def patch_requests() -> bool:
@@ -419,6 +525,9 @@ def patch_all() -> Dict[str, bool]:
     # Patch OpenAI
     results["openai"] = patch_openai()
     
+    # Patch Google Gemini
+    results["gemini"] = patch_gemini()
+    
     # Patch generic requests (covers LangChain, Anthropic, etc.)
     results["requests"] = patch_requests()
     

{epi_recorder-2.1.2 → epi_recorder-2.1.3/epi_recorder.egg-info}/PKG-INFO

@@ -1,15 +1,15 @@
 Metadata-Version: 2.4
 Name: epi-recorder
-Version: 2.1.2
-Summary: Evidence Packaged Infrastructure (.epi) for AI Evidence - Record, Verify, and Replay AI runs with automatic PATH setup
+Version: 2.1.3
+Summary: The PDF for AI Evidence - Cryptographic proof of what Autonomous AI Systems actually did.
 Author-email: Mohd Ibrahim Afridi <epitechforworld@outlook.com>
 Maintainer-email: Mohd Ibrahim Afridi <epitechforworld@outlook.com>
 License: Apache-2.0
-Project-URL: Homepage, https://github.com/mohdibrahimaiml/EPI-V2.1.2
-Project-URL: Documentation, https://github.com/mohdibrahimaiml/EPI-V2.1.2#readme
-Project-URL: Repository, https://github.com/mohdibrahimaiml/EPI-V2.1.2
-Project-URL: Issues, https://github.com/mohdibrahimaiml/EPI-V2.1.2/issues
-Keywords: ai,reproducibility,verification,llm,evidence,openai,cryptography,workflow,audit
+Project-URL: Homepage, https://github.com/mohdibrahimaiml/EPI-V2.1.3
+Project-URL: Documentation, https://github.com/mohdibrahimaiml/EPI-V2.1.3#readme
+Project-URL: Repository, https://github.com/mohdibrahimaiml/EPI-V2.1.3
+Project-URL: Issues, https://github.com/mohdibrahimaiml/EPI-V2.1.3/issues
+Keywords: ai,reproducibility,verification,llm,evidence,openai,gemini,cryptography,workflow,audit
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -29,6 +29,7 @@ Requires-Dist: cryptography>=41.0.0
 Requires-Dist: cbor2>=5.6.0
 Requires-Dist: typer[all]>=0.12.0
 Requires-Dist: rich>=13.0.0
+Requires-Dist: google-generativeai>=0.4.0
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
@@ -43,18 +44,18 @@ Dynamic: license-file
 
 # EPI Recorder
 
-### The PDF for AI Evidence™
+### The PDF for AI Evidence
 
-**Record. Verify. Trust.**
+**Cryptographic proof of what Autonomous AI Systems actually did.**
 
-[![PyPI](https://img.shields.io/pypi/v/epi-recorder?color=blue&label=PyPI&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
-[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?logo=python&logoColor=white)](https://www.python.org)
-[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)](LICENSE)
-[![Status](https://img.shields.io/badge/Status-MVP-orange.svg)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
-[![Downloads](https://img.shields.io/pypi/dm/epi-recorder?color=blue&label=Downloads)](https://pypi.org/project/epi-recorder/)
-[![Stars](https://img.shields.io/github/stars/mohdibrahimaiml/EPI-V2.0.0?style=social)](https://github.com/mohdibrahimaiml/EPI-V2.0.0)
+[![PyPI](https://img.shields.io/badge/PyPI-v2.1.3-blue?style=flat&logo=pypi&logoColor=white)](https://pypi.org/project/epi-recorder/)
+[![Python](https://img.shields.io/badge/Python-3.11+-3776AB?style=flat&logo=python&logoColor=white)](https://www.python.org)
+[![License](https://img.shields.io/badge/License-Apache%202.0-green.svg?style=flat)](LICENSE)
+[![Status](https://img.shields.io/badge/Status-v2.1.3-blue?style=flat)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
+[![Downloads](https://img.shields.io/badge/Downloads-3.8k-blue?style=flat)](https://pypi.org/project/epi-recorder/)
+[![Stars](https://img.shields.io/badge/Stars-16-ea4aaa?style=social&logo=github)](https://github.com/mohdibrahimaiml/EPI-V2.1.3)
 
-[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.0.0/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.2/blob/main/colab_demo.ipynb)
+[**🚀 Quick Start**](#-quick-start-30-seconds) • [**📖 Docs**](https://epilabs.org/docs) • [**💬 Community**](https://github.com/mohdibrahimaiml/EPI-V2.1.3/discussions) • [**🎥 Demo**](https://colab.research.google.com/github/mohdibrahimaiml/EPI-V2.1.3/blob/main/colab_demo.ipynb)
 
 </div>
 
@@ -73,12 +74,12 @@ Dynamic: license-file
 
 **Unix/Mac:**
 ```bash
-curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.sh | sh
+curl -sSL https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.sh | sh
 ```
 
 **Windows:**
 ```powershell
-iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.2/main/scripts/install.ps1 -useb | iex
+iwr https://raw.githubusercontent.com/mohdibrahimaiml/EPI-V2.1.3/main/scripts/install.ps1 -useb | iex
 ```
 
 **Manual (pip):**
@@ -476,21 +477,23 @@ Use `python -m epi_cli` instead (always works). Or run `python -m epi_cli doctor
 
 ---
 
-## 🆕 What's New in v2.1.2
+## 🆕 What's New in v2.1.3
 
 > [!IMPORTANT]
-> **Critical Security Update:** Implemented client-side Ed25519 signature verification in the embedded viewer. The viewer now cryptographically verifies that the standard signature matches the content and the embedded public key, preventing potential tamper attacks.
+> **Gemini Native:** EPI now natively supports Google Gemini. Just run your script, and Gemini calls are captured automatically. Plus, talk to your evidence with `epi chat`.
 
-### 🛡️ Security & Integrity
+### 🤖 Gemini Integration
+- **Automatic Patcher:** Intercepts `google.generativeai` calls without code changes. Logs prompts, responses, and token usage.
+- **Evidence Chat:** New command `epi chat` allows you to query your `.epi` files using Gemini AI. "What happened in this run?"
+- **Error Capture:** Automatically records API errors like Quota Exceeded (429) or Blocked Content.
+
+### 🛡️ Security & Integrity (v2.1.2)
 - **Client-Side Verification:** The HTML viewer now includes a bundled crypto library to verify signatures offline.
-- **Spec Update (v1.1-json):** Manifests now use Canonical JSON hashing and include the `public_key` for self-contained proof.
 - **Trust Badges:** UI now explicitly shows "Verified" (Green), "Unsigned" (Yellow), or "Tampered" (Red).
 
-### ✨ Enhanced CLI Reliability (v2.1.1)
-
-**1. Python Module Fallback**
-- `python -m epi_cli` now works as a 100% reliable alternative
-- Perfect for restricted environments
+### ✨ Enhanced CLI Reliability
+- **Windows Compatibility:** Fixed Unicode issues in CLI for legacy terminals.
+- **Python Module Fallback:** `python -m epi_cli` works reliably everywhere.
 
 **2. Automatic PATH Configuration**
 - Post-install script auto-fixes PATH on Windows

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_recorder.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ pyproject.toml
 setup.py
 epi_cli/__init__.py
 epi_cli/__main__.py
+epi_cli/chat.py
 epi_cli/keys.py
 epi_cli/ls.py
 epi_cli/main.py

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/epi_recorder.egg-info/requires.txt

@@ -3,6 +3,7 @@ cryptography>=41.0.0
 cbor2>=5.6.0
 typer[all]>=0.12.0
 rich>=13.0.0
+google-generativeai>=0.4.0
 
 [dev]
 pytest>=8.0.0

{epi_recorder-2.1.2 → epi_recorder-2.1.3}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "epi-recorder"
-version = "2.1.2"
-description = "Evidence Packaged Infrastructure (.epi) for AI Evidence - Record, Verify, and Replay AI runs with automatic PATH setup"
+version = "2.1.3"
+description = "The PDF for AI Evidence - Cryptographic proof of what Autonomous AI Systems actually did."
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.11"
 license = {text = "Apache-2.0"}
@@ -15,7 +15,7 @@ authors = [
 maintainers = [
     {name = "Mohd Ibrahim Afridi", email = "epitechforworld@outlook.com"}
 ]
-keywords = ["ai", "reproducibility", "verification", "llm", "evidence", "openai", "cryptography", "workflow", "audit"]
+keywords = ["ai", "reproducibility", "verification", "llm", "evidence", "openai", "gemini", "cryptography", "workflow", "audit"]
 classifiers = [
     "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
@@ -36,6 +36,7 @@ dependencies = [
     "cbor2>=5.6.0",
     "typer[all]>=0.12.0",
     "rich>=13.0.0",
+    "google-generativeai>=0.4.0",
 ]
 
 [project.optional-dependencies]
@@ -51,10 +52,10 @@ dev = [
 epi = "epi_cli.main:app"
 
 [project.urls]
-Homepage = "https://github.com/mohdibrahimaiml/EPI-V2.1.2"
-Documentation = "https://github.com/mohdibrahimaiml/EPI-V2.1.2#readme"
-Repository = "https://github.com/mohdibrahimaiml/EPI-V2.1.2"
-Issues = "https://github.com/mohdibrahimaiml/EPI-V2.1.2/issues"
+Homepage = "https://github.com/mohdibrahimaiml/EPI-V2.1.3"
+Documentation = "https://github.com/mohdibrahimaiml/EPI-V2.1.3#readme"
+Repository = "https://github.com/mohdibrahimaiml/EPI-V2.1.3"
+Issues = "https://github.com/mohdibrahimaiml/EPI-V2.1.3/issues"
 
 [tool.setuptools.packages.find]
 where = ["."]