kyp-mem 0.7.4 → 0.9.0

package/kyp_mem/cli.py

@@ -49,6 +49,10 @@ def main():
     cfg_parser.add_argument("key", nargs="?", help="Config key (e.g. session_model)")
     cfg_parser.add_argument("value", nargs="?", help="Value to set")
 
+    obj_parser = subparsers.add_parser("objective", help="Get or set a project's objective (injected at session start)")
+    obj_parser.add_argument("project", nargs="?", help="Project name (defaults to current directory name)")
+    obj_parser.add_argument("text", nargs="*", help="Objective text to set (omit to read the current objective)")
+
     hook_parser = subparsers.add_parser("hook", help="Handle Claude Code hook events (internal)")
     hook_sub = hook_parser.add_subparsers(dest="hook_command")
     hook_sub.add_parser("session-start", help="Inject project context at session start")
@@ -79,6 +83,8 @@ def main():
         _run_install_hooks(global_config=args.global_config, remove=args.remove)
     elif args.command == "config":
         _run_config(args.key, args.value)
+    elif args.command == "objective":
+        _run_objective(args.project, " ".join(args.text).strip())
     elif args.command == "uninstall":
         _run_uninstall(purge=args.purge)
     elif args.command == "doctor":
@@ -460,6 +466,39 @@ def _run_config(key, value):
     print(f"  {G}✓{R} {key} = {value}")
 
 
+def _run_objective(project, text):
+    from .config import get_vault_path
+    from .vault import Vault
+
+    project = project or Path.cwd().name
+    vault = Vault(get_vault_path())
+    path = f"{project}/Objective.md"
+
+    if not text:
+        note = vault.read(path)
+        print()
+        print(f"  {C}KYP-MEM{R} — Objective for {G}{project}{R}")
+        print()
+        if not note:
+            print(f"  {Y}(not set){R}")
+            print(f"  {D}  Set one: kyp-mem objective {project} \"<your goal>\"{R}")
+        else:
+            content = note.content.strip()
+            lines = content.split("\n")
+            if lines and lines[0].lstrip().startswith("# "):
+                content = "\n".join(lines[1:]).strip()
+            print(f"  {content}")
+        print()
+        return
+
+    content = f"# Objective\n\n{text}\n"
+    vault.write_note(path, content, ["objective", project.lower().replace(" ", "-")], {})
+    print()
+    print(f"  {G}✓{R} Objective saved for {G}{project}{R} ({path})")
+    print(f"  {D}  Injected at every session start.{R}")
+    print()
+
+
 def _run_stats():
     from .config import get_vault_path
     from .vault import Vault

package/kyp_mem/hooks.py

@@ -145,8 +145,42 @@ def _build_stats_line(project_name, injected_chars, session_ids):
         return None
 
 
+def _objective_note_path(project_name):
+    """Canonical vault path for a project's objective note."""
+    return f"{project_name}/Objective.md"
+
+
+def _find_objective_path(vault, project_name):
+    """Case-insensitive lookup of the objective note (vault casing may differ
+    from the cwd basename on case-insensitive filesystems)."""
+    target = f"{project_name}/objective.md".lower()
+    for p in vault.index.notes:
+        if p.lower() == target:
+            return p
+    return None
+
+
+def _read_objective(vault, project_name):
+    """Return the objective text for a project, or None if not set.
+
+    Strips an optional leading ``# ...`` heading so only the objective body
+    is surfaced.
+    """
+    path = _find_objective_path(vault, project_name)
+    if not path:
+        return None
+    note = vault.read(path)
+    if not note:
+        return None
+    lines = note.content.strip().split("\n")
+    if lines and lines[0].lstrip().startswith("# "):
+        lines = lines[1:]
+    text = "\n".join(lines).strip()
+    return text or None
+
+
 def handle_session_start():
-    """Inject recent session memory into the conversation at session start."""
+    """Inject the project objective and recent session memory at session start."""
     sys.stdin.read()
     if _is_subprocess():
         return
@@ -167,36 +201,63 @@ def handle_session_start():
         # filesystems "KYP-MEM" and "kyp-mem" are the same directory).
         prefix = f"{project_name}/".lower()
         project_notes = [p for p in vault.index.notes if p.lower().startswith(prefix)]
-        if not project_notes:
-            return
+
+        objective = _read_objective(vault, project_name)
 
         sessions = sorted(
             (p for p in project_notes if "/sessions/" in p.lower()),
             reverse=True,
         )[:10]
-        if not sessions:
-            return
 
-        parts = [f"# [kyp-mem] {project_name} — Recent Sessions"]
+        # Nothing to say: no objective to surface, no objective to request
+        # (project is already known but sessions just haven't been captured),
+        # and no sessions. Only stay silent when this is an established project
+        # with an objective but zero sessions — otherwise we always at least
+        # surface or request the objective.
+        if not objective and not project_notes and not sessions:
+            # Brand-new / unknown directory: still ask for an objective so the
+            # project starts with a clear goal.
+            pass
+
+        parts = [f"# [kyp-mem] {project_name} — Session Context"]
         parts.append(f"Use `kyp_search` or `kyp_project_context` for architecture/project knowledge on demand.\n")
 
-        parts.append(f"## Last {len(sessions)} Sessions")
-        for sp in sessions:
-            note = vault.read(sp)
-            if not note:
-                continue
-            parts.append(f"### {note.title}")
-            summary = _extract_session_summary(note.content)
-            parts.append(summary)
+        # --- Objective (always first) ---
+        if objective:
+            parts.append("## 🎯 Objective")
+            parts.append(objective)
             parts.append("")
+        else:
+            parts.append("## 🎯 Objective — NOT SET")
+            parts.append(
+                f"No objective is recorded for **{project_name}**. Before anything else, "
+                "ask the user: **\"What is the main objective / goal for this project?\"** "
+                "When they answer, save it by calling "
+                f"`kyp_objective_set(project=\"{project_name}\", objective=\"...\")` "
+                f"(or `kyp_write(\"{project_name}/Objective.md\", ...)`). "
+                "Keep your work aligned to this objective every session."
+            )
+            parts.append("")
+
+        # --- Recent sessions ---
+        if sessions:
+            parts.append(f"## Last {len(sessions)} Sessions")
+            for sp in sessions:
+                note = vault.read(sp)
+                if not note:
+                    continue
+                parts.append(f"### {note.title}")
+                summary = _extract_session_summary(note.content)
+                parts.append(summary)
+                parts.append("")
 
-        session_ids = {Path(sp).stem for sp in sessions}
-        stats_line = _build_stats_line(project_name, len("\n".join(parts)), session_ids)
-        if stats_line:
-            parts.append(stats_line)
+            session_ids = {Path(sp).stem for sp in sessions}
+            stats_line = _build_stats_line(project_name, len("\n".join(parts)), session_ids)
+            if stats_line:
+                parts.append(stats_line)
 
         parts.append("")
-        parts.append("**CRITICAL: Your FIRST response to the user MUST be displaying the session summaries below. Do NOT skip this. Do NOT wait for user input. Display them immediately, formatted cleanly, before doing anything else.**")
+        parts.append("**CRITICAL: Your FIRST response to the user MUST surface this context — display the session summaries above (if any) and the objective. If the objective is NOT SET, ask the user for it as instructed. Do this immediately, formatted cleanly, before anything else.**")
 
         output = "\n".join(parts)
 

package/kyp_mem/server.py

@@ -22,6 +22,7 @@ YOU MUST FOLLOW THESE INSTRUCTIONS when kyp-mem tools are available.
 2. Call `kyp_project_context(project)` to load the project's knowledge base, notes, and recent session summaries.
 3. If no project exists yet, call `kyp_project_context` anyway — if it returns empty, ask the user if you should create one.
 4. Use the returned context to ground yourself: understand architecture, known bugs, past decisions, and what was done in recent sessions. Do NOT ask the user questions that are already answered in the project context.
+5. Check the project objective with `kyp_objective_get(project)`. If none is set, ask the user for the project's main goal and save it with `kyp_objective_set(project, objective)`. Keep your work aligned to this objective.
 
 ### DURING WORK — WHEN TO SEARCH SESSIONS
 Call `kyp_session_search(query)` when:
@@ -342,6 +343,31 @@ def kyp_sessions(project: str = "", limit: int = 10) -> str:
     return "\n".join(lines)
 
 
+@mcp.tool()
+def kyp_objective_get(project: str) -> str:
+    """Get the recorded objective / main goal for a project. The objective is injected at every session start. Returns a not-set message if none exists yet."""
+    note = vault.read(f"{project}/Objective.md")
+    if not note:
+        return f"No objective set for '{project}'. Ask the user for the project's main goal, then call kyp_objective_set."
+    content = note.content.strip()
+    lines = content.split("\n")
+    if lines and lines[0].lstrip().startswith("# "):
+        content = "\n".join(lines[1:]).strip()
+    return content or f"No objective set for '{project}'."
+
+
+@mcp.tool()
+def kyp_objective_set(project: str, objective: str) -> str:
+    """Set (or replace) the objective / main goal for a project. This is injected into every future session start so work stays aligned. Call this once the user tells you what the project is for."""
+    objective = objective.strip()
+    if not objective:
+        return "Objective text is empty — nothing saved."
+    path = f"{project}/Objective.md"
+    content = f"# Objective\n\n{objective}\n"
+    vault.write_note(path, content, ["objective", project.lower().replace(" ", "-")], {})
+    return f"Objective saved for '{project}' ({path}). It will be injected at every session start."
+
+
 @mcp.tool()
 def kyp_project_context(project: str) -> str:
     """CALL THIS AT SESSION START. Returns the project's full context: Knowledge.md (ground truth), project notes, and recent session summaries. Use this to understand architecture, known bugs, past decisions, and what was done recently. This prevents hallucination and avoids repeating past work."""