From 47d482c1a04acba6344777eeb1e0a4b00efd0673 Mon Sep 17 00:00:00 2001
From: Marc Klingen <git@marcklingen.com>
Date: Thu, 17 Apr 2025 11:28:39 +0200
Subject: [PATCH 1/2] feat: add env filter to fetch_traces() and
 fetch_observations()

---
 langfuse/client.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/langfuse/client.py b/langfuse/client.py
index dc7ed9409..213771252 100644
--- a/langfuse/client.py
+++ b/langfuse/client.py
@@ -660,6 +660,7 @@ def fetch_traces(
         session_id: Optional[str] = None,
         from_timestamp: Optional[dt.datetime] = None,
         to_timestamp: Optional[dt.datetime] = None,
+        environment: Optional[Union[str, Sequence[str]]] = None,
         order_by: Optional[str] = None,
         tags: Optional[Union[str, Sequence[str]]] = None,
     ) -> FetchTracesResponse:
@@ -673,6 +674,7 @@ def fetch_traces(
             session_id (Optional[str]): Filter by session_id. Defaults to None.
             from_timestamp (Optional[dt.datetime]): Retrieve only traces with a timestamp on or after this datetime. Defaults to None.
             to_timestamp (Optional[dt.datetime]): Retrieve only traces with a timestamp before this datetime. Defaults to None.
+            environment (Optional[Union[str, Sequence[str]]]): Filter by environment. Defaults to None.
             order_by (Optional[str]): Format of the string `[field].[asc/desc]`. Fields: id, timestamp, name, userId, release, version, public, bookmarked, sessionId. Example: `timestamp.asc`. Defaults to None.
             tags (Optional[Union[str, Sequence[str]]]): Filter by tags. Defaults to None.
 
@@ -684,7 +686,7 @@ def fetch_traces(
         """
         try:
             self.log.debug(
-                f"Getting traces... {page}, {limit}, {name}, {user_id}, {session_id}, {from_timestamp}, {to_timestamp}, {order_by}, {tags}"
+                f"Getting traces... {page}, {limit}, {name}, {user_id}, {session_id}, {from_timestamp}, {to_timestamp}, {environment}, {order_by}, {tags}"
             )
             res = self.client.trace.list(
                 page=page,
@@ -694,6 +696,7 @@ def fetch_traces(
                 session_id=session_id,
                 from_timestamp=from_timestamp,
                 to_timestamp=to_timestamp,
+                environment=environment,
                 order_by=order_by,
                 tags=tags,
             )
@@ -768,6 +771,7 @@ def fetch_observations(
         parent_observation_id: typing.Optional[str] = None,
         from_start_time: typing.Optional[dt.datetime] = None,
         to_start_time: typing.Optional[dt.datetime] = None,
+        environment: Optional[Union[str, Sequence[str]]] = None,
         type: typing.Optional[str] = None,
     ) -> FetchObservationsResponse:
         """Get a list of observations in the current project matching the given parameters.
@@ -781,6 +785,7 @@ def fetch_observations(
             parent_observation_id (Optional[str]): Parent observation identifier. Defaults to None.
             from_start_time (Optional[dt.datetime]): Retrieve only observations with a start_time on or after this datetime. Defaults to None.
             to_start_time (Optional[dt.datetime]): Retrieve only observations with a start_time before this datetime. Defaults to None.
+            environment (Optional[Union[str, Sequence[str]]]): Filter by environment. Defaults to None.
             type (Optional[str]): Type of the observation. Defaults to None.
 
         Returns:
@@ -802,6 +807,7 @@ def fetch_observations(
                 parent_observation_id=parent_observation_id,
                 from_start_time=from_start_time,
                 to_start_time=to_start_time,
+                environment=environment,
                 type=type,
             )
             return FetchObservationsResponse(data=res.data, meta=res.meta)

From a43e858fe21b36f5f7aa64659061ee33d5cba3fc Mon Sep 17 00:00:00 2001
From: Marc Klingen <git@marcklingen.com>
Date: Thu, 17 Apr 2025 11:31:54 +0200
Subject: [PATCH 2/2] log

---
 langfuse/client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/langfuse/client.py b/langfuse/client.py
index 213771252..794d3c491 100644
--- a/langfuse/client.py
+++ b/langfuse/client.py
@@ -796,7 +796,7 @@ def fetch_observations(
         """
         try:
             self.log.debug(
-                f"Getting observations... {page}, {limit}, {name}, {user_id}, {trace_id}, {parent_observation_id}, {from_start_time}, {to_start_time}, {type}"
+                f"Getting observations... {page}, {limit}, {name}, {user_id}, {trace_id}, {parent_observation_id}, {from_start_time}, {to_start_time}, {environment}, {type}"
             )
             res = self.client.observations.get_many(
                 page=page,