slide-narrator 0.2.1__py3-none-any.whl

narrator/__init__.py

@@ -0,0 +1,18 @@
+"""
+The Narrator - Thread and file storage components for conversational AI
+"""
+
+from .database.thread_store import ThreadStore
+from .storage.file_store import FileStore
+from .models.thread import Thread
+from .models.message import Message
+from .models.attachment import Attachment
+
+__version__ = "0.2.1"
+__all__ = [
+    "ThreadStore",
+    "FileStore", 
+    "Thread",
+    "Message",
+    "Attachment",
+] 

narrator/database/__init__.py

@@ -0,0 +1,8 @@
+"""
+Database package for Tyler Stores
+"""
+
+from .thread_store import ThreadStore
+from .models import ThreadRecord, MessageRecord
+
+__all__ = ["ThreadStore", "ThreadRecord", "MessageRecord"] 

narrator/database/cli.py

@@ -0,0 +1,66 @@
+"""Database CLI for Tyler Stores"""
+import asyncio
+import click
+from .thread_store import ThreadStore
+from ..utils.logging import get_logger
+
+logger = get_logger(__name__)
+
+@click.group()
+def main():
+    """Tyler Stores Database CLI"""
+    pass
+
+@click.command()
+@click.option('--database-url', help='Database URL for initialization')
+async def init(database_url):
+    """Initialize database tables"""
+    try:
+        if database_url:
+            store = await ThreadStore.create(database_url)
+        else:
+            # Use environment variables or default
+            store = await ThreadStore.create()
+        
+        logger.info("Database initialized successfully")
+        click.echo("Database initialized successfully")
+    except Exception as e:
+        logger.error(f"Failed to initialize database: {e}")
+        click.echo(f"Error: Failed to initialize database: {e}")
+        raise click.Abort()
+
+@click.command()
+@click.option('--database-url', help='Database URL')
+async def status(database_url):
+    """Check database status"""
+    try:
+        if database_url:
+            store = await ThreadStore.create(database_url)
+        else:
+            store = await ThreadStore.create()
+        
+        # Get some basic stats
+        threads = await store.list_recent(limit=5)
+        click.echo(f"Database connection: OK")
+        click.echo(f"Recent threads count: {len(threads)}")
+        
+    except Exception as e:
+        logger.error(f"Database status check failed: {e}")
+        click.echo(f"Error: Database status check failed: {e}")
+        raise click.Abort()
+
+# Add async wrapper for commands
+def async_command(f):
+    def wrapper(*args, **kwargs):
+        return asyncio.run(f(*args, **kwargs))
+    return wrapper
+
+# Apply async wrapper to commands
+init = click.command()(async_command(init))
+status = click.command()(async_command(status))
+
+main.add_command(init)
+main.add_command(status)
+
+if __name__ == '__main__':
+    main() 

narrator/database/migrations/__init__.py

@@ -0,0 +1,6 @@
+"""
+Database migrations for Tyler Stores
+
+This directory will contain future database migrations.
+For new installations, the schema is created directly from the models.
+""" 

narrator/database/models.py

@@ -0,0 +1,69 @@
+import json
+from sqlalchemy.types import TypeDecorator, TEXT, JSON
+
+"""Database models for SQLAlchemy"""
+from sqlalchemy import Column, String, DateTime, Text, ForeignKey, Integer
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.orm import declarative_base
+from sqlalchemy.orm import relationship
+from datetime import datetime, UTC
+
+class JSONBCompat(TypeDecorator):
+    impl = TEXT
+    cache_ok = True
+
+    def load_dialect_impl(self, dialect):
+        if dialect.name == 'postgresql':
+            return dialect.type_descriptor(JSONB())
+        else:
+            return dialect.type_descriptor(JSON())
+
+    def process_bind_param(self, value, dialect):
+        if dialect.name == 'postgresql':
+            return value
+        if value is not None:
+            return value
+        return value
+
+    def process_result_value(self, value, dialect):
+        if dialect.name == 'postgresql':
+            return value
+        if value is not None:
+            return value
+        return value
+
+Base = declarative_base()
+
+class ThreadRecord(Base):
+    __tablename__ = 'threads'
+    
+    id = Column(String, primary_key=True)
+    title = Column(String, nullable=True)
+    attributes = Column(JSONBCompat, nullable=False, default={})
+    platforms = Column(JSONBCompat, nullable=True)
+    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
+    updated_at = Column(DateTime(timezone=True), default=lambda: datetime.now(UTC), onupdate=lambda: datetime.now(UTC))
+    
+    messages = relationship("MessageRecord", back_populates="thread", cascade="all, delete-orphan")
+
+class MessageRecord(Base):
+    __tablename__ = 'messages'
+    
+    id = Column(String, primary_key=True)
+    thread_id = Column(String, ForeignKey('threads.id', ondelete='CASCADE'), nullable=False)
+    sequence = Column(Integer, nullable=False)
+    turn = Column(Integer, nullable=True)
+    role = Column(String, nullable=False)
+    content = Column(Text, nullable=True)
+    name = Column(String, nullable=True)
+    tool_call_id = Column(String, nullable=True)
+    tool_calls = Column(JSONBCompat, nullable=True)
+    attributes = Column(JSONBCompat, nullable=False, default={})
+    timestamp = Column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
+    source = Column(JSONBCompat, nullable=True)
+    platforms = Column(JSONBCompat, nullable=True)
+    attachments = Column(JSONBCompat, nullable=True)
+    metrics = Column(JSONBCompat, nullable=False, default={})
+    reactions = Column(JSONBCompat, nullable=True)
+    
+    thread = relationship("ThreadRecord", back_populates="messages")