📁 File Manager Pro
v10.0.2 | PHP: 8.1.34
Server: LiteSpeed
2026-06-27 12:40:42
📂
/
/
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
__pycache__
✏️
Editing: cohere.cpython-312.pyc
� Yr*j�) � �. � d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z mZ d dlm Z d dlmZ d d lmZ erd d lmZmZmZmZ d dlmZ d dlZd dlmZmZ d d lmZ d dlmZmZm Z d dl!m"Z"m#Z# d dl$m%Z% d dl&m'Z' erd dl!m(Z( d dl!m*Z* e jX e jZ e j\ e j^ e j` e jb e jd d�Z3e jh e jj d�Z6e jn e jp e jr d�Z:e jv e jx e jz e j| e j~ d�Z@ G d� de� ZAd$d�ZBd%d�ZCdd d!eDdd fd"�ZEd&d#�ZFy# e)$ r ed� �w xY w# e)$ r d dl!m+Z* Y ��w xY w)'� N��wraps)� TYPE_CHECKING)�consts)�record_token_usage)�get_start_span_function�set_data_normalized)�SPANDATA)�StreamedSpan)�has_span_streaming_enabled)�Any�Callable�Iterator�Union)�Span)�DidNotEnable�Integration)�should_send_default_pii)�capture_internal_exceptions�event_from_exception�reraise)�ChatStreamEndEvent�NonStreamedChatResponse)� BaseCohere)�Client)�StreamedChatResponsezCohere not installed)�StreamEndStreamedChatResponse)�StreamedChatResponse_StreamEnd)�model�k�p�seed�frequency_penalty�presence_penalty� raw_prompting)�tools�preamble)� generation_id�is_search_required� finish_reason)� citations� documents�search_queries�search_results� tool_callsc �B � e Zd ZdZde� �Zd dd deddfd�Zed d�� Zy)�CohereIntegration�coherezauto.ai.�self�include_prompts�returnNc � � || _ y �N)r4 )r3 r4 s �M/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/cohere.py�__init__zCohereIntegration.__init__L s � �.��� c �� � t t j d�� t _ t t j � t _ t t j d�� t _ y )NF)� streamingT)� _wrap_chatr �chat�_wrap_embedr �embed�chat_stream� r: r8 � setup_oncezCohereIntegration.setup_onceO s9 � �$�Z�_�_��F� ��"�6�<�<�0���!+�J�,B�,B�d�!S� �r: )T)r5 N) �__name__� __module__�__qualname__� identifier�origin�boolr9 �staticmethodrC rB r: r8 r1 r1 H sD � ��J�� �|� $�F�/�*� /�T� /�T� /� �T� �Tr: r1 r5 c � � t | t j � j ddd��� \ }}t j ||�� y )Nr2 F)�type�handled)�client_options� mechanism)�hint)r � sentry_sdk� get_client�options� capture_event)�exc�eventrP s r8 �_capture_exceptionrW V s? � �&��!�,�,�.�6�6�#��6��K�E�4� ���U��.r: c �l � t | t � r| j � y | j d d d � y r7 )� isinstancer �end�__exit__)�spans r8 � _end_spanr] _ s% � ��$��%���� �� � �d�D�$�'r: �f�Callable[..., Any]r<