migrations fixed

backend/alembic/versions/{0017_calendar_kit.py → 0020_calendar_kit.py}

@@ -1,14 +1,14 @@
 """calendar_items.kit_id: the Launch Kit a scheduled post was written from, kept apart from its attachment
 
-Revision ID: 0017_calendar_kit
-Revises: 0016_support_chat
+Revision ID: 0020_calendar_kit
+Revises: 0019_infographics_limit
 Create Date: 2026-10-06
 """
 import sqlalchemy as sa
 from alembic import op
 
-revision = "0017_calendar_kit"
-down_revision = "0016_support_chat"
+revision = "0020_calendar_kit"
+down_revision = "0019_infographics_limit"
 branch_labels = None
 depends_on = None
 

backend/app/infographics/image.py

@@ -32,7 +32,8 @@ def chrome() -> str | None:
         if found := shutil.which(name):
             return found
     patterns = ["/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
-                os.path.expanduser("~/.cache/puppeteer/chrome/*/chrome-mac-*/Google Chrome for Testing.app/Contents/MacOS/Google Chrome for Testing")]
+                os.path.expanduser("~/.cache/puppeteer/chrome/*/chrome-mac-*/"
+                                   "Google Chrome for Testing.app/Contents/MacOS/Google Chrome for Testing")]
     return next((p for pat in patterns for p in glob.glob(pat)), None)
 
 

backend/app/infographics/render.py

@@ -5,8 +5,8 @@ import hashlib
 import random
 from pathlib import Path
 
-from markupsafe import Markup
 from jinja2 import Environment, FileSystemLoader, StrictUndefined, select_autoescape
+from markupsafe import Markup
 
 from app.infographics.content import HEIGHT, WIDTH
 from app.infographics.themes import DEFAULT_THEME, THEMES, theme_id

backend/app/llm/signatures.py

@@ -451,8 +451,12 @@ class WriteReport(dspy.Signature):
     "Section 2" or "Details"), and they read in a logical order. Use "### " inside sections, lists, and tables where
     they help. When the material has numbered lines (FILE blocks), put a [n] marker after each
     claim it supports and list each n in citations with the path and lines it came from; when it is a chat transcript,
-    write no markers and no citations. Do not write a quiz, flashcards, practice or review questions into the report unless the reader's instructions
-    ask for one: those are offered to the reader separately as visuals they can add. Never write about visuals, mind maps, flashcards, quizzes, tables or timelines in the text, and never suggest or describe one, even if the reader's instructions mention them: the app offers those to the reader separately, so write only the report itself. Write in the language requested.
+    write no markers and no citations. Do not write a quiz, flashcards, practice or review questions into the report
+    unless the reader's instructions
+    ask for one: those are offered to the reader separately as visuals they can add. Never write about visuals, mind
+    maps, flashcards, quizzes, tables or timelines in the text, and never suggest or describe one, even if the
+    reader's instructions mention them: the app offers those to the reader separately, so write only the report
+    itself. Write in the language requested.
     Never use em dashes."""
 
     title: str = dspy.InputField(desc="The notebook, post or chat the material comes from")
@@ -483,7 +487,8 @@ class PlanInteractiveReport(dspy.Signature):
     section that is itself a quiz, flashcards, practice questions or a mind map, because those are visuals, and are
     suggested through a section's visual instead. Choose 4 to 8 sections in a sensible
     order, starting with an Introduction and ending with a Summary unless the reader asks for something else. Each
-    heading is specific and descriptive of what the section covers, never generic like "Details" or "Section 2". Then decide, section by section, where a visual would help the reader and which one. These are only
+    heading is specific and descriptive of what the section covers, never generic like "Details" or "Section 2".
+    Then decide, section by section, where a visual would help the reader and which one. These are only
     suggestions: nothing is built until the reader clicks Add, so suggest the spots where each really fits: a mind_map to show how
     many linked ideas fit together (at most one, usually near the start or the end), flashcards to learn a set of
     terms or facts, a quiz to check understanding after a dense section, an infographic for a one page picture of the
@@ -502,11 +507,15 @@ class PlanInteractiveReport(dspy.Signature):
 
 class WriteReportSection(dspy.Signature):
     """Write one section of a report, using only the material given. Follow the reader's instructions on style, tone
-    and length; a section is two to five short paragraphs, or a list or table where that reads better. Do not write quiz questions, flashcards or
+    and length; a section is two to five short paragraphs, or a list or table where that reads better. Do not write
+    quiz questions, flashcards or
     review questions in the section (those are offered to the reader separately). Do not repeat
     the heading, and do not use "#" or "##" headings; "###" subheadings are fine. When the material has numbered lines (FILE
     blocks), put a [n] marker after each claim it supports and list each n in citations with the path and lines; when it
-    is a chat transcript, write no markers and no citations. Never write about visuals, mind maps, flashcards, quizzes, tables or timelines in the text, and never suggest or describe one, even if the reader's instructions mention them: the app offers those to the reader separately, so write only the report itself. Write in the language requested. Never use em dashes."""
+    is a chat transcript, write no markers and no citations. Never write about visuals, mind maps, flashcards,
+    quizzes, tables or timelines in the text, and never suggest or describe one, even if the reader's instructions
+    mention them: the app offers those to the reader separately, so write only the report itself. Write in the
+    language requested. Never use em dashes."""
 
     title: str = dspy.InputField()
     report_title: str = dspy.InputField()

backend/app/pipeline/report.py

@@ -10,6 +10,7 @@ import uuid
 
 from sqlalchemy.orm import Session
 
+from app.infographics.themes import theme_id
 from app.llm import run
 from app.llm.provider import fast_lm, report_lm
 from app.llm.signatures import (
@@ -22,10 +23,9 @@ from app.models import Artifact, Job
 from app.pipeline.flashcards import make_cards
 from app.pipeline.generate import NothingToDo, mind_map_content
 from app.pipeline.material import Material, load_material
-from app.pipeline.quiz import make_questions
 from app.pipeline.passages import as_citations
+from app.pipeline.quiz import make_questions
 from app.pipeline.research import verify_citations
-from app.infographics.themes import theme_id
 from app.services.artifacts import start_artifact
 from app.services.jobs import update_job
 from app.services.renderer import PermanentJobError
@@ -319,7 +319,8 @@ def make_block(db: Session, job: Job, report: Artifact, kind: str, brief: str, t
     if kind == "infographic":
         # A picture is drawn by the renderer, so it is its own infographic (listed in the Library, counted in the
         # plan) that the block points at; the report page shows it once it is ready.
-        child, _ = start_artifact(db, report.workspace_id, "infographic", title=f"Infographic: {brief[:60] or report.content_json.get('title', '')}",
+        title = f"Infographic: {brief[:60] or report.content_json.get('title', '')}"
+        child, _ = start_artifact(db, report.workspace_id, "infographic", title=title,
                                   notebook_id=report.notebook_id,
                                   params={**_source_params(source), "theme": theme_id(theme), "instructions": brief[:600]})
         return {"title": f"Infographic: {brief[:60]}", "artifact_id": str(child.id), "theme": theme_id(theme)}

backend/app/routers/artifacts.py

@@ -8,20 +8,21 @@ from sqlalchemy import func, or_, select
 
 from app.auth import Ctx, get_ctx
 from app.config import settings
+from app.infographics.image import ImageUnavailable, render_png
+from app.infographics.themes import theme_id
 from app.models import Artifact, CalendarItem, Chat, Document, Job, Notebook, UserSavedVoice
 from app.routers.notebooks import ensure_archive_notebook
-from app.infographics.image import ImageUnavailable, render_png
 from app.services.artifacts import latest_jobs, serialize_artifact, start_artifact
 from app.services.jobs import create_job, serialize_job
 from app.services.plans import effective_plan, plan_limit_error
-from app.infographics.themes import theme_id
 from app.services.renderer import COMPOSITIONS
 from app.services.storage import storage
 from app.services.usage import check_limit
 
 router = APIRouter(prefix="/api/artifacts", tags=["artifacts"])
 
-ArtifactType = Literal["summary", "audio_overview", "video", "quote_card", "carousel", "launch_kit", "mind_map", "quiz", "flashcards", "report", "infographic"]
+ArtifactType = Literal["summary", "audio_overview", "video", "quote_card", "carousel", "launch_kit", "mind_map", "quiz",
+                       "flashcards", "report", "infographic"]
 
 
 SOURCE_TYPES = ("quiz", "flashcards", "report", "infographic")  # made from posts or chats picked in the dialog
@@ -33,7 +34,7 @@ class RenderIn(BaseModel):
 
 
 class GenerateIn(BaseModel):
-    type: Literal["summary", "audio_overview", "video", "quote_card", "carousel", "launch_kit", "mind_map", "quiz", "flashcards", "report", "infographic"]
+    type: ArtifactType
     notebook_id: uuid.UUID | None = None
     document_id: uuid.UUID | None = None
     archive: bool = False  # no notebook or post picked: use the "All posts" notebook

backend/app/routers/reports.py

@@ -15,8 +15,8 @@ from app.routers.artifacts import get_artifact_or_404
 from app.routers.videos import FocusIn, _focus_material
 from app.services.artifacts import latest_jobs, serialize_artifact
 from app.services.jobs import create_job
-from app.services.usage import check_limit
 from app.services.report_templates import INTERACTIVE, TEMPLATES
+from app.services.usage import check_limit
 
 router = APIRouter(prefix="/api/reports", tags=["reports"])
 

backend/app/services/usage.py

@@ -60,8 +60,15 @@ def usage_report(db: Session, workspace: Workspace) -> dict:
         "plan": plan.id,
         "used": used,
         "videos_resets_at": videos["resets_at"],
-        "limits": {"audio_minutes": plan.audio_minutes, "videos": videos["limit"],
-                   "launch_kits": plan.launch_kits, "reports": plan.reports, "infographics": plan.infographics, "sources": plan.sources, "indexed_posts": plan.indexed_posts},
+        "limits": {
+            "audio_minutes": plan.audio_minutes,
+            "videos": videos["limit"],
+            "launch_kits": plan.launch_kits,
+            "reports": plan.reports,
+            "infographics": plan.infographics,
+            "sources": plan.sources,
+            "indexed_posts": plan.indexed_posts,
+        },
     }
 
 

backend/app/support/scope.py

@@ -47,7 +47,10 @@ _ABOUT_MY_DATA = re.compile(
     """
 )
 # Words that mean something went wrong with their work; the reply also offers the human form.
-_BROKEN = re.compile(r"(?i)\b(?:fail\w*|slow|forever|too\s+long|taking\s+(?:so\s+|way\s+)?long|hang\w*|never\s+(?:finish|load|appear)\w*|error|broken|stuck|crash\w*|not\s+working|won'?t|doesn'?t\s+work|missing|lost)\b")
+_BROKEN = re.compile(
+    r"(?i)\b(?:fail\w*|slow|forever|too\s+long|taking\s+(?:so\s+|way\s+)?long|hang\w*"
+    r"|never\s+(?:finish|load|appear)\w*|error|broken|stuck|crash\w*|not\s+working|won'?t|doesn'?t\s+work|missing|lost)\b"
+)
 
 OUT_OF_SCOPE_REPLY = (
     "I can help with how Notestack works, but I can't see your notebooks, posts, files or account, so I can't answer "

backend/tests/test_buildout.py

@@ -1822,7 +1822,6 @@ def test_infographic_is_a_page_at_once_and_can_be_shared_by_link(client, auth, r
 
 
 def test_infographic_needs_a_source_and_enough_text_and_obeys_its_limit(client, auth, run_jobs, feed, llm, monkeypatch):
-    from app.pipeline import infographic
 
     pricing, nb = _notebook_with_pricing(client, auth, run_jobs)
     llm["ExtractIdeas"] = {"ideas": []}