providers.cpython-311.pyc•6.41 kB
�
�h� � � � d dl Z d dlZd dlZd dlmZmZmZ d dlmZ e� � G d� d� � Z G d� d� � Z
e � � e
� � d�Z dS ) � N)�Dict�Any�Optional)� load_dotenvc �@ � e Zd Zd� Zdefd�Zdedeeef fd�Z dS )�AzureOpenAIProviderc �� � t j d� � | _ t j d� � | _ t j d� � | _ t j dd� � | _ d S )N�AZURE_OPENAI_ENDPOINT�AZURE_OPENAI_DEPLOYMENT�AZURE_OPENAI_API_KEY�AZURE_OPENAI_API_VERSIONz2025-01-01-preview)�os�getenv�endpoint�
deployment�api_key� api_version��selfs �G/Users/kulraj/agent09/mcp-starter/mcp-bearer-token/../data/providers.py�__init__z AzureOpenAIProvider.__init__
sT � �� �"9�:�:��
� �)�$=�>�>����y�!7�8�8�� � �9�%?�AU�V�V����� �returnc �D � t | j | j | j g� � S �N)�allr r r r s r � is_availablez AzureOpenAIProvider.is_available s � ��D�M�4�?�D�L�A�B�B�Br �promptc � �� K � | � � � sddiS | j � d| j � d| j � �}| j dd�}ddd �d
|d �gd d d
�}t j � � } t
j � � 4 �d {V ��}|� |||d�� � � d {V ��}|� � � |�
� � }t t j � � |z
dz � � } |d d d d |� di � � � d� � | d�cd d d � � �d {V �� S # 1 �d {V ��swxY w Y d S # t $ r }
ddt |
� � � �icY d }
~
S d }
~
ww xY w)N�error�skipped_no_keyz/openai/deployments/z /chat/completions?api-version=�application/json)zapi-key� Content-Type�system�FYou are a neutral assistant generating concise, citation-rich answers.��role�content�user�� �333333�?)�messages�
max_tokens� temperature� >@��headers�json�timeout�� �choicesr �messager( �usage� total_tokens��
raw_answer� tokens_used�
latency_ms� api_error: )r r r r r �time�httpx� AsyncClient�post�raise_for_statusr2 �int�get� Exception�str� r r �urlr1 �data�
start_time�client�response�resultr< �es r �callzAzureOpenAIProvider.call sb � � � �� � �"�"� /�� -�.� .�� �u�u�D�O�u�u�cg�cs�u�u���|� .�
�
�� "�.v�w�w� �F�3�3�� � �
�
�� �Y�[�[�
� 5� �(�*�*�
�
�
�
�
�
�
�f�!'���S�'��VZ��![�![�[�[�[�[�[�[���)�)�+�+�+�!� � � � �� �$�)�+�+�
�":�d�!B� C� C�
� #)��"3�A�"6�y�"A�)�"L�#)�:�:�g�r�#:�#:�#>�#>�~�#N�#N�",�� �
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
����
�
�
�
�
�
�� � 5� 5� 5�� 3�3�q�6�6� 3� 3�4� 4� 4� 4� 4� 4� 4����� 5���sC � E �9B2D>�+E �>
E�E � E� E �
E9�E4�.E9�4E9N�
�__name__�
__module__� __qualname__r �boolr rF r r rO � r r r r sp � � � � � �W� W� W� C�d� C� C� C� C�#5�� #5��c�3�h� � #5� #5� #5� #5� #5� #5r r c �@ � e Zd Zd� Zdefd�Zdedeeef fd�Z dS )�PerplexityProviderc �8 � t j d� � | _ d S )N�PERPLEXITY_API_KEY)r r r r s r r zPerplexityProvider.__init__9 s � ��y�!5�6�6�� � � r r c �* � t | j � � S r )rT r r s r r z PerplexityProvider.is_available< s � ��D�L�!�!�!r r c � �� K � | � � � sddiS d}d| j � �dd�}ddd d
�d |d
�gd d
d�}t j � � } t j � � 4 �d {V ��}|� |||d�� � � d {V ��}|� � � |� � � }t t j � � |z
dz � � } |d d d d |� di � � � d� � | d�cd d d � � �d {V �� S # 1 �d {V ��swxY w Y d S # t $ r }
ddt |
� � � �icY d }
~
S d }
~
ww xY w)Nr r! z*https://api.perplexity.ai/chat/completionszBearer r" )�
Authorizationr# z!llama-3.1-sonar-small-128k-onliner$ r% r& r) r* r+ )�modelr, r- r. r/ r0 r4 r5 r r6 r( r7 r8 r9 r= ) r r r> r? r@ rA rB r2 rC rD rE rF rG s r rO zPerplexityProvider.call? sO � � � �� � �"�"� /�� -�.� .�:�� 5�t�|� 5� 5� .�
�
�� 9�!�.v�w�w� �F�3�3�� � �
�
�� �Y�[�[�
� 5� �(�*�*�
�
�
�
�
�
�
�f�!'���S�'��VZ��![�![�[�[�[�[�[�[���)�)�+�+�+�!� � � � �� �$�)�+�+�
�":�d�!B� C� C�
� #)��"3�A�"6�y�"A�)�"L�#)�:�:�g�r�#:�#:�#>�#>�~�#N�#N�",�� �
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
����
�
�
�
�
�
�� � 5� 5� 5�� 3�3�q�6�6� 3� 3�4� 4� 4� 4� 4� 4� 4����� 5���sC �
D> �&B2D+�D> �+
D5�5D> �8D5�9D> �>
E&�E!�E&�!E&NrP rU r r rW rW 8 sh � � � � � �7� 7� 7�"�d� "� "� "� "�$5�� $5��c�3�h� � $5� $5� $5� $5� $5� $5r rW )� azure_openai�
perplexity) r? r> r �typingr r r �dotenvr r rW � PROVIDERSrU r r �<module>rc s� �� � � � � � � � � � � � � &� &� &� &� &� &� &� &� &� &� � � � � � � � �
�
�
�-5� -5� -5� -5� -5� -5� -5� -5�^+5� +5� +5� +5� +5� +5� +5� +5�^ (�'�)�)�$�$�&�&�
�
� � � r