Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
7 changes: 2 additions & 5 deletions sentry_sdk/integrations/litellm.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ def _input_callback(kwargs: "Dict[str, Any]") -> None:
record_inputs = False
if has_data_collection_enabled(client.options):
record_inputs = client.options["data_collection"]["gen_ai"]["inputs"]
elif should_send_default_pii() and integration.include_prompts:
elif should_send_default_pii():
record_inputs = True

if record_inputs:
Expand Down Expand Up @@ -198,7 +198,7 @@ def _success_callback(
record_outputs = False
if has_data_collection_enabled(client.options):
record_outputs = client.options["data_collection"]["gen_ai"]["outputs"]
elif should_send_default_pii() and integration.include_prompts:
elif should_send_default_pii():
record_outputs = True

if record_outputs:
Expand Down Expand Up @@ -336,9 +336,6 @@ class LiteLLMIntegration(Integration):
identifier = "litellm"
origin = f"auto.ai.{identifier}"

def __init__(self: "LiteLLMIntegration", include_prompts: bool = True) -> None:
self.include_prompts = include_prompts

@staticmethod
def setup_once() -> None:
"""Set up LiteLLM callbacks for monitoring."""
Expand Down
Loading
Loading