arize-phoenix 4.11.0__tar.gz → 4.12.0__tar.gz

{arize_phoenix-4.11.0 → arize_phoenix-4.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 4.11.0
+Version: 4.12.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-4.11.0 → arize_phoenix-4.12.0}/src/phoenix/session/client.py

@@ -51,6 +51,9 @@ from phoenix.trace.otel import encode_span_to_otlp
 
 logger = logging.getLogger(__name__)
 
+
+DEFAULT_TIMEOUT_IN_SECONDS = 5
+
 DatasetAction: TypeAlias = Literal["create", "append"]
 
 
@@ -120,6 +123,7 @@ class Client(TraceDataExtractor):
         project_name: Optional[str] = None,
         # Deprecated
         stop_time: Optional[datetime] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
         """
         Queries spans from the Phoenix server or active session based on specified criteria.
@@ -159,6 +163,7 @@ class Client(TraceDataExtractor):
                 "limit": limit,
                 "root_spans_only": root_spans_only,
             },
+            timeout=timeout,
         )
         if response.status_code == 404:
             logger.info("No spans found.")

{arize_phoenix-4.11.0 → arize_phoenix-4.12.0}/src/phoenix/session/data_extractor.py

@@ -9,6 +9,7 @@ from phoenix.trace.dsl import SpanQuery
 from phoenix.trace.trace_dataset import TraceDataset
 
 DEFAULT_SPAN_LIMIT = 1000
+DEFAULT_TIMEOUT_IN_SECONDS = 5
 
 
 class TraceDataExtractor(ABC):
@@ -26,6 +27,7 @@ class TraceDataExtractor(ABC):
         limit: Optional[int] = DEFAULT_SPAN_LIMIT,
         root_spans_only: Optional[bool] = None,
         project_name: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]: ...
 
     def get_spans_dataframe(
@@ -37,6 +39,7 @@ class TraceDataExtractor(ABC):
         limit: Optional[int] = DEFAULT_SPAN_LIMIT,
         root_spans_only: Optional[bool] = None,
         project_name: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[pd.DataFrame]:
         return cast(
             Optional[pd.DataFrame],
@@ -47,6 +50,7 @@ class TraceDataExtractor(ABC):
                 limit=limit,
                 root_spans_only=root_spans_only,
                 project_name=project_name,
+                timeout=timeout,
             ),
         )
 
@@ -59,8 +63,23 @@ class TraceDataExtractor(ABC):
     def get_trace_dataset(
         self,
         project_name: Optional[str] = None,
+        *,
+        start_time: Optional[datetime] = None,
+        end_time: Optional[datetime] = None,
+        limit: Optional[int] = DEFAULT_SPAN_LIMIT,
+        root_spans_only: Optional[bool] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[TraceDataset]:
-        if (dataframe := self.get_spans_dataframe(project_name=project_name)) is None:
+        if (
+            dataframe := self.get_spans_dataframe(
+                project_name=project_name,
+                start_time=start_time,
+                end_time=end_time,
+                limit=limit,
+                root_spans_only=root_spans_only,
+                timeout=timeout,
+            )
+        ) is None:
             return None
         evaluations = self.get_evaluations(project_name=project_name)
         return TraceDataset(dataframe=dataframe, evaluations=evaluations)

{arize_phoenix-4.11.0 → arize_phoenix-4.12.0}/src/phoenix/session/session.py

@@ -75,6 +75,9 @@ global _session_working_dir
 _session_working_dir: Optional["TemporaryDirectory[str]"] = None
 
 
+DEFAULT_TIMEOUT_IN_SECONDS = 5
+
+
 class NotebookEnvironment(Enum):
     COLAB = "colab"
     LOCAL = "local"
@@ -152,6 +155,7 @@ class Session(TraceDataExtractor, ABC):
         project_name: Optional[str] = None,
         # Deprecated fields
         stop_time: Optional[datetime] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
         """
         Queries the spans in the project based on the provided parameters.

arize_phoenix-4.12.0/src/phoenix/version.py

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

arize_phoenix-4.11.0/src/phoenix/version.py

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