memra 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

memra-ops/scripts/check_database.py

@@ -0,0 +1,37 @@
+import psycopg2
+from prettytable import PrettyTable
+import time
+
+def check_database():
+    # Connect to the Postgres database
+    conn = psycopg2.connect(
+        dbname="memra_invoice_db",
+        user="memra",          # From docker-compose.yml
+        password="memra123",   # From docker-compose.yml
+        host="localhost",
+        port=5432
+    )
+
+    # Create a cursor and run the query
+    cur = conn.cursor()
+    cur.execute("SELECT * FROM invoices ORDER BY created_at DESC LIMIT 10;")
+    rows = cur.fetchall()
+    columns = [desc[0] for desc in cur.description]
+
+    # Create and populate the pretty table
+    table = PrettyTable()
+    table.field_names = columns
+    for row in rows:
+        table.add_row(row)
+
+    # Print the table
+    print(f"\n📊 Current invoices in database (as of {time.strftime('%H:%M:%S')}):")
+    print(table)
+    print(f"Total rows: {len(rows)}")
+
+    # Clean up
+    cur.close()
+    conn.close()
+
+if __name__ == "__main__":
+    check_database() 

memra-ops/scripts/clear_database.py

@@ -0,0 +1,48 @@
+import psycopg2
+
+def clear_database():
+    """Clear all data from the invoices table"""
+    
+    # Connect to the Postgres database
+    conn = psycopg2.connect(
+        dbname="memra_invoice_db",
+        user="memra",          # From docker-compose.yml
+        password="memra123",   # From docker-compose.yml
+        host="localhost",
+        port=5432
+    )
+
+    try:
+        # Create a cursor and run the query
+        cur = conn.cursor()
+        
+        # First, let's see how many rows we have
+        cur.execute("SELECT COUNT(*) FROM invoices;")
+        count_before = cur.fetchone()[0]
+        print(f"📊 Current invoice count: {count_before}")
+        
+        if count_before > 0:
+            # Clear all data from the invoices table
+            cur.execute("DELETE FROM invoices;")
+            conn.commit()
+            
+            # Verify the deletion
+            cur.execute("SELECT COUNT(*) FROM invoices;")
+            count_after = cur.fetchone()[0]
+            
+            print(f"🗑️  Deleted {count_before} invoice records")
+            print(f"📊 New invoice count: {count_after}")
+            print("✅ Database cleared successfully!")
+        else:
+            print("📊 Database is already empty")
+        
+        cur.close()
+        
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        conn.rollback()
+    finally:
+        conn.close()
+
+if __name__ == "__main__":
+    clear_database() 

memra-ops/scripts/monitor_database.py

@@ -0,0 +1,67 @@
+import psycopg2
+from prettytable import PrettyTable
+import time
+import os
+
+def monitor_database():
+    """Continuously monitor the database for new invoice rows"""
+    
+    # Connect to the Postgres database
+    conn = psycopg2.connect(
+        dbname="memra_invoice_db",
+        user="memra",          # From docker-compose.yml
+        password="memra123",   # From docker-compose.yml
+        host="localhost",
+        port=5432
+    )
+
+    print("🔍 Monitoring database for new invoice rows...")
+    print("Press Ctrl+C to stop monitoring\n")
+    
+    last_count = 0
+    
+    try:
+        while True:
+            # Create a cursor and run the query
+            cur = conn.cursor()
+            cur.execute("SELECT COUNT(*) FROM invoices;")
+            current_count = cur.fetchone()[0]
+            
+            # Get the latest invoices
+            cur.execute("SELECT id, invoice_number, vendor_name, total_amount, created_at FROM invoices ORDER BY created_at DESC LIMIT 5;")
+            rows = cur.fetchall()
+            
+            # Clear screen (works on Unix-like systems)
+            os.system('clear' if os.name == 'posix' else 'cls')
+            
+            print(f"📊 Database Monitor - {time.strftime('%Y-%m-%d %H:%M:%S')}")
+            print(f"Total invoices: {current_count}")
+            
+            if current_count > last_count:
+                print(f"🆕 New invoices detected! (+{current_count - last_count})")
+                last_count = current_count
+            
+            if rows:
+                # Create and populate the pretty table
+                table = PrettyTable()
+                table.field_names = ["ID", "Invoice #", "Vendor", "Amount", "Created"]
+                for row in rows:
+                    table.add_row(row)
+                print("\nLatest invoices:")
+                print(table)
+            else:
+                print("\nNo invoices found in database.")
+            
+            print(f"\nMonitoring... (refresh every 2 seconds)")
+            cur.close()
+            time.sleep(2)
+            
+    except KeyboardInterrupt:
+        print("\n\n🛑 Monitoring stopped by user.")
+    except Exception as e:
+        print(f"\n❌ Error: {e}")
+    finally:
+        conn.close()
+
+if __name__ == "__main__":
+    monitor_database() 

