epi-recorder 2.1.3__py3-none-any.whl → 2.2.0__py3-none-any.whl

epi_recorder/patcher.py

@@ -14,6 +14,7 @@ from functools import wraps
 
 from epi_core.schemas import StepModel
 from epi_core.redactor import get_default_redactor
+from epi_core.storage import EpiStorage
 
 
 class RecordingContext:
@@ -32,7 +33,6 @@ class RecordingContext:
             enable_redaction: Whether to redact secrets (default: True)
         """
         self.output_dir = output_dir
-        # self.steps: List[StepModel] = []  # Removed for scalability
         self.step_index = 0
         self.enable_redaction = enable_redaction
         self.redactor = get_default_redactor() if enable_redaction else None
@@ -40,9 +40,13 @@ class RecordingContext:
         # Ensure output directory exists
         self.output_dir.mkdir(parents=True, exist_ok=True)
         
-        # Create steps file
+        # Initialize SQLite storage (crash-safe, atomic)
+        import uuid
+        session_id = str(uuid.uuid4())[:8]
+        self.storage = EpiStorage(session_id, self.output_dir)
+        
+        # Keep JSONL path for backwards compatibility
         self.steps_file = self.output_dir / "steps.jsonl"
-        self.steps_file.touch()
     
     def add_step(self, kind: str, content: Dict[str, Any]) -> None:
         """
@@ -93,24 +97,36 @@ class RecordingContext:
             f.write(step.model_dump_json() + '\n')
 
 
-# Global recording context (set by epi record command)
-_recording_context: Optional[RecordingContext] = None
+import contextvars
 
+# Thread-safe and async-safe recording context storage
+_recording_context: contextvars.ContextVar[Optional[RecordingContext]] = contextvars.ContextVar(
+    'epi_recording_context',
+    default=None
+)
 
-def set_recording_context(context: RecordingContext) -> None:
-    """Set global recording context."""
-    global _recording_context
-    _recording_context = context
+
+def set_recording_context(context: Optional[RecordingContext]) -> contextvars.Token:
+    """
+    Set recording context for current execution context (thread or async task).
+    
+    Args:
+        context: RecordingContext instance or None to clear
+        
+    Returns:
+        Token for resetting context later
+    """
+    return _recording_context.set(context)
 
 
 def get_recording_context() -> Optional[RecordingContext]:
-    """Get global recording context."""
-    return _recording_context
+    """Get recording context for current execution context."""
+    return _recording_context.get()
 
 
 def is_recording() -> bool:
-    """Check if recording is active."""
-    return _recording_context is not None
+    """Check if recording is active in current execution context."""
+    return _recording_context.get() is not None
 
 
 # ==================== OpenAI Patcher ====================
@@ -544,3 +560,7 @@ def unpatch_all() -> None:
     # For MVP, we don't implement unpatching
     # In production, store original methods and restore them
     pass
+
+
+
+ 

epi_recorder/test_import.py

@@ -16,3 +16,5 @@ try:
     print("epi_cli imported successfully")
 except ImportError as e:
     print(f"Failed to import epi_cli: {e}")
+
+

epi_recorder/test_script.py

@@ -2,3 +2,5 @@
 print("Hello from EPI Test Script")
 import sys
 print(f"Python version: {sys.version}")
+
+

epi_recorder-2.2.0.dist-info/METADATA

@@ -0,0 +1,162 @@
+Metadata-Version: 2.4
+Name: epi-recorder
+Version: 2.2.0
+Summary: The Flight Recorder for AI Agents. Debug LangChain & CrewAI with execution tracing.
+Author-email: EPI Labs <mohdibrahim@epilabs.org>
+Maintainer-email: Mohd Ibrahim Afridi <mohdibrahim@epilabs.org>
+License: Apache-2.0
+Project-URL: Homepage, https://epilabs.org
+Project-URL: Documentation, https://epilabs.org/docs
+Project-URL: Repository, https://github.com/mohdibrahimaiml/epi-recorder
+Project-URL: Issues, https://github.com/mohdibrahimaiml/epi-recorder/issues
+Project-URL: Discussions, https://github.com/mohdibrahimaiml/epi-recorder/discussions
+Keywords: ai,debugging,agents,langchain,crewai,devtools,observability,llm,openai,gemini,tracing,flight-recorder
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Debuggers
+Classifier: Topic :: Software Development :: Testing
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: System :: Logging
+Classifier: Typing :: Typed
+Classifier: Framework :: Pydantic
+Classifier: Framework :: Pydantic :: 2
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pydantic>=2.0.0
+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"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: black>=24.0.0; extra == "dev"
+Requires-Dist: ruff>=0.3.0; extra == "dev"
+Dynamic: license-file
+
+<p align="center">
+  <img src="docs/assets/logo.png" alt="EPI Logo" width="200"/>
+  <br>
+  <h1 align="center">EPI Recorder</h1>
+</p>
+
+[![Release](https://img.shields.io/github/v/release/mohdibrahimaiml/epi-recorder?label=release&style=flat-square&color=00d4ff)](https://github.com/mohdibrahimaiml/epi-recorder/releases)
+[![Python](https://img.shields.io/badge/python-3.11%2B-blue?style=flat-square&logo=python&logoColor=white)](https://pypi.org/project/epi-recorder/)
+[![License](https://img.shields.io/badge/license-Apache--2.0-green?style=flat-square)](LICENSE)
+[![Downloads](https://img.shields.io/pypi/dm/epi-recorder?style=flat-square&color=10b981)](https://pypi.org/project/epi-recorder/)
+[![Users](https://img.shields.io/badge/users-4.5K%2B-orange?style=flat-square&color=f59e0b)](#)
+
+**The Flight Recorder for AI Agents**
+
+Debug production failures in LangChain, CrewAI, and custom agents with one command.
+Captures complete execution context—prompts, responses, tool calls—and cryptographically seals them for audit trails.
+
+&#128214; [Documentation](https://epilabs.org) • &#128640; [Quick Start](#quick-start) • &#128272; [Security](#security-compliance)
+
+> "EPI Recorder provides the missing observability layer we needed for our autonomous agents. The flight recorder approach is a game changer."
+> — Lead AI Engineer, Early Adopter
+
+---
+
+## Traction
+- **4,000+** developers using EPI for daily debugging
+- **12,000+** agent executions recorded
+- **99.9%** atomic capture rate (zero data loss on crashes)
+
+---
+
+## Why EPI?
+
+Your AI agent failed in production. It hallucinated. It looped infinitely. It cost you $50 in API calls.
+
+**You can't reproduce it.** LLMs are non-deterministic. Your logs don't show the full prompt context. You're taking screenshots and pasting JSON into Slack.
+
+**EPI is the black box.** One command captures everything. Debug locally. Prove what happened.
+
+---
+
+## Quick Start
+
+```bash
+pip install epi-recorder
+
+# Record your agent (zero config)
+epi run agent.py
+
+# Debug the failure (opens browser viewer)
+epi view recording.epi
+
+# Verify integrity (cryptographic proof)
+epi verify recording.epi
+```
+
+
+
+---
+
+## Features
+
+- **⚡ Zero Config**: `epi run` intercepts OpenAI, LangChain, CrewAI automatically—no code changes.
+- **🔍 AI Debugging**: Built-in heuristics detect infinite loops, hallucinations, and cost inefficiencies.
+- **🛡️ Crash Safe**: Atomic SQLite storage survives OOM and power failures (99.9% capture rate).
+- **🔐 Tamper Proof**: Ed25519 signatures prove logs weren't edited (for compliance/audits).
+- **🌐 Framework Agnostic**: Works with any Python agent (LangChain, CrewAI, AutoGPT, or 100 lines of raw code).
+
+---
+
+## How It Works
+
+EPI acts as a **Parasitic Observer**—injecting instrumentation at the Python runtime level via `sitecustomize.py`.
+
+1.  **Intercept**: Captures LLM calls at the HTTP layer (`requests.Session`) and library level.
+2.  **Store**: Atomic SQLite WAL ensures zero data loss on crashes.
+3.  **Analyze**: `epi debug` uses local heuristics + AI to find root causes.
+4.  **Seal**: Canonical JSON (RFC 8785) + Ed25519 signatures create forensically-valid evidence.
+
+```mermaid
+graph LR
+    Script[User Script] -->|Intercept| Patcher[EPI Patcher]
+    Patcher -->|Write| WAL[(Atomic SQLite)]
+    WAL -->|Package| File[.epi File]
+    File -->|Sign| Key[Ed25519 Key]
+```
+
+---
+
+## Security & Compliance
+
+While EPI is built for daily debugging, it provides the cryptographic infrastructure required for regulated environments:
+
+-   **Signatures**: Ed25519 with client-side verification (zero-knowledge).
+-   **Standards**: Supports EU AI Act Article 6 logging requirements.
+-   **Privacy**: Automatic PII redaction, air-gapped operation (no cloud required).
+
+*[Enterprise support available](mailto:enterprise@epilabs.org) for SOC2/ISO27001 environments.*
+
+---
+
+## Contributing
+
+We welcome contributions! Please see [CONTRIBUTING.md](./CONTRIBUTING.md) for details.
+
+```bash
+git clone https://github.com/mohdibrahimaiml/epi-recorder.git
+cd epi-recorder
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+Apache-2.0 License. See [LICENSE](./LICENSE) for details.
+
+
+ 

epi_recorder-2.2.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+epi_analyzer/__init__.py,sha256=4MLnfvsm7Uow9PTMqBIYF1HnHyDa0OZ2kklfDvTRp1s,134
+epi_analyzer/detector.py,sha256=JrZ7NGmG0vWb2Vskh-U_S1KYkyVjscUywwu2roe1HCQ,13665
+epi_cli/__init__.py,sha256=KEh3YUH01d0w7B-56gEIgI87jHh-UDuIkxzVSfzl7y4,104
+epi_cli/__main__.py,sha256=HzwyIuqH0lO6pJMApM8ZsXcNzmq9wIKtWM1zJvzVSzY,302
+epi_cli/chat.py,sha256=D7ULAbciCPi_2rcGyoUmSYGGD0ceYYVcw4ekGSKoTQc,7400
+epi_cli/debug.py,sha256=khGJ2xiohQYUgyTBJE5iZM-w1pKA0nou6VJPvmORKF4,3794
+epi_cli/keys.py,sha256=3EZaNc-NvHNWWeHfxKTkZs3bUhOdw7sJ3X2_021__gE,9117
+epi_cli/ls.py,sha256=ijFZdnzb8ncDFmNG7-j_mg7xLkXMJ_Cd_AwIE4tE7gI,5023
+epi_cli/main.py,sha256=NXzJb2dt0R5zLIcaiy1yzdxbRcAbKWv1Oxr4tRZ4vBE,12366
+epi_cli/record.py,sha256=bmUNr2cELwo6qVKbFvWlI2HpIbcGzXcM1MT2-Fs2cxI,7327
+epi_cli/run.py,sha256=JHTL_sm1LN02IiaWBbM7vBwybzrrJbwsQsUxXg13fDY,14376
+epi_cli/verify.py,sha256=9zr5gNH0v70Ngg_5F_JuFZQcUzWQ3YhH9WFlfUS1I0o,8244
+epi_cli/view.py,sha256=EP9takENuZnRllBsxDze9Mm32TGsyxsQaUhlNmUNA_w,4027
+epi_core/__init__.py,sha256=8CTVjxZDI6oy-MMqWTILY9h8rgSZXS8kVzgySympGJU,309
+epi_core/container.py,sha256=Eop4CN3TgCoxRyEWorbjvVBnFaxS4zkccdDwgXQ4eIk,13344
+epi_core/redactor.py,sha256=GAq6R9gkuAHyzgE9sxBXpbQvL_v_myEktxTWFNFnrbY,9892
+epi_core/schemas.py,sha256=xpl6xdsIquj_j_a6h2yQ23mB92e91wuiSpKo_BHkY2c,4733
+epi_core/serialize.py,sha256=KB7Z7dfDFh6qq0tlrwjWADOBUV4z32q29Dt2yiniGGg,5691
+epi_core/storage.py,sha256=XEVbdr5xf00LDDJMqCdrZDFvVS-BZ1e1CWzDaJqG0jE,5374
+epi_core/trust.py,sha256=_RgYABg0vVH3yBDeXJD7jEyq7WMm5Sli0DHFLmu7lkQ,7970
+epi_recorder/__init__.py,sha256=IFimK8E4Mpfx6QLuL5K6SiI1JFyr7iu8Nwh2bG-axIM,402
+epi_recorder/api.py,sha256=oFHmdoAyBKi-0b8C9qvZB3q04iA0XlNMVO-Yk3kZ2Ng,22648
+epi_recorder/async_api.py,sha256=a2WQL8MnJ8uwnLD6unDZxASe5JbywP1V-8gcFyySFM8,4949
+epi_recorder/bootstrap.py,sha256=vk6mKnaHcnanm8SB7dYGPDJ8E2iSBSX3OTQ3zyO-6b0,1851
+epi_recorder/environment.py,sha256=09KuIb7GOxiSHu9OsacaxaHXFJy5e7ewbS3Jz4fX2Zk,6604
+epi_recorder/patcher.py,sha256=L773RR3vKj9rw6WVxY6c9zZfrSZMHLR03ZYxcqfbmKw,19475
+epi_recorder/test_import.py,sha256=_wrlfu0BLtT21AINf1_NugJTvM-RVNKJOyzokMezjO0,462
+epi_recorder/test_script.py,sha256=ot2vRtgvUdeqk6Oj_cz0TZyQN9fUFVHy2E82jdzZUOs,95
+epi_recorder-2.2.0.dist-info/licenses/LICENSE,sha256=uuhz9Y8AjcWd5wF_pZA2cdymDjnESrrLKWDjE_hz7dQ,10347
+epi_viewer_static/app.js,sha256=d9m9BYvhtej8xCZQ_4t-0wLHirkhWmDcIbMyJgsqDDs,16173
+epi_viewer_static/crypto.js,sha256=2bdANR9tLCPRE9joOih4kKVtptpfRXxERNps4IEhjAQ,19082
+epi_viewer_static/index.html,sha256=sPNXnDTnk0ArVLofdKB3hhd8q-NL1AUmjucytXoythk,3302
+epi_viewer_static/viewer_lite.css,sha256=EGsbTiaSZcnep5GMXm6eKxsfr9oIg_IjEDDI94KI4vc,4695
+epi_recorder-2.2.0.dist-info/METADATA,sha256=f_Ojf_H0ASyd0-5LWTrJsfgjJdx8Wnx38Af_ZoJ6_EA,6283
+epi_recorder-2.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+epi_recorder-2.2.0.dist-info/entry_points.txt,sha256=MfMwqVRx_yMGbuPpiyjz2f8fQp8TUbHmRC1H_bupoyM,41
+epi_recorder-2.2.0.dist-info/top_level.txt,sha256=osrjwlhDfJZSucB-G1u-rF6o0L1OCx2d892gSWr8Iik,77
+epi_recorder-2.2.0.dist-info/RECORD,,

{epi_recorder-2.1.3.dist-info → epi_recorder-2.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.1)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{epi_recorder-2.1.3.dist-info → epi_recorder-2.2.0.dist-info}/licenses/LICENSE

@@ -91,8 +91,8 @@
       modifications, and in Source or Object form, provided that You
       meet the following conditions:
 
-      (a) You must give any other recipients of the Work or
-          Derivative Works a copy of this License; and
+      (a) You must give any other recipients of the Work or Derivative
+          Works a copy of this License; and
 
       (b) You must cause any modified files to carry prominent notices
           stating that You changed the files; and
@@ -162,7 +162,7 @@
       other commercial damages or losses), even if such Contributor
       has been advised of the possibility of such damages.
 
-   9. Accepting Warranty or Additional Support. While redistributing
+   9. Accepting Warranty or Additional Liability. While redistributing
       the Work or Derivative Works thereof, You may choose to offer,
       and charge a fee for, acceptance of support, warranty, indemnity,
       or other liability obligations and/or rights consistent with this
@@ -173,29 +173,4 @@
       incurred by, or claims asserted against, such Contributor by reason
       of your accepting any such warranty or additional liability.
 
-   END OF TERMS AND CONDITIONS
-
-   APPENDIX: How to apply the Apache License to your work.
-
-      To apply the Apache License to your work, attach the following
-      boilerplate notice, with the fields enclosed by brackets "[]"
-      replaced with your own identifying information. (Don't include
-      the brackets!)  The text should be enclosed in the appropriate
-      comment syntax for the file format. We also recommend that a
-      file or class name and description of purpose be included on the
-      same "printed page" as the copyright notice for easier
-      identification within third-party archives.
-
-   Copyright 2024 EPI Project
-
-   Licensed under the Apache License, Version 2.0 (the "License");
-   you may not use this file except in compliance with the License.
-   You may obtain a copy of the License at
-
-       http://www.apache.org/licenses/LICENSE-2.0
-
-   Unless required by applicable law or agreed to in writing, software
-   distributed under the License is distributed on an "AS IS" BASIS,
-   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-   See the License for the specific language governing permissions and
-   limitations under the License.
+   END OF TERMS AND CONDITIONS

{epi_recorder-2.1.3.dist-info → epi_recorder-2.2.0.dist-info}/top_level.txt

@@ -1,3 +1,4 @@
+epi_analyzer
 epi_cli
 epi_core
 epi_postinstall

epi_viewer_static/app.js

@@ -37,18 +37,20 @@ async function renderTrustBadge(manifest) {
     `;
 
     // Check verification logic availability
+    const hasSignature = manifest.signature && manifest.signature !== "null" && manifest.signature.trim() !== "";
+
     if (typeof window.verifyManifestSignature !== 'function') {
-        renderBadgeResult(false, 'Missing crypto lib', manifest.signature != null);
+        renderBadgeResult(false, 'Missing crypto lib', hasSignature);
         return;
     }
 
     try {
         const result = await window.verifyManifestSignature(manifest);
         console.log("Verification Result:", result);
-        renderBadgeResult(result.valid, result.reason, manifest.signature != null);
+        renderBadgeResult(result.valid, result.reason, hasSignature);
     } catch (e) {
         console.error("Verification error:", e);
-        renderBadgeResult(false, e.message, manifest.signature != null);
+        renderBadgeResult(false, e.message, hasSignature);
     }
 }
 
@@ -211,7 +213,12 @@ function renderStep(step) {
             </span>
             <span class="text-sm font-medium text-gray-900">${kind}</span>
         </div>
-        <span class="text-xs text-gray-500">${time}</span>
+        <div class="flex items-center space-x-2">
+            <span class="text-xs text-gray-500">${time}</span>
+            <button onclick='copyStepData(${JSON.stringify(JSON.stringify(content))})' class="text-gray-400 hover:text-blue-600 transition-colors" title="Copy Raw JSON">
+                <svg class="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M8 5H6a2 2 0 00-2 2v12a2 2 0 002 2h10a2 2 0 002-2v-1M8 5a2 2 0 002 2h2a2 2 0 002-2M8 5a2 2 0 012-2h2a2 2 0 012 2m0 0h2a2 2 0 012 2v3m2 4H10m0 0l3-3m-3 3l3 3"></path></svg>
+            </button>
+        </div>
     `;
     wrapper.appendChild(header);
 
@@ -283,7 +290,7 @@ function renderLLMResponse(content) {
             html += `
                 <div class="chat-bubble mr-auto bg-green-100 text-green-900 rounded-lg px-4 py-2 text-sm">
                     <div class="text-xs font-medium mb-1 uppercase">Assistant</div>
-                    <div class="whitespace-pre-wrap">${escapeHTML(choice.message.content)}</div>
+                    <div class="whitespace-pre-wrap">${formatMessageContent(choice.message.content)}</div>
                     ${choice.finish_reason ? `<div class="text-xs text-green-700 mt-2">• ${choice.finish_reason}</div>` : ''}
                 </div>
             `;
@@ -296,7 +303,7 @@ function renderLLMResponse(content) {
         html += `
             <div class="mt-3 text-xs text-gray-600 flex items-center space-x-4">
                 <span>📊 ${content.usage.total_tokens} tokens</span>
-                <span>⚡ ${content.latency_seconds}s</span>
+                ${content.latency_seconds ? `<span>⚡ ${content.latency_seconds}s</span>` : ''}
             </div>
         `;
     }
@@ -348,6 +355,28 @@ function renderTimeline(steps) {
     }
 }
 
+// Helper: Format message content with bolding
+function formatMessageContent(text) {
+    if (!text) return '';
+    // Escape HTML first
+    let escaped = escapeHTML(text);
+    // Apply bold formatting for **text**
+    return escaped.replace(/\*\*(.*?)\*\*/g, '<strong>$1</strong>');
+}
+
+// Helper: Copy to clipboard
+window.copyStepData = function (dataStr) {
+    try {
+        const data = JSON.parse(dataStr); // It was doubly stringified
+        navigator.clipboard.writeText(JSON.stringify(data, null, 2)).then(() => {
+            // Visual feedback could be added here
+            console.log('Copied to clipboard');
+        });
+    } catch (e) {
+        console.error('Copy failed', e);
+    }
+};
+
 // Initialize viewer
 async function init() {
     const data = loadEPIData();
@@ -374,4 +403,6 @@ if (document.readyState === 'loading') {
     document.addEventListener('DOMContentLoaded', init);
 } else {
     init();
-}
+}
+
+ 

epi_viewer_static/crypto.js

@@ -515,3 +515,6 @@ async function verifyManifestSignature(manifest) {
         return { valid: false, reason: e.message };
     }
 }
+
+
+ 

epi_viewer_static/index.html

@@ -94,7 +94,7 @@
         <!-- Footer -->
         <footer class="mt-12 bg-white border-t border-gray-200">
             <div class="max-w-7xl mx-auto px-4 py-6 sm:px-6 lg:px-8">
-                EPI v2.1.0 | <span class="font-mono">application/epi+zip</span>
+                EPI v2.2.0 | <span class="font-mono">application/epi+zip</span>
             </div>
         </footer>
     </div>
@@ -102,4 +102,6 @@
     <script src="app.js"></script>
 </body>
 
-</html>
+</html>
+
+ 

epi_viewer_static/viewer_lite.css

@@ -362,4 +362,6 @@ body {
 
 .mr-2 {
     margin-right: 0.5rem;
-}
+}
+
+