mcp_integration.cpython-312.pyc•6.01 kB
�
�Y�g� � �6 � d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z d dl
Z
d dlZd dlZdZe j" j% dd� Ze d Ze G d� d � � Ze G d
� d � � Z G d � d
� Z G d� d� Zdeeef deeef fd�Zy)� N)�Dict�List�Any�Optional�Literal)� dataclass�asdictzhttps://api.duckduckgo.com�CLAUDE_API_KEY� �claudec �X � e Zd ZU eed< dZeed< dZeed< dZeed< de ee
f fd�Z y )
�
DDGRequest�q�json�format� �no_html�
skip_disambig�returnc � � t | � S �N)r )�selfs �>C:\Users\risha\OneDrive\Desktop\MCP_Project\mcp_integration.py�to_dictzDDGRequest.to_dict s � ��d�|�� N) �__name__�
__module__� __qualname__�str�__annotations__r r �intr r r r � r r r r s9 � �
�F��F�C���G�S���M�3�� ��c�3�h� � r r c �, � e Zd ZU eed< eed< eed< y)� WebResult�title�url� descriptionN)r r r r r r" r r r$ r$ s � � �J� �H��r r$ c �8 � e Zd Zefdefd�Zddededee fd�Z y) � MCPClient�endpointc � � || _ y r )r* )r r* s r �__init__zMCPClient.__init__! s � � ��
r �query�countr c
�� � t |�� } t j | j |j � �� }|j � |j
� }g }|j d� rM|j t |j dd� |j dd� |j dd� �� � |S # t $ r}t |� g cY d }~S d }~ww xY w)N)r )�params�Abstract�Headingr � AbstractURL)r% r& r' ) r �requests�getr* r �raise_for_statusr �appendr$ � Exception�print)r r- r. �request�response�data�results�es r �searchzMCPClient.search$ s� � � �u�%�� � �|�|��
�
� ���(��H�
� %� %� '��=�=�?�D��G��x�x�
�#����y� �(�(�9�b�1� �(�(�=�"�5� $����R� 8� � �
�N��� � �!�H��I�� �s �B0B? �? C � C�C �C N)�
)
r r r �DUCKDUCKGO_ENDPOINTr r, r! r r$ r? r" r r r) r) s/ � �':� !�� !��C� �� �T�)�_� r r) c �F � e Zd Zddefd�Zdedee fd�Zdedee fd�Zy) �ClaudeMCPBridge� llm_providerc � � t � | _ || _ |dk( r5t j � }t j t |�� | _ y y )Nr )�api_key� http_client) r) �
mcp_clientrD �httpx�Client� anthropic� Anthropicr
�
claude_client)r rD rG s r r, zClaudeMCPBridge.__init__? sD � �#�+��� (��� �8� #� �,�,�.�K�!*�!4�!4�&� '�"�D� � $r � user_messager c �H � | j dk( r| j |� S dgS )Nr �error)rD �_extract_with_claude)r rN s r � extract_website_queries_with_llmz0ClaudeMCPBridge.extract_website_queries_with_llmK s) � � � � �� (��,�,�\�:� :��9� r c �� � | j j j ddddd|d�g�� }|j d j }t j d |t
j � }|r%t j |j d
� � }n t j |� }|j d g � }|S # Y y xY w# t $ r}t |� g cY d }~S d }~ww xY w)
Nzclaude-3-sonnet-20240229i� g�������?a& You are a helpful assistant that identifies web search queries in user message. Extract any specific website or topic queries the user wants information about. Return results as a JSON object with a 'queries' field containing an array of strings. If no queries are found, return an empty array.�user)�role�content)�model�
max_tokens� temperature�system�messagesr z ```(?:json)?\s*({.*?})\s*```r rP �queries)rM r[ �createrV �text�rer? �DOTALLr �loads�groupr5 r8 r9 )r rN r; rV �
json_match�resultr\ r> s r rQ z$ClaudeMCPBridge._extract_with_claudeQ s� � � ��)�)�2�2�9�9�0� � � @� #� �=�� :� �H� �&�&�q�)�.�.�G����#B�G�R�Y�Y�W�J�� �
�
�:�#3�#3�A�#6�7��#� !�Z�Z�� 0�F� �j�j��B�/�G��N��#�"�� � � �!�H��I�� �s< �BC �B? �+C �?C�C �C � C&� C!�C&�!C&N)r ) r r r � LLMProviderr, r r rR rQ r" r r rC rC = s< � �
�[�
� �S� �T�#�Y� � �� ��c�� r rC � tool_paramsr c � � | j dd� }|sddiS t � }|j j |� }d|D �cg c]
}t |� �� c}iS c c}w )Nr- r rP zno queryr= )r5 rC rH r? r )rf r- �bridger= rd s r �handle_claude_tool_callri o se � � �O�O�G�R� (�E� ���$�$�
�
�F�� � �&�&�u�-�G� ��9��v�F�6�N��9� � ��9s �A)�osr_ r r4 �typingr r r r r � dataclassesr r �openairK rI rA �environr5 r
re r r$ r) rC r ri r" r r �<module>ro s� �� � � � � 5� 5� )�
� � �2� ���� � � 0�"�5���h� � �
� � � � � �� � ��
� �:0� 0�d
��c�3�h� �
�D��c��N�
r