arize-phoenix 3.25.0__py3-none-any.whl → 4.0.1__py3-none-any.whl

phoenix/storage/span_store/text_file.py

@@ -1,85 +0,0 @@
-import weakref
-from base64 import urlsafe_b64decode, urlsafe_b64encode
-from pathlib import Path
-from queue import SimpleQueue
-from threading import Thread
-from typing import Iterator, Optional, Tuple
-
-from opentelemetry.proto.trace.v1.trace_pb2 import TracesData
-from typing_extensions import TypeAlias
-
-from phoenix.utilities.project import get_project_name
-
-_Queue: TypeAlias = "SimpleQueue[Optional[TracesData]]"
-
-_END_OF_QUEUE = None
-_DIR_PREFIX = "project."
-
-
-class TextFileSpanStoreImpl:
-    def __init__(self, root: Path) -> None:
-        self._root = root
-        self._projects = dict(_load_projects(self._root))
-
-    def save(self, traces_data: TracesData) -> None:
-        for resource_spans in traces_data.resource_spans:
-            name = get_project_name(resource_spans.resource.attributes)
-            if (project := self._projects.get(name)) is None:
-                self._projects[name] = project = _Project(name, self._root)
-            project.save(TracesData(resource_spans=[resource_spans]))
-
-    def load(self) -> Iterator[TracesData]:
-        queue: _Queue = SimpleQueue()
-        Thread(target=self._load_traces_data, args=(queue,)).start()
-        while (item := queue.get()) is not _END_OF_QUEUE:
-            yield item
-
-    def _load_traces_data(self, queue: _Queue) -> None:
-        """Load traces data from all projects into the queue"""
-        for project in self._projects.values():
-            project.load(queue)
-        queue.put(_END_OF_QUEUE)
-
-
-class _Project:
-    def __init__(self, name: str, root: Path) -> None:
-        self._path = root / f"project.{_b64encode(name.encode())}"
-        spans_path = self._path / "spans"
-        spans_path.mkdir(parents=True, exist_ok=True)
-        self._spans = _Spans(spans_path / "spans.txt")
-
-    def save(self, traces_data: TracesData) -> None:
-        self._spans.save(traces_data)
-
-    def load(self, queue: _Queue) -> None:
-        self._spans.load(queue)
-
-
-class _Spans:
-    def __init__(self, file_path: Path):
-        self._path = file_path
-        self._file = self._path.open("a")
-        weakref.finalize(self, self._file.close)
-
-    def save(self, traces_data: TracesData) -> None:
-        self._file.write(_b64encode(traces_data.SerializeToString()))
-        self._file.write("\n")
-
-    def load(self, queue: _Queue) -> None:
-        with self._path.open("r") as f:
-            while line := f.readline():
-                queue.put(TracesData.FromString(_b64decode(line)))
-
-
-def _load_projects(root: Path) -> Iterator[Tuple[str, _Project]]:
-    for dir_path in root.glob(f"{_DIR_PREFIX}*/"):
-        name = _b64decode(dir_path.name[len(_DIR_PREFIX) :]).decode()
-        yield name, _Project(name, root)
-
-
-def _b64encode(s: bytes) -> str:
-    return urlsafe_b64encode(s).decode()
-
-
-def _b64decode(name: str) -> bytes:
-    return urlsafe_b64decode(name.encode())

phoenix/trace/dsl/missing.py

@@ -1,60 +0,0 @@
-from typing import Any
-
-
-class _Missing:
-    """
-    Falsify all comparisons except those with self; return self when getattr()
-    is called. Also, self is callable returning self. All this may seem peculiar
-    but is useful for getting the desired (and intuitive) behavior from any
-    boolean (i.e. comparative) expression without needing error handling when
-    missing values are encountered. `_Missing()` is intended to be a (fancier)
-    replacement for `None`.
-    """
-
-    def __lt__(self, _: Any) -> bool:
-        return False
-
-    def __le__(self, _: Any) -> bool:
-        return False
-
-    def __gt__(self, _: Any) -> bool:
-        return False
-
-    def __ge__(self, _: Any) -> bool:
-        return False
-
-    def __eq__(self, other: Any) -> bool:
-        return isinstance(other, _Missing)
-
-    def __ne__(self, _: Any) -> bool:
-        return False
-
-    def __len__(self) -> int:
-        return 0
-
-    def __iter__(self) -> Any:
-        return self
-
-    def __next__(self) -> Any:
-        raise StopIteration()
-
-    def __contains__(self, _: Any) -> bool:
-        return False
-
-    def __str__(self) -> str:
-        return ""
-
-    def __float__(self) -> float:
-        return float("nan")
-
-    def __bool__(self) -> bool:
-        return False
-
-    def __getattr__(self, _: Any) -> "_Missing":
-        return self
-
-    def __call__(self, *_: Any, **__: Any) -> "_Missing":
-        return self
-
-
-MISSING: Any = _Missing()