adaptive-memory-multi-model-router 2.15.4 → 2.15.5

package/adapters/a3m_adapter/tests/test_adapters.py

@@ -18,73 +18,75 @@ sys.path.insert(0, '.')
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+
 def test_langchain_adapter():
     """Test LangChain adapter."""
     print("Testing LangChain adapter...")
     
     try:
-        from a3m_llm_adapter import A3MChatModel
+        from a3m_adapter import A3MLangChainAdapter
         
         # Initialize
-        llm = A3MChatModel(model="auto", temperature=0.7)
-        print(f"��✅ Initialized: {llm}")
+        llm = A3MLangChainAdapter(model="auto", temperature=0.7)
+        print(f"✅ Initialized: {llm}")
         
         # Test simple generation
         # Note: This would make actual API calls - we'll skip for now
         # In a real test, we'd mock the A3M router
-        print("��✅ LangChain adapter structure OK")
-        return True
+        print("✅ LangChain adapter structure OK")
+        assert llm is not None
         
     except Exception as e:
-        print(f"��❌ LangChain adapter failed: {e}")
-        return False
+        print(f"❌ LangChain adapter failed: {e}")
+        raise
+
 
 def test_llamaindex_adapter():
     """Test LlamaIndex adapter."""
     print("Testing LlamaIndex adapter...")
     
     try:
-        from a3m_llama_index_adapter import A3MLlamaIndexLLM
+        from a3m_adapter import A3MLlamaIndexAdapter
         
         # Initialize
-        llm = A3MLlamaIndexLLM(model="auto", temperature=0.5)
-        print(f"��✅ Initialized: {llm}")
+        llm = A3MLlamaIndexAdapter(model="auto", temperature=0.5)
+        print(f"✅ Initialized: {llm}")
         
         # Check metadata
         metadata = llm.metadata
-        print(f"��✅ Metadata: {metadata.model_name}, tokens: {metadata.num_output}")
-        return True
+        print(f"✅ Metadata: {metadata}")
+        assert metadata is not None
         
     except Exception as e:
-        print(f"��❌ LlamaIndex adapter failed: {e}")
-        return False
+        print(f"❌ LlamaIndex adapter failed: {e}")
+        raise
+
 
 def test_config():
     """Test configuration."""
     print("Testing configuration...")
     
     try:
-        from a3m_adapter_config import A3MConfig
+        from a3m_adapter import A3MConfig
         
         # Test defaults
         config = A3MConfig()
-        print(f"��✅ Default config: model={config.model}")
+        print(f"✅ Default config: model={config.model}")
         
         # Test to_dict
         data = config.to_dict()
         assert 'model' in data
-        print("��✅ Config to_dict works")
+        print("✅ Config to_dict works")
         
         # Test JSON serialization
         json_str = config.to_json()
         assert '"model"' in json_str
-        print("��✅ Config JSON serialization works")
-        
-        return True
+        print("✅ Config JSON serialization works")
         
     except Exception as e:
-        print(f"��❌ Config test failed: {e}")
-        return False
+        print(f"❌ Config test failed: {e}")
+        raise
+
 
 def main():
     """Run all tests."""
@@ -102,19 +104,15 @@ def main():
     total = len(tests)
     
     for test in tests:
-        if test():
-            passed += 1
+        test()
+        passed += 1
         print()
     
     print("=" * 50)
     print(f"Results: {passed}/{total} tests passed")
-    
-    if passed == total:
-        print("���🎉 All tests passed!")
-        return 0
-    else:
-        print("��❌ Some tests failed")
-        return 1
+    print("🎉 All tests passed!")
+    return 0
+
 
 if __name__ == "__main__":
     sys.exit(main())

package/adapters/a3m_adapter/tests/test_integration.py

@@ -0,0 +1,80 @@
+"""
+Integration tests for A3M Router adapters.
+Requires A3M Router server running on localhost:8787
+"""
+
+import pytest
+import os
+
+
+@pytest.fixture
+def a3m_server_url():
+    """Get A3M Router server URL."""
+    return os.environ.get("A3M_SERVER_URL", "http://localhost:8787")
+
+
+@pytest.fixture
+def skip_if_no_server():
+    """Skip test if server is not available."""
+    import requests
+    try:
+        resp = requests.get("http://localhost:8787/health", timeout=2)
+        if resp.status_code != 200:
+            pytest.skip("A3M Router server not running")
+    except:
+        pytest.skip("A3M Router server not running")
+
+
+@pytest.mark.integration
+def test_simple_chat_completion(a3m_server_url, skip_if_no_server):
+    """Test simple chat completion via HTTP API."""
+    import requests
+    
+    response = requests.post(
+        f"{a3m_server_url}/v1/chat/completions",
+        json={
+            "model": "auto",
+            "messages": [{"role": "user", "content": "What is 2+2?"}]
+        },
+        timeout=30
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert "choices" in data
+    assert len(data["choices"]) > 0
+    assert "message" in data["choices"][0]
+    assert data["choices"][0]["message"]["content"]
+
+
+@pytest.mark.integration
+def test_parallel_ensemble(a3m_server_url, skip_if_no_server):
+    """Test parallel ensemble with multiple providers."""
+    import requests
+    
+    response = requests.post(
+        f"{a3m_server_url}/v1/chat/completions",
+        json={
+            "model": "auto",
+            "messages": [{"role": "user", "content": "Explain gravity"}],
+            "parallel_ensemble": 3,
+        },
+        timeout=60
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert "choices" in data
+    assert "provider" in data
+
+
+@pytest.mark.integration  
+def test_health_endpoint(a3m_server_url, skip_if_no_server):
+    """Test health endpoint."""
+    import requests
+    
+    response = requests.get(f"{a3m_server_url}/health", timeout=10)
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert "providers" in data or "status" in data

package/adapters/requirements-dev.txt

@@ -0,0 +1,6 @@
+-r requirements.txt
+black>=21.12
+isort>=5.10.1
+flake8>=3.9.0
+build>=0.10.0
+twine>=4.0.0

package/adapters/requirements.txt

@@ -0,0 +1,4 @@
+requests>=2.25.1
+pydantic>=1.9.0
+pytest>=6.0
+pytest-asyncio>=0.21.0

package/apps/cost-calculator/README.md

@@ -0,0 +1,72 @@
+# A3M Router Cost Calculator
+
+**Live:** https://a3m.router/calculator
+
+Drop-in interactive calculator. No build step. No npm dependencies for the calculator itself.
+
+## What it does
+
+- Input: monthly request volume + model mix
+- Output: side-by-side cost comparison (baseline vs A3M routed)
+- Generates shareable URL with encoded config (base64url)
+- Exports comparison as CSV
+
+## Quick Start
+
+```bash
+# Open directly
+open apps/cost-calculator/index.html
+
+# Or serve locally
+npx serve apps/cost-calculator
+```
+
+## Embed in any site
+
+```html
+<iframe
+  src="https://a3m.router/calculator"
+  width="100%"
+  height="700"
+  frameborder="0"
+  style="border-radius: 12px;"
+></iframe>
+```
+
+## Architecture
+
+```
+calculator.js    — Pure JS, no dependencies. All pricing, routing, and encoding logic.
+calculator.css   — Standalone styles. Dark theme, responsive.
+index.html       — Self-contained. Vanilla JS. Loads calculator.js.
+package.json    — vitest for unit testing the pure JS logic.
+```
+
+## Testing
+
+```bash
+cd apps/cost-calculator
+npm install
+npm test
+```
+
+## Models Supported
+
+| Premium Model | Routed To | Savings |
+|---------------|-----------|---------|
+| GPT-4o | GPT-4o-mini | ~35% |
+| Claude 3.5 Sonnet | Claude 3 Haiku | ~40% |
+| Gemini 1.5 Pro | Gemini 1.5 Flash | ~45% |
+| Groq Llama 3.3 70B | Groq Mixtral 8x7B | ~30% |
+
+## Pricing Source
+
+OpenRouter provider rates (per 1M tokens). Update `PRICING` constant in `calculator.js` to reflect latest rates.
+
+## Next Steps
+
+- [ ] Deploy to `a3m.router/calculator` (Vercel static)
+- [ ] Add UTM tracking on share links
+- [ ] Add "embed widget" script for dynamic resize
+- [ ] Write HN + Reddit post with actual example calculations
+- [ ] Add unit tests with vitest

package/apps/cost-calculator/calculator.css

@@ -0,0 +1,280 @@
+/* A3M Router Cost Calculator Styles */
+
+:root {
+  --a3m-primary: #6366f1;
+  --a3m-accent: #818cf8;
+  --a3m-bg: #0f0f1a;
+  --a3m-surface: #1a1a2e;
+  --a3m-border: #2d2d4a;
+  --a3m-text: #e2e8f0;
+  --a3m-muted: #94a3b8;
+  --a3m-green: #22c55e;
+  --a3m-red: #ef4444;
+}
+
+* { box-sizing: border-box; margin: 0; padding: 0; }
+
+body {
+  font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+  background: var(--a3m-bg);
+  color: var(--a3m-text);
+  min-height: 100vh;
+  padding: 2rem 1rem;
+}
+
+.container {
+  max-width: 900px;
+  margin: 0 auto;
+}
+
+h1 {
+  font-size: 1.75rem;
+  font-weight: 700;
+  text-align: center;
+  margin-bottom: 0.5rem;
+}
+
+.subtitle {
+  text-align: center;
+  color: var(--a3m-muted);
+  margin-bottom: 2rem;
+}
+
+.card {
+  background: var(--a3m-surface);
+  border: 1px solid var(--a3m-border);
+  border-radius: 12px;
+  padding: 1.5rem;
+  margin-bottom: 1.5rem;
+}
+
+.card-title {
+  font-size: 0.875rem;
+  font-weight: 600;
+  text-transform: uppercase;
+  letter-spacing: 0.05em;
+  color: var(--a3m-muted);
+  margin-bottom: 1rem;
+}
+
+/* Input grid */
+.input-grid {
+  display: grid;
+  grid-template-columns: 1fr 1fr;
+  gap: 1rem;
+}
+
+@media (max-width: 600px) {
+  .input-grid { grid-template-columns: 1fr; }
+}
+
+.input-group {
+  display: flex;
+  flex-direction: column;
+  gap: 0.5rem;
+}
+
+.input-group label {
+  font-size: 0.875rem;
+  font-weight: 500;
+  color: var(--a3m-muted);
+}
+
+.input-group input,
+.input-group select {
+  padding: 0.625rem 0.875rem;
+  background: var(--a3m-bg);
+  border: 1px solid var(--a3m-border);
+  border-radius: 8px;
+  color: var(--a3m-text);
+  font-size: 1rem;
+  width: 100%;
+}
+
+.input-group input:focus,
+.input-group select:focus {
+  outline: none;
+  border-color: var(--a3m-primary);
+}
+
+/* Model mix rows */
+.model-mix {
+  display: flex;
+  flex-direction: column;
+  gap: 0.75rem;
+  margin-top: 1rem;
+}
+
+.model-row {
+  display: grid;
+  grid-template-columns: 1fr 100px auto;
+  gap: 0.5rem;
+  align-items: center;
+}
+
+.model-row select,
+.model-row input {
+  padding: 0.5rem;
+  background: var(--a3m-bg);
+  border: 1px solid var(--a3m-border);
+  border-radius: 6px;
+  color: var(--a3m-text);
+  font-size: 0.875rem;
+}
+
+.model-row span {
+  color: var(--a3m-muted);
+  font-size: 0.875rem;
+  white-space: nowrap;
+}
+
+.btn {
+  display: inline-flex;
+  align-items: center;
+  gap: 0.5rem;
+  padding: 0.625rem 1.25rem;
+  background: var(--a3m-primary);
+  color: white;
+  border: none;
+  border-radius: 8px;
+  font-size: 0.875rem;
+  font-weight: 600;
+  cursor: pointer;
+  transition: background 0.2s;
+}
+
+.btn:hover { background: var(--a3m-accent); }
+.btn-secondary {
+  background: var(--a3m-border);
+  color: var(--a3m-text);
+}
+.btn-secondary:hover { background: #3d3d5a; }
+
+.btn-row {
+  display: flex;
+  gap: 0.75rem;
+  margin-top: 1.25rem;
+  flex-wrap: wrap;
+}
+
+/* Results */
+.results-grid {
+  display: grid;
+  grid-template-columns: repeat(3, 1fr);
+  gap: 1rem;
+  margin-bottom: 1.5rem;
+}
+
+@media (max-width: 600px) {
+  .results-grid { grid-template-columns: 1fr; }
+}
+
+.result-card {
+  background: var(--a3m-bg);
+  border: 1px solid var(--a3m-border);
+  border-radius: 10px;
+  padding: 1.25rem;
+  text-align: center;
+}
+
+.result-label {
+  font-size: 0.75rem;
+  font-weight: 600;
+  text-transform: uppercase;
+  letter-spacing: 0.05em;
+  color: var(--a3m-muted);
+  margin-bottom: 0.5rem;
+}
+
+.result-value {
+  font-size: 1.75rem;
+  font-weight: 700;
+}
+
+.result-value.savings { color: var(--a3m-green); }
+.result-value.baseline { color: var(--a3m-muted); text-decoration: line-through; }
+
+/* Breakdown table */
+.breakdown-table {
+  width: 100%;
+  border-collapse: collapse;
+  font-size: 0.875rem;
+  margin-top: 1rem;
+}
+
+.breakdown-table th {
+  text-align: left;
+  padding: 0.625rem 0.75rem;
+  color: var(--a3m-muted);
+  font-weight: 600;
+  font-size: 0.75rem;
+  text-transform: uppercase;
+  border-bottom: 1px solid var(--a3m-border);
+}
+
+.breakdown-table td {
+  padding: 0.625rem 0.75rem;
+  border-bottom: 1px solid var(--a3m-border);
+}
+
+.breakdown-table tr:last-child td { border-bottom: none; }
+.breakdown-table .num { text-align: right; font-variant-numeric: tabular-nums; }
+.breakdown-table .savings-cell { color: var(--a3m-green); }
+
+/* Share bar */
+.share-bar {
+  display: flex;
+  gap: 0.5rem;
+  align-items: center;
+  margin-top: 1rem;
+}
+
+.share-bar input {
+  flex: 1;
+  padding: 0.5rem 0.75rem;
+  background: var(--a3m-bg);
+  border: 1px solid var(--a3m-border);
+  border-radius: 6px;
+  color: var(--a3m-text);
+  font-size: 0.8rem;
+  font-family: monospace;
+}
+
+/* Chart */
+.chart-container {
+  position: relative;
+  height: 250px;
+  margin: 1.5rem 0;
+}
+
+.chart-bar {
+  position: absolute;
+  bottom: 0;
+  width: 40%;
+  border-radius: 6px 6px 0 0;
+  transition: height 0.5s ease;
+}
+
+.chart-bar.baseline {
+  left: 15%;
+  background: var(--a3m-muted);
+  opacity: 0.5;
+}
+
+.chart-bar.a3m {
+  right: 15%;
+  background: var(--a3m-primary);
+}
+
+/* Hidden until calculated */
+.hidden { display: none; }
+
+/* Footer */
+.footer {
+  text-align: center;
+  color: var(--a3m-muted);
+  font-size: 0.75rem;
+  margin-top: 2rem;
+}
+
+.footer a { color: var(--a3m-accent); text-decoration: none; }