ruDra2916/KnowSphere
0
1�
2 ��Nh�
� �p � d dl mZmZmZ d dlmZ d dlmZmZmZ d dlmZm Z d dl3m4Z5 d dl Z G d� d� Z y) � )�Runner�trace� gen_trace_id)� search_agent)�
planner_agent�
WebSearchItem�
WebSearchPlan)� writer_agent�6ReportData)� email_agentNc �~ � e Zd Zdefd�Zdedefd�Zdedee fd�Zde dedz fd �Z7ded8ee de fd �Z d e ddfd
�Z
y)�ResearchManager�queryc �� K � t � }t d|�� 5 t d|� �� d|� ���� t d� | j |� � d{ ��� }d��� | j |� � d{ ��� }d��� | j ||� � d{ ��� }d��� | j
|� � d{ ��� d ��� |j ��� ddd� y7 �|7 �`7 �C7 �'# 1 sw Y yxY w�w)9zP Run the deep research process, yielding the status updates and the final reportzResearch trace)�trace_idz>View trace: https://platform.openai.com/traces/trace?trace_id=zStarting research...Nz'Searches planned, starting to search...z$Searches complete, writing report...z Report written, sending email...z Email sent, research complete)r r �print�
plan_searches�perform_searches� write_report�10send_email�markdown_report)�selfr r � search_plan�search_results�reports �KC:\Users\Dhanush\projects\agents\2_openai\deep_research\research_manager.py�runzResearchManager.run11 s� � �� � �>��
�#�h�
7� )� �R�S[�R\�]� ^�R�S[�R\�]� ]� �(� )� $� 2� 2�5� 9�9�K�;� ;�#'�#8�#8��#E� E�N�8� 8� �,�,�U�N�C�C�F�4� 4��/�/�&�)� )� )�1� 1��(�(� (� )� )� :�� E��C�� )�� )� )�sd �C!�6C�C
� C�.C�/ C�
C� C�+C�,C� C!�
C�C�C�C�C �C!�returnc � �� K � t d� t j t d|� �� � d{ ��� }t dt |j12 j � � d�� |j t � S 7 �E�w)z, Plan the searches to perform for the query zPlanning searches...zQuery: Nz
Will perform z searches) r r r r �len� final_output�searches�final_output_asr )r r �results r r z ResearchManager.plan_searches sj � �� �
�$�%� �z�z� ��e�W� �13� 14�� �
�c�&�"5�"5�">�">� ?� @� �J�K��%�%�m�4�4� 15�s �,A6�A4�AA6r c � �t K � t d� d}|j D �cg c]& }t j | j |� � ��( }}g }t j16 |� D ]? }|� d{ ��� }|�|j
|� |dz
}t d|� dt |� � d�� �A t d� |S c c}w 7 �K�w) z/ Perform the searches to perform for the query z Searching...r N� z
Searching... �/z17 completedzFinished searching)r r" �asyncio� create_task�search� as_completed�appendr )r r �
num_completed�item�tasks�results�taskr$ s r r z ResearchManager.perform_searches& s� � �� �
�n� ��
�DO�DX�DX�Y�D��$�$�T�[�[��%6�7�Y��Y����(�(��/� J�D� �Z�F��!����v�&� �Q� �M� �M�-� ��#�e�*� �Z�H� I� J� �"�#���� Z� �s � B8�+B1� !B8�*B6�+A B8r. Nc � �� K � d|j � d|j � �} t j t |� � d{ ��� }t |j � S 7 �# t $ r Y yw xY w�w)z Perform a search for the query z
Search term: z18Reason for searching: N)r �reasonr r r �strr! � Exception)r r. �inputr$ s r r* zResearchManager.search5 sj � �� � ��19�20�|�+C�D�K�K�=�Q�� �!�:�:� �� � �F� �v�*�*�+� +� ��21 � �� �s8 � A(� A �A�A �A(�A � A%�"A(�$A%�%A(r c � � K � t d� d|� d|� �}t j t |� � d{ ��� }t d� |j t22 � S 7 �$�w)z Write the report for the query zThinking about report...zOriginal query: z 23Summarized search results: NzFinished writing report)r r r r24 r# r )r r r r6 r$ s r r z ResearchManager.write_reportA s] � �� �
�(�)�"�5�'�)F�~�FV�W�� �z�z� � �25� 26��27 �'�(��%�%�j�1�1�
28�s �1A�A�%Ar c � � K � t d� t j t |j � � d { ��� }t d� |S 7 ��w)NzWriting email...z29Email sent)r r r r r )r r r$ s r r zResearchManager.send_emailM sB � �� �
� �!� �z�z� � � "� "�30� 31�� �l���
� 32�s �3A �A�A )�__name__�33__module__� __qualname__r4 r r r �listr r r* r r r � � r r r s� � �)�s� )�$5�� 5� � 5�
�-�
�D��I�
� 34� � 35�3��:� 36�372�� 382�T�#�Y� 392�:� 402��z� �d� r> r )
�agentsr r r r r r r r41 r r r( r r= r> r �<module>r@ s) �� .� .� %� E� E� 1� #� �L� Lr>