arize-phoenix 3.16.1__py3-none-any.whl → 3.16.3__py3-none-any.whl

phoenix/trace/dsl/helpers.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import List, Optional, Protocol, Union, cast
 
 import pandas as pd
@@ -24,11 +25,20 @@ IS_RETRIEVER = "span_kind == 'RETRIEVER'"
 class CanQuerySpans(Protocol):
     # Implemented by phoenix.session.client.Client
     def query_spans(
-        self, *query: SpanQuery, project_name: Optional[str] = None
+        self,
+        *query: SpanQuery,
+        start_time: Optional[datetime] = None,
+        stop_time: Optional[datetime] = None,
+        project_name: Optional[str] = None,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]: ...
 
 
-def get_retrieved_documents(obj: CanQuerySpans, project_name: Optional[str] = None) -> pd.DataFrame:
+def get_retrieved_documents(
+    obj: CanQuerySpans,
+    start_time: Optional[datetime] = None,
+    stop_time: Optional[datetime] = None,
+    project_name: Optional[str] = None,
+) -> pd.DataFrame:
     project_name = project_name or get_env_project_name()
     return cast(
         pd.DataFrame,
@@ -41,12 +51,19 @@ def get_retrieved_documents(obj: CanQuerySpans, project_name: Optional[str] = No
                 reference=DOCUMENT_CONTENT,
                 document_score=DOCUMENT_SCORE,
             ),
+            start_time=start_time,
+            stop_time=stop_time,
             project_name=project_name,
         ),
     )
 
 
-def get_qa_with_reference(obj: CanQuerySpans, project_name: Optional[str] = None) -> pd.DataFrame:
+def get_qa_with_reference(
+    obj: CanQuerySpans,
+    start_time: Optional[datetime] = None,
+    stop_time: Optional[datetime] = None,
+    project_name: Optional[str] = None,
+) -> pd.DataFrame:
     project_name = project_name or get_env_project_name()
     return pd.concat(
         cast(
@@ -60,6 +77,8 @@ def get_qa_with_reference(obj: CanQuerySpans, project_name: Optional[str] = None
                     RETRIEVAL_DOCUMENTS,
                     reference=DOCUMENT_CONTENT,
                 ),
+                start_time=start_time,
+                stop_time=stop_time,
                 project_name=project_name,
             ),
         ),

phoenix/trace/dsl/query.py

@@ -14,6 +14,7 @@ from typing import (
     Mapping,
     Optional,
     Sequence,
+    Sized,
     Tuple,
     cast,
 )
@@ -125,7 +126,7 @@ class Explosion(Projection):
         return replace(self, primary_index_key=primary_index_key)
 
     def __call__(self, span: Span) -> Iterator[Dict[str, Any]]:
-        if not isinstance(seq := self.value(span), Sequence):
+        if not isinstance(seq := self.value(span), Iterable):
             return
         has_mapping = False
         for item in seq:
@@ -193,7 +194,7 @@ class Concatenation(Projection):
         return replace(self, separator=separator)
 
     def __call__(self, span: Span) -> Iterator[Tuple[str, str]]:
-        if not isinstance(seq := self.value(span), Sequence):
+        if not isinstance(seq := self.value(span), Iterable):
             return
         if not self.kwargs:
             yield self.key, self.separator.join(map(str, seq))
@@ -282,12 +283,12 @@ class SpanQuery:
             spans = filter(self._filter, spans)
         if self._explode:
             spans = filter(
-                lambda span: (isinstance(seq := self._explode.value(span), Sequence) and len(seq)),
+                lambda span: (isinstance(seq := self._explode.value(span), Sized) and len(seq)),
                 spans,
             )
         if self._concat:
             spans = filter(
-                lambda span: (isinstance(seq := self._concat.value(span), Sequence) and len(seq)),
+                lambda span: (isinstance(seq := self._concat.value(span), Sized) and len(seq)),
                 spans,
             )
         if not (self._select or self._explode or self._concat):

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.16.1"
+__version__ = "3.16.3"