PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
google_genai
/
__pycache__
� o0i�, � � � d dl mZ d dlmZmZmZmZmZ d dlZd dl m Z d dlmZm Z d dlmZmZ d dlmZ d dlmZmZ d dlmZmZmZ d dlmZmZmZmZ d d l m!Z!m"Z" G d� de � Z#d� Z$d� Z%d� Z&d� Z'y# e$ r ed � �w xY w)� ��wraps)�Any� AsyncIterator�Callable�Iterator�ListN)�get_start_span_function)�DidNotEnable�Integration)�OP�SPANDATA)� SPANSTATUS)�Models�AsyncModelszgoogle-genai not installed� )� IDENTIFIER�ORIGIN� GEN_AI_SYSTEM)�set_span_data_for_request�set_span_data_for_response�_capture_exception�prepare_generate_content_args)�$set_span_data_for_streaming_response�accumulate_streaming_responsec �, � e Zd ZeZeZdd�Zed� � Z y)�GoogleGenAIIntegrationc � � || _ y �N)�include_prompts)�selfr s �\/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/google_genai/__init__.py�__init__zGoogleGenAIIntegration.__init__( s � �.��� c � � t t j � t _ t t j � t _ t t j � t _ t t j � t _ y r )�_wrap_generate_contentr �generate_content�_wrap_generate_content_stream�generate_content_stream�_wrap_async_generate_contentr �#_wrap_async_generate_content_stream� r$ r"