dyno-phi 0.2.2__tar.gz → 0.2.3__tar.gz

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/.claude/skills/phi/SKILL.md

@@ -63,8 +63,6 @@ phi research \
   --dataset-id <ID>
 
 # Stream results live
-phi research --question "..." --stream
-
 # Append notes to a local file and cloud storage
 phi research --question "..." --notes-file ./research.md --dataset-id <ID>
 

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/CHANGELOG.md

@@ -7,6 +7,19 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.2.3] - 2026-03-17
+
+### Added
+- `phi research` now accepts `--dataset_id` (underscore) as an alias for `--dataset-id` — consistent with all other commands
+- `phi notes` command implemented: fetches and displays accumulated research notes for a dataset
+
+### Changed
+- `phi design` (`rfdiffusion3`) and `phi boltzgen` no longer appear in `phi --help` or usage line — they remain accessible internally via `DYNO_ENABLE_DESIGN=1`
+
+### Fixed
+- Missing imports (`_request`, `PhiApiError`, `Panel`, `Markdown`, `_C_BLUE`, `_C_SAND`, `json`) in `phi notes` command
+- mypy `Cannot assign to a method` error replaced with a proper `_FilteredSubParsersAction` subclass
+
 ## [0.2.2] - 2026-03-17
 
 ### Added

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/CLI-REFERENCE.md

@@ -679,7 +679,6 @@ phi research --question QUESTION [options]
 | `--dataset-id ID` | — | Associate notes with a dataset and sync to cloud storage |
 | `--notes-file FILE` | `./research.md` | Local append-only notes file |
 | `--no-save` | — | Skip saving the report to the local notes file |
-| `--stream` | — | Stream results live via SSE (skips job tracking) |
 
 **Examples:**
 ```bash

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyno-phi
-Version: 0.2.2
+Version: 0.2.3
 Summary: Phi CLI and biomodals for the dyno protein design platform
 Project-URL: Homepage, https://dynotx.com
 Project-URL: Repository, https://github.com/dynotx/phi-cli

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/phi-plugin/skills/phi/SKILL.md

@@ -63,8 +63,6 @@ phi research \
   --dataset-id <ID>
 
 # Stream results live
-phi research --question "..." --stream
-
 # Append notes to a local file and cloud storage
 phi research --question "..." --notes-file ./research.md --dataset-id <ID>
 

dyno_phi-0.2.3/src/phi/_version.py

@@ -0,0 +1 @@
+__version__ = "0.2.3"

dyno_phi-0.2.3/src/phi/commands/research.py

@@ -0,0 +1,90 @@
+from __future__ import annotations
+
+import argparse
+import json
+import os
+from pathlib import Path
+
+from rich.markdown import Markdown
+from rich.panel import Panel
+
+from phi.api import _request
+from phi.display import _C_BLUE, _C_ROSE, _C_SAND, _die, console
+from phi.research import _stream_research
+from phi.types import PhiApiError
+
+_ERROR_REPORT_PREFIXES = ("# Research Failed", "# Error", "Error: Command failed")
+
+
+def _is_error_report(report: str) -> bool:
+    stripped = report.strip()
+    return any(stripped.startswith(prefix) for prefix in _ERROR_REPORT_PREFIXES)
+
+
+def cmd_research(args: argparse.Namespace) -> None:
+    notes_file: Path | None = None if args.no_save else Path(args.notes_file)
+    dataset_id: str | None = args.dataset_id
+
+    stream_url = os.environ.get("DYNO_RESEARCH_STREAM_URL") or args.url
+
+    context: str | None = None
+    if args.context_file:
+        try:
+            prior = Path(args.context_file).read_text(encoding="utf-8")
+            existing = args.context or ""
+            context = (prior + "\n\n" + existing).strip() if existing else prior
+        except Exception as exc:
+            console.print(f"[{_C_ROSE}]Warning: could not read context file:[/] {exc}")
+    elif args.context:
+        context = args.context
+
+    _stream_research(
+        question=args.question,
+        base_url=stream_url,
+        notes_file=notes_file,
+        dataset_id=dataset_id,
+        context=context,
+    )
+
+
+def cmd_notes(args: argparse.Namespace) -> None:
+    dataset_id: str = args.dataset_id
+
+    try:
+        data = _request("GET", f"/datasets/{dataset_id}/research-notes")
+    except PhiApiError as exc:
+        _die(str(exc))
+
+    if not data.get("exists"):
+        console.print(f"[{_C_ROSE}]No research notes found[/] for dataset {dataset_id}")
+        return
+
+    content: str = data.get("content") or ""
+    data.get("gcs_url")
+    data.get("gcs_uri")
+
+    if args.out:
+        out = Path(args.out)
+        if out.suffix.lower() == ".md":
+            dest = out
+        else:
+            out.mkdir(parents=True, exist_ok=True)
+            dest = out / "research.md"
+        dest.parent.mkdir(parents=True, exist_ok=True)
+        dest.write_text(content, encoding="utf-8")
+        console.print(f"[{_C_SAND}]Notes saved[/] → {dest}")
+        return
+
+    if args.json:
+        print(json.dumps(data, indent=2))
+        return
+
+    console.print()
+    console.print(
+        Panel(
+            Markdown(content),
+            title=f"[{_C_BLUE}]Research Notes[/] — {dataset_id[:8]}…",
+            border_style=_C_BLUE,
+            padding=(1, 2),
+        )
+    )

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/src/phi/parser.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import argparse
 
 from phi._version import __version__
@@ -44,6 +46,8 @@ def _add_fasta_args(p: argparse.ArgumentParser) -> None:
     g.add_argument("--fasta-str", metavar="FASTA", help="FASTA content as a string (for scripting)")
     g.add_argument(
         "--dataset-id",
+        "--dataset_id",
+        dest="dataset_id",
         metavar="DATASET_ID",
         help="Pre-ingested dataset ID (for batch runs of 100–50,000 files)",
     )
@@ -55,6 +59,8 @@ def _add_pdb_args(p: argparse.ArgumentParser) -> None:
     g.add_argument("--pdb-gcs", metavar="URI", help="Cloud storage URI to PDB (gs://…)")
     g.add_argument(
         "--dataset-id",
+        "--dataset_id",
+        dest="dataset_id",
         metavar="DATASET_ID",
         help="Pre-ingested dataset ID (for batch runs of 100–50,000 files)",
     )
@@ -233,6 +239,13 @@ def _add_boltzgen_args(p: argparse.ArgumentParser) -> None:
     )
 
 
+class _FilteredSubParsersAction(argparse._SubParsersAction):
+    """Subparser action that omits choices with help=SUPPRESS from help output."""
+
+    def _get_subactions(self) -> list[argparse.Action]:
+        return [a for a in self._choices_actions if a.help is not argparse.SUPPRESS]
+
+
 def build_parser() -> argparse.ArgumentParser:
     root = argparse.ArgumentParser(
         prog="phi",
@@ -252,6 +265,7 @@ def build_parser() -> argparse.ArgumentParser:
         metavar="S",
         help=f"Seconds between status-poll requests (default: {POLL_INTERVAL})",
     )
+    root.register("action", "parsers", _FilteredSubParsersAction)
     sub = root.add_subparsers(dest="command", required=True)
 
     p = sub.add_parser(
@@ -427,14 +441,14 @@ def build_parser() -> argparse.ArgumentParser:
     p = sub.add_parser(
         "rfdiffusion3",
         aliases=["design"],
-        help="All-atom backbone generation: binder design, de novo, motif scaffolding (2–5 min/design)",
+        help=argparse.SUPPRESS,
     )
     _add_rfdiffusion3_args(p)
     _add_job_args(p)
 
     p = sub.add_parser(
         "boltzgen",
-        help="All-atom generative design from a YAML spec — proteins, peptides, small molecules (10–20 min)",
+        help=argparse.SUPPRESS,
     )
     _add_boltzgen_args(p)
     _add_job_args(p)
@@ -481,7 +495,9 @@ def build_parser() -> argparse.ArgumentParser:
     )
     p.add_argument(
         "--dataset-id",
+        "--dataset_id",
         metavar="ID",
+        dest="dataset_id",
         help="Associate notes with a dataset and sync to cloud storage",
     )
     p.add_argument(
@@ -494,17 +510,12 @@ def build_parser() -> argparse.ArgumentParser:
         "--no-save", action="store_true", help="Skip saving the report to the local notes file"
     )
     p.add_argument(
-        "--stream",
-        action="store_true",
-        help="Stream results live from Modal SSE endpoint (skips job tracking)",
-    )
-    p.add_argument(
-        "--stream-url",
+        "--url",
         metavar="URL",
         default="https://dynotx--research-agent-streaming-fastapi-app.modal.run",
-        help="Override the SSE base URL",
+        help="Override the SSE streaming base URL",
     )
-    _add_job_args(p)
+    p.add_argument("--out", metavar="DIR", help="Save report and citations to DIR")
 
     p = sub.add_parser("notes", help="View accumulated research notes for a dataset")
     p.add_argument(
@@ -559,6 +570,8 @@ def build_parser() -> argparse.ArgumentParser:
     )
     p.add_argument(
         "--dataset-id",
+        "--dataset_id",
+        dest="dataset_id",
         default=None,
         metavar="ID",
         help="Dataset of PDB/CIF designs (default: cached from last upload or `phi use`)",
@@ -652,4 +665,8 @@ def build_parser() -> argparse.ArgumentParser:
     p.add_argument("--out", metavar="FILE", help="Save scores CSV to file")
     p.add_argument("--json", action="store_true", help="Output raw JSON")
 
+    # Hide internal/unreleased commands from the usage line.
+    _hidden_cmds = {"rfdiffusion3", "design", "boltzgen"}
+    sub.metavar = "{" + ",".join(c for c in sub.choices if c not in _hidden_cmds) + "}"
+
     return root

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/src/phi/research.py

@@ -150,13 +150,14 @@ def _stream_research(
     max_turns: int = 15,
     notes_file: Path | None = None,
     dataset_id: str | None = None,
+    context: str | None = None,
 ) -> None:
-    url = f"{base_url.rstrip('/')}/research/stream?" + urllib.parse.urlencode(
-        {"question": question, "max_turns": max_turns}
-    )
+    params: dict = {"question": question, "max_turns": max_turns}
+    if context:
+        params["context"] = context
+    url = f"{base_url.rstrip('/')}/research/stream?" + urllib.parse.urlencode(params)
 
-    console.print(f"\n[{_C_BLUE}]Connecting to research stream…[/]")
-    console.print(f"[dim]{url}[/]\n")
+    console.print(f"\n[{_C_BLUE}]Connecting to research stream…[/]\n")
 
     state = _ResearchStreamState()
     req = urllib.request.Request(url, headers={"Accept": "text/event-stream"})

{dyno_phi-0.2.2 → dyno_phi-0.2.3}/tests/test_research.py

@@ -3,6 +3,7 @@ from pathlib import Path
 
 sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
 
+from phi.commands.research import _is_error_report
 from phi.research import (
     _build_research_section,
     _dispatch_sse_event,
@@ -85,6 +86,31 @@ def test_build_research_section_includes_citations() -> None:
     assert "Sources (2)" in section
 
 
+def test_is_error_report_detects_research_failed() -> None:
+    report = "# Research Failed\n\nError: Command failed with exit code 1"
+    assert _is_error_report(report) is True
+
+
+def test_is_error_report_detects_error_header() -> None:
+    assert _is_error_report("# Error\n\nSomething went wrong") is True
+
+
+def test_is_error_report_detects_bare_error_prefix() -> None:
+    assert _is_error_report("Error: Command failed with exit code 1") is True
+
+
+def test_is_error_report_passes_valid_report() -> None:
+    assert _is_error_report("# AAV Biology\n\nAAVs are...") is False
+
+
+def test_is_error_report_passes_empty() -> None:
+    assert _is_error_report("") is False
+
+
+def test_is_error_report_handles_leading_whitespace() -> None:
+    assert _is_error_report("  # Research Failed\n\nError") is True  # strip() normalises whitespace
+
+
 def test_build_research_section_handles_invalid_citations() -> None:
     section = _build_research_section("query", "report", "not-valid-json")
     assert "query" in section

dyno_phi-0.2.2/src/phi/_version.py

@@ -1 +0,0 @@
-__version__ = "0.2.2"

dyno_phi-0.2.2/src/phi/commands/research.py

@@ -1,141 +0,0 @@
-from __future__ import annotations
-
-import argparse
-import json
-from pathlib import Path
-
-from rich.markdown import Markdown
-from rich.panel import Panel
-
-import phi.config as config
-from phi.api import _request, _submit
-from phi.display import _C_BLUE, _C_ROSE, _C_SAND, _die, _print_status, _print_submission, console
-from phi.download import _download_job
-from phi.polling import _poll
-from phi.research import _append_research_notes, _stream_research
-from phi.types import PhiApiError
-
-
-def cmd_research(args: argparse.Namespace) -> None:
-    notes_file: Path | None = None if args.no_save else Path(args.notes_file)
-    dataset_id: str | None = args.dataset_id
-
-    if args.stream:
-        _stream_research(
-            question=args.question,
-            base_url=args.stream_url,
-            notes_file=notes_file,
-            dataset_id=dataset_id,
-        )
-        return
-
-    params: dict = {
-        "question": args.question,
-        "databases": [d.strip() for d in args.databases.split(",") if d.strip()],
-        "max_papers": args.max_papers,
-        "include_structures": args.structures,
-    }
-    if args.target:
-        params["target"] = args.target
-
-    if args.context_file:
-        try:
-            prior = Path(args.context_file).read_text(encoding="utf-8")
-            existing = args.context or ""
-            params["context"] = (prior + "\n\n" + existing).strip() if existing else prior
-        except Exception as exc:
-            console.print(f"[{_C_ROSE}]Warning: could not read context file:[/] {exc}")
-    elif args.context:
-        params["context"] = args.context
-
-    result = _submit("research", params, run_id=args.run_id)
-    _print_submission(result)
-    if args.wait:
-        console.print(f"\n[dim]Polling every {config.POLL_INTERVAL}s …[/]")
-        final = _poll(result["job_id"])
-        _print_status(final)
-
-        report = (final.get("outputs") or {}).get("report_md")
-        citations_raw = (final.get("outputs") or {}).get("citations")
-
-        if report:
-            if notes_file is not None:
-                _append_research_notes(
-                    notes_file=notes_file,
-                    question=args.question,
-                    report=report,
-                    citations_raw=citations_raw,
-                    dataset_id=dataset_id,
-                )
-
-            if args.out:
-                out = Path(args.out)
-                out.mkdir(parents=True, exist_ok=True)
-                (out / "research_report.md").write_text(report)
-                console.print(f"\n[{_C_SAND}]Report written[/] → {out}/research_report.md")
-                if citations_raw:
-                    try:
-                        citations = json.loads(citations_raw)
-                        (out / "citations.json").write_text(json.dumps(citations, indent=2))
-                        console.print(f"[{_C_SAND}]Citations written[/] → {out}/citations.json")
-                    except Exception:
-                        pass
-            else:
-                console.print("\n" + "─" * 60)
-                console.print(report)
-                if citations_raw:
-                    try:
-                        citations = json.loads(citations_raw)
-                        console.print(f"\n[{_C_BLUE}]Sources ({len(citations)})[/]")
-                        for i, src in enumerate(citations[:10], 1):
-                            title = src.get("title") or src.get("pmid") or str(src)
-                            console.print(f"  [{_C_BLUE}]{i}.[/] {title}")
-                        if len(citations) > 10:
-                            console.print(f"  [dim]… and {len(citations) - 10} more[/]")
-                    except Exception:
-                        pass
-        elif args.out and final.get("status") == "completed":
-            _download_job(final, args.out)
-
-
-def cmd_notes(args: argparse.Namespace) -> None:
-    dataset_id: str = args.dataset_id
-
-    try:
-        data = _request("GET", f"/datasets/{dataset_id}/research-notes")
-    except PhiApiError as exc:
-        _die(str(exc))
-
-    if not data.get("exists"):
-        console.print(f"[{_C_ROSE}]No research notes found[/] for dataset {dataset_id}")
-        return
-
-    content: str = data.get("content") or ""
-    data.get("gcs_url")
-    data.get("gcs_uri")
-
-    if args.out:
-        out = Path(args.out)
-        if out.suffix.lower() == ".md":
-            dest = out
-        else:
-            out.mkdir(parents=True, exist_ok=True)
-            dest = out / "research.md"
-        dest.parent.mkdir(parents=True, exist_ok=True)
-        dest.write_text(content, encoding="utf-8")
-        console.print(f"[{_C_SAND}]Notes saved[/] → {dest}")
-        return
-
-    if args.json:
-        print(json.dumps(data, indent=2))
-        return
-
-    console.print()
-    console.print(
-        Panel(
-            Markdown(content),
-            title=f"[{_C_BLUE}]Research Notes[/] — {dataset_id[:8]}…",
-            border_style=_C_BLUE,
-            padding=(1, 2),
-        )
-    )