memra-ops/scripts/reset_database.py

@@ -0,0 +1,65 @@
+import psycopg2
+
+def reset_database():
+    """Clear all data from the invoices table and reset the sequence"""
+    
+    # Connect to the Postgres database
+    conn = psycopg2.connect(
+        dbname="memra_invoice_db",
+        user="memra",          # From docker-compose.yml
+        password="memra123",   # From docker-compose.yml
+        host="localhost",
+        port=5432
+    )
+
+    try:
+        # Create a cursor and run the query
+        cur = conn.cursor()
+        
+        # First, let's see how many rows we have and current sequence value
+        cur.execute("SELECT COUNT(*) FROM invoices;")
+        count_before = cur.fetchone()[0]
+        
+        cur.execute("SELECT last_value FROM invoices_id_seq;")
+        sequence_before = cur.fetchone()[0]
+        
+        print(f"📊 Current invoice count: {count_before}")
+        print(f"🔢 Current sequence value: {sequence_before}")
+        
+        if count_before > 0:
+            # Clear all data from the invoices table
+            cur.execute("DELETE FROM invoices;")
+            
+            # Reset the sequence to start from 1
+            cur.execute("ALTER SEQUENCE invoices_id_seq RESTART WITH 1;")
+            
+            conn.commit()
+            
+            # Verify the deletion and sequence reset
+            cur.execute("SELECT COUNT(*) FROM invoices;")
+            count_after = cur.fetchone()[0]
+            
+            cur.execute("SELECT last_value FROM invoices_id_seq;")
+            sequence_after = cur.fetchone()[0]
+            
+            print(f"🗑️  Deleted {count_before} invoice records")
+            print(f"🔄 Reset sequence from {sequence_before} to {sequence_after}")
+            print(f"📊 New invoice count: {count_after}")
+            print("✅ Database reset successfully!")
+        else:
+            print("📊 Database is already empty")
+            # Still reset the sequence
+            cur.execute("ALTER SEQUENCE invoices_id_seq RESTART WITH 1;")
+            conn.commit()
+            print("🔄 Reset sequence to start from 1")
+        
+        cur.close()
+        
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        conn.rollback()
+    finally:
+        conn.close()
+
+if __name__ == "__main__":
+    reset_database() 

memra-ops/server_tool_registry.py

@@ -137,7 +137,9 @@ class ServerToolRegistry:
         
         elif tool_name == "PDFProcessor":
             file_path = input_data.get("file", "")
-            return tool_instance.process_pdf(file_path)
+            # Handle both "schema" and "invoice_schema" keys
+            schema = input_data.get("schema") or input_data.get("invoice_schema", {})
+            return tool_instance.process_pdf(file_path, schema)
         
         elif tool_name == "OCRTool":
             # Assume PDF processor output is passed as input

memra-sdk/memra/__init__.py

@@ -6,7 +6,7 @@ Think of it as "Kubernetes for business logic" where agents are the pods and
 departments are the deployments.
 """
 
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 
 # Core imports
 from .models import Agent, Department, Tool, LLM

memra-sdk/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="memra",
-    version="0.2.3",
+    version="0.2.4",
     author="Memra",
     author_email="support@memra.com",
     description="Declarative framework for enterprise workflows with MCP integration - Client SDK",

memra-0.2.4.dist-info/METADATA

@@ -1,145 +0,0 @@
-Metadata-Version: 2.4
-Name: memra
-Version: 0.2.4
-Summary: Declarative framework for enterprise workflows with MCP integration - Client SDK
-Home-page: https://github.com/memra/memra-sdk
-Author: Memra
-Author-email: Memra <support@memra.com>
-License: MIT
-Project-URL: Homepage, https://memra.co
-Project-URL: Repository, https://github.com/memra-platform/memra-sdk
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: pydantic>=1.8.0
-Requires-Dist: httpx>=0.24.0
-Requires-Dist: typing-extensions>=4.0.0
-Requires-Dist: aiohttp>=3.8.0
-Requires-Dist: aiohttp-cors>=0.7.0
-Provides-Extra: dev
-Requires-Dist: pytest>=6.0; extra == "dev"
-Requires-Dist: pytest-asyncio; extra == "dev"
-Requires-Dist: black; extra == "dev"
-Requires-Dist: flake8; extra == "dev"
-Provides-Extra: mcp
-Requires-Dist: psycopg2-binary>=2.9.0; extra == "mcp"
-Dynamic: author
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-python
-
-# Memra SDK
-
-The core Memra framework for building AI-powered business workflows.
-
-## Installation
-
-```bash
-pip install memra
-```
-
-## Quick Start
-
-```python
-from memra import Agent, Department, LLM, ExecutionEngine
-
-# Define an agent
-agent = Agent(
-    role="Data Analyst",
-    job="Analyze customer data",
-    llm=LLM(model="llama-3.2-11b-vision-preview"),
-    sops=["Load data", "Perform analysis", "Generate report"],
-    output_key="analysis_result"
-)
-
-# Create a department
-department = Department(
-    name="Analytics",
-    mission="Provide data insights",
-    agents=[agent],
-    workflow_order=["Data Analyst"]
-)
-
-# Execute the workflow
-engine = ExecutionEngine()
-result = engine.execute_department(department, {"data": "customer_data.csv"})
-```
-
-## Core Components
-
-### Agent
-An AI worker that performs specific tasks using LLMs and tools.
-
-### Department
-A team of agents working together to accomplish a mission.
-
-### ExecutionEngine
-Orchestrates the execution of departments and their workflows.
-
-### LLM
-Configuration for language models used by agents.
-
-## Examples
-
-See the `examples/` directory for basic usage examples:
-- `simple_text_to_sql.py` - Basic text-to-SQL conversion
-- `ask_questions.py` - Simple question answering
-
-## Documentation
-
-For detailed documentation, visit [docs.memra.co](https://docs.memra.co)
-
-Documentation is also available locally in the `examples/` directory.
-
-## Example: Propane Delivery Workflow
-
-See the `examples/propane_delivery.py` file for a complete example of how to use Memra to orchestrate a propane delivery workflow.
-
-## 🔍 Smart File Discovery
-
-Memra includes intelligent file discovery and management capabilities:
-
-### File Discovery Tools
-- **FileDiscovery**: Automatically scan directories for files matching patterns
-- **FileCopy**: Copy files from external locations to standard processing directories
-- **Smart Routing**: Automatically handle file paths and directory management
-
-### Example: Smart Invoice Processing
-```python
-from memra import Agent
-
-# Smart agent that discovers and processes files automatically
-smart_parser = Agent(
-    role="Smart Invoice Parser",
-    job="Discover and process invoice files intelligently",
-    tools=[
-        {"name": "FileDiscovery", "hosted_by": "memra"},
-        {"name": "FileCopy", "hosted_by": "memra"},
-        {"name": "InvoiceExtractionWorkflow", "hosted_by": "memra"}
-    ]
-)
-
-# Three modes of operation:
-# 1. Auto-discovery: Scan invoices/ directory
-# 2. External file: Copy from Downloads to invoices/
-# 3. Specific file: Process exact file path
-```
-
-See `examples/accounts_payable_smart.py` for a complete implementation.
-
-## Contributing
-
-We welcome contributions! Please see our [contributing guide](CONTRIBUTING.md) for details.
-
-## License
-
-MIT License - see LICENSE file for details. 

memra-0.2.4.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-memra = memra.cli:main