wafer-cli 0.2.62__tar.gz → 0.2.64__tar.gz

{wafer_cli-0.2.62 → wafer_cli-0.2.64}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.62
+Version: 0.2.64
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{wafer_cli-0.2.62 → wafer_cli-0.2.64}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wafer-cli"
-version = "0.2.62"
+version = "0.2.64"
 description = "CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels"
 readme = "README.md"
 requires-python = ">=3.11"

{wafer_cli-0.2.62 → wafer_cli-0.2.64}/tests/test_direct_streaming.py

@@ -354,6 +354,30 @@ class TestStreamDirectEndpoint:
         assert "reading" in stderr_text
         assert any("Final answer." in s for s in stdout)
 
+    def test_sources_event_renders_to_stderr(self) -> None:
+        events = [
+            {"type": "text", "content": "Answer here."},
+            {"type": "sources", "files": ["./memory.md", "./guide.md"]},
+            {"type": "done"},
+        ]
+        _, _, stderr = self._run(events)
+        stderr_text = " ".join(stderr)
+        assert "Sources" in stderr_text
+        assert "./memory.md" in stderr_text
+        assert "./guide.md" in stderr_text
+
+    def test_json_mode_sources_event(self) -> None:
+        events = [
+            {"type": "text", "content": "Answer"},
+            {"type": "sources", "files": ["./memory.md"]},
+            {"type": "done"},
+        ]
+        _, stdout, _ = self._run(events, json_output=True)
+        json_events = [json.loads(line) for line in stdout if line.strip()]
+        sources_events = [e for e in json_events if e.get("type") == "sources"]
+        assert len(sources_events) == 1
+        assert sources_events[0]["files"] == ["./memory.md"]
+
     def test_turn_start_renders_to_stderr(self) -> None:
         """turn_start event should print turn number to stderr."""
         events = [

{wafer_cli-0.2.62 → wafer_cli-0.2.64}/wafer/wevin_cli.py

@@ -550,6 +550,17 @@ async def _stream_direct_endpoint(
                     else:
                         print(f"\033[2m  Executing {tool_name} ({index}/{total})...\033[0m", file=sys.stderr)
 
+                elif event_type == "sources":
+                    files = event.get("files", [])
+                    if json_output:
+                        assert frontend is not None
+                        frontend._emit({"type": "sources", "files": files})
+                    else:
+                        if files:
+                            print(f"\033[2m  Sources:\033[0m", file=sys.stderr)
+                            for f in files:
+                                print(f"\033[2m    {f}\033[0m", file=sys.stderr)
+
                 elif event_type == "done":
                     break
 

{wafer_cli-0.2.62 → wafer_cli-0.2.64}/wafer_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.62
+Version: 0.2.64
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown