chore: import upstream snapshot with attribution

python/tests/core/langfuse/__init__.py

@@ -0,0 +1,13 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

python/tests/core/langfuse/test_read.py

@@ -0,0 +1,136 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+import json
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.accessor.langfuse import LangfuseAccessor
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.langfuse.read import read
+from mirage.resource.langfuse.config import LangfuseConfig
+from mirage.types import PathSpec
+
+
+@pytest.fixture
+def accessor():
+    config = LangfuseConfig(
+        public_key="pk-test",
+        secret_key="sk-test",
+    )
+    with patch("mirage.accessor.langfuse.Langfuse"):
+        return LangfuseAccessor(config=config)
+
+
+@pytest.fixture
+def index():
+    return RAMIndexCacheStore()
+
+
+@pytest.mark.asyncio
+async def test_read_trace(accessor, index):
+    trace_data = {"id": "abc123", "name": "chat", "input": "hello"}
+    with patch(
+            "mirage.core.langfuse.read.fetch_trace",
+            new_callable=AsyncMock,
+            return_value=trace_data,
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="traces/abc123.json",
+                     virtual="/traces/abc123.json",
+                     directory="/traces/abc123.json"), index)
+
+    parsed = json.loads(result)
+    assert parsed["id"] == "abc123"
+    assert parsed["name"] == "chat"
+
+
+@pytest.mark.asyncio
+async def test_read_prompt_version(accessor, index):
+    prompt_data = {"name": "summarize", "version": 1, "prompt": "Summarize:"}
+    with patch(
+            "mirage.core.langfuse.read.fetch_prompt",
+            new_callable=AsyncMock,
+            return_value=prompt_data,
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="prompts/summarize/1.json",
+                     virtual="/prompts/summarize/1.json",
+                     directory="/prompts/summarize/1.json"), index)
+
+    parsed = json.loads(result)
+    assert parsed["name"] == "summarize"
+    assert parsed["version"] == 1
+
+
+@pytest.mark.asyncio
+async def test_read_dataset_items(accessor, index):
+    items = [
+        {
+            "id": "item1",
+            "input": "q1",
+            "output": "a1"
+        },
+        {
+            "id": "item2",
+            "input": "q2",
+            "output": "a2"
+        },
+    ]
+    with patch(
+            "mirage.core.langfuse.read.fetch_dataset_items",
+            new_callable=AsyncMock,
+            return_value=items,
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="datasets/qa-eval/items.jsonl",
+                     virtual="/datasets/qa-eval/items.jsonl",
+                     directory="/datasets/qa-eval/items.jsonl"), index)
+
+    lines = result.decode().strip().split("\n")
+    assert len(lines) == 2
+    assert json.loads(lines[0])["id"] == "item1"
+    assert json.loads(lines[1])["id"] == "item2"
+
+
+@pytest.mark.asyncio
+async def test_read_invalid_path_raises(accessor, index):
+    with pytest.raises(FileNotFoundError):
+        await read(
+            accessor,
+            PathSpec(resource_path="not_a_valid_path",
+                     virtual="/not_a_valid_path",
+                     directory="/not_a_valid_path"), index)
+
+
+@pytest.mark.asyncio
+async def test_read_session_trace(accessor, index):
+    trace_data = {"id": "tid1", "session_id": "sid1"}
+    with patch(
+            "mirage.core.langfuse.read.fetch_trace",
+            new_callable=AsyncMock,
+            return_value=trace_data,
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="sessions/sid1/tid1.json",
+                     virtual="/sessions/sid1/tid1.json",
+                     directory="/sessions/sid1/tid1.json"), index)
+
+    parsed = json.loads(result)
+    assert parsed["id"] == "tid1"

python/tests/core/langfuse/test_readdir.py

@@ -0,0 +1,171 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.accessor.langfuse import LangfuseAccessor
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.langfuse.readdir import readdir
+from mirage.resource.langfuse.config import LangfuseConfig
+from mirage.types import PathSpec
+
+
+@pytest.fixture
+def accessor():
+    config = LangfuseConfig(
+        public_key="pk-test",
+        secret_key="sk-test",
+    )
+    with patch("mirage.accessor.langfuse.Langfuse"):
+        return LangfuseAccessor(config=config)
+
+
+@pytest.fixture
+def index():
+    return RAMIndexCacheStore()
+
+
+@pytest.mark.asyncio
+async def test_readdir_root(accessor, index):
+    result = await readdir(
+        accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+        index)
+    assert result == ["/traces", "/sessions", "/prompts", "/datasets"]
+
+
+@pytest.mark.asyncio
+async def test_readdir_traces(accessor, index):
+    with patch(
+            "mirage.core.langfuse.readdir.fetch_traces",
+            new_callable=AsyncMock,
+            return_value=[
+                {
+                    "id": "abc123",
+                    "name": "chat"
+                },
+                {
+                    "id": "def456",
+                    "name": "search"
+                },
+            ],
+    ):
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="traces",
+                     virtual="/traces",
+                     directory="/traces"), index)
+
+    assert "/traces/abc123.json" in result
+    assert "/traces/def456.json" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_sessions(accessor, index):
+    with patch(
+            "mirage.core.langfuse.readdir.fetch_sessions",
+            new_callable=AsyncMock,
+            return_value=[{
+                "id": "session-1"
+            }, {
+                "id": "session-2"
+            }],
+    ):
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="sessions",
+                     virtual="/sessions",
+                     directory="/sessions"), index)
+
+    assert "/sessions/session-1" in result
+    assert "/sessions/session-2" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_prompts(accessor, index):
+    with patch(
+            "mirage.core.langfuse.readdir.fetch_prompts",
+            new_callable=AsyncMock,
+            return_value=[
+                {
+                    "name": "summarize",
+                    "version": 1
+                },
+                {
+                    "name": "translate",
+                    "version": 1
+                },
+            ],
+    ):
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="prompts",
+                     virtual="/prompts",
+                     directory="/prompts"), index)
+
+    assert "/prompts/summarize" in result
+    assert "/prompts/translate" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_datasets(accessor, index):
+    with patch(
+            "mirage.core.langfuse.readdir.fetch_datasets",
+            new_callable=AsyncMock,
+            return_value=[{
+                "name": "qa-eval"
+            }, {
+                "name": "chat-eval"
+            }],
+    ):
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="datasets",
+                     virtual="/datasets",
+                     directory="/datasets"), index)
+
+    assert "/datasets/qa-eval" in result
+    assert "/datasets/chat-eval" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_dataset_contents(accessor, index):
+    result = await readdir(
+        accessor,
+        PathSpec(resource_path="datasets/qa-eval",
+                 virtual="/datasets/qa-eval",
+                 directory="/datasets/qa-eval"), index)
+    assert "/datasets/qa-eval/items.jsonl" in result
+    assert "/datasets/qa-eval/runs" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_dotfile_raises(accessor, index):
+    with pytest.raises(FileNotFoundError):
+        await readdir(
+            accessor,
+            PathSpec(resource_path=".hidden",
+                     virtual="/.hidden",
+                     directory="/.hidden"), index)
+
+
+@pytest.mark.asyncio
+async def test_readdir_dotfile_nested_raises(accessor, index):
+    with pytest.raises(FileNotFoundError):
+        await readdir(
+            accessor,
+            PathSpec(resource_path="traces/.DS_Store",
+                     virtual="/traces/.DS_Store",
+                     directory="/traces/.DS_Store"), index)

python/tests/core/langfuse/test_scope.py

@@ -0,0 +1,127 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.core.langfuse.scope import detect_scope
+from mirage.types import PathSpec
+from mirage.utils.key_prefix import mount_key
+
+
+def test_root_path():
+    scope = detect_scope("/")
+    assert scope.level == "root"
+
+
+def test_traces_dir():
+    scope = detect_scope("/traces")
+    assert scope.level == "traces"
+    assert scope.resource_type == "traces"
+
+
+def test_traces_file():
+    scope = detect_scope("/traces/abc.json")
+    assert scope.level == "file"
+    assert scope.resource_type == "traces"
+    assert scope.resource_id == "abc"
+
+
+def test_sessions_dir():
+    scope = detect_scope("/sessions")
+    assert scope.level == "sessions"
+    assert scope.resource_type == "sessions"
+
+
+def test_sessions_id():
+    scope = detect_scope("/sessions/sid1")
+    assert scope.level == "sessions"
+    assert scope.resource_type == "sessions"
+    assert scope.resource_id == "sid1"
+
+
+def test_sessions_trace_file():
+    scope = detect_scope("/sessions/sid1/tid1.json")
+    assert scope.level == "file"
+    assert scope.resource_type == "sessions"
+    assert scope.resource_id == "sid1"
+
+
+def test_prompts_dir():
+    scope = detect_scope("/prompts")
+    assert scope.level == "prompts"
+    assert scope.resource_type == "prompts"
+
+
+def test_prompts_name():
+    scope = detect_scope("/prompts/summarize")
+    assert scope.level == "prompts"
+    assert scope.resource_type == "prompts"
+    assert scope.resource_id == "summarize"
+
+
+def test_prompts_version_file():
+    scope = detect_scope("/prompts/summarize/1.json")
+    assert scope.level == "file"
+    assert scope.resource_type == "prompts"
+    assert scope.resource_id == "summarize"
+
+
+def test_datasets_dir():
+    scope = detect_scope("/datasets")
+    assert scope.level == "datasets"
+    assert scope.resource_type == "datasets"
+
+
+def test_datasets_name():
+    scope = detect_scope("/datasets/qa-eval")
+    assert scope.level == "datasets"
+    assert scope.resource_type == "datasets"
+    assert scope.resource_id == "qa-eval"
+
+
+def test_glob_scope_root():
+    gs = PathSpec(
+        resource_path=mount_key("/langfuse/", "/langfuse"),
+        virtual="/langfuse/",
+        directory="/langfuse/",
+        pattern=None,
+        resolved=False,
+    )
+    scope = detect_scope(gs)
+    assert scope.level == "root"
+
+
+def test_glob_scope_traces():
+    gs = PathSpec(
+        resource_path=mount_key("/langfuse/traces", "/langfuse"),
+        virtual="/langfuse/traces",
+        directory="/langfuse/",
+        pattern=None,
+        resolved=False,
+    )
+    scope = detect_scope(gs)
+    assert scope.level == "traces"
+    assert scope.resource_type == "traces"
+
+
+def test_glob_scope_file():
+    gs = PathSpec(
+        resource_path=mount_key("/langfuse/traces/abc.json", "/langfuse"),
+        virtual="/langfuse/traces/abc.json",
+        directory="/langfuse/traces/",
+        pattern="*.json",
+        resolved=True,
+    )
+    scope = detect_scope(gs)
+    assert scope.level == "file"
+    assert scope.resource_type == "traces"
+    assert scope.resource_id == "abc"

python/tests/core/langfuse/test_stat.py

@@ -0,0 +1,123 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from unittest.mock import patch
+
+import pytest
+
+from mirage.accessor.langfuse import LangfuseAccessor
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.langfuse.stat import stat
+from mirage.resource.langfuse.config import LangfuseConfig
+from mirage.types import FileType, PathSpec
+
+
+@pytest.fixture
+def accessor():
+    config = LangfuseConfig(
+        public_key="pk-test",
+        secret_key="sk-test",
+    )
+    with patch("mirage.accessor.langfuse.Langfuse"):
+        return LangfuseAccessor(config=config)
+
+
+@pytest.fixture
+def index():
+    return RAMIndexCacheStore()
+
+
+@pytest.mark.asyncio
+async def test_stat_root(accessor, index):
+    result = await stat(accessor,
+                        PathSpec(resource_path="", virtual="/", directory="/"),
+                        index)
+    assert result.type == FileType.DIRECTORY
+    assert result.name == "/"
+
+
+@pytest.mark.asyncio
+async def test_stat_traces_dir(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="traces",
+                 virtual="/traces",
+                 directory="/traces"), index)
+    assert result.type == FileType.DIRECTORY
+    assert result.name == "traces"
+
+
+@pytest.mark.asyncio
+async def test_stat_trace_file(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="traces/abc.json",
+                 virtual="/traces/abc.json",
+                 directory="/traces/abc.json"), index)
+    assert result.type == FileType.JSON
+    assert result.name == "abc.json"
+
+
+@pytest.mark.asyncio
+async def test_stat_session_dir(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="sessions/sid1",
+                 virtual="/sessions/sid1",
+                 directory="/sessions/sid1"), index)
+    assert result.type == FileType.DIRECTORY
+    assert result.extra["session_id"] == "sid1"
+
+
+@pytest.mark.asyncio
+async def test_stat_prompt_version_file(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="prompts/summarize/1.json",
+                 virtual="/prompts/summarize/1.json",
+                 directory="/prompts/summarize/1.json"), index)
+    assert result.type == FileType.JSON
+    assert result.name == "1.json"
+
+
+@pytest.mark.asyncio
+async def test_stat_dataset_items(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="datasets/qa-eval/items.jsonl",
+                 virtual="/datasets/qa-eval/items.jsonl",
+                 directory="/datasets/qa-eval/items.jsonl"), index)
+    assert result.type == FileType.TEXT
+    assert result.name == "items.jsonl"
+
+
+@pytest.mark.asyncio
+async def test_stat_dotfile_raises(accessor, index):
+    with pytest.raises(FileNotFoundError):
+        await stat(
+            accessor,
+            PathSpec(resource_path=".hidden",
+                     virtual="/.hidden",
+                     directory="/.hidden"), index)
+
+
+@pytest.mark.asyncio
+async def test_stat_dataset_runs_dir(accessor, index):
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="datasets/qa-eval/runs",
+                 virtual="/datasets/qa-eval/runs",
+                 directory="/datasets/qa-eval/runs"), index)
+    assert result.type == FileType.DIRECTORY
+    assert result.name == "runs"