Skip to content

feat: add env filter to fetch_traces() and fetch_observations() #1166

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
marcklingen merged 2 commits into from
Apr 17, 2025
Merged

feat: add env filter to fetch_traces() and fetch_observations() #1166

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
47d482c
feat: add env filter to fetch_traces() and fetch_observations()
marcklingen Apr 17, 2025
a43e858
log
marcklingen Apr 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions langfuse/client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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.

Expand All @@ -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,
Expand All @@ -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,
)
Expand Down Expand Up @@ -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.
Expand All @@ -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:
Expand All @@ -791,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,
Expand All @@ -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)
Expand Down