(hY f d dl mZ d dlmZmZmZmZmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZmZmZmZ d dlmZ d dlmZmZmZm Z d d l m Z d d
l m Z d d l!m"Z" d d l#m$Z$ d d
l%m&Z& d dl%m'Z' d dl(m)Z) d dl*m+Z+m,Z, d dl-m.Z. e de/de/fdZ0e"de/dee/e/f fd Z1e"de/dee/e/f fd Z2e"de/dee/e/f fd Z3 ee. Z4e2e1e3gZ5d Z6e7dk( rTd Z8 e6 Z9 ee8 gZ: e;d e8 e9jy d e:i Z: e;d e:d D ] Z=e=j} y y )! ) load_dotenv) TypedDictListDictAnyOptional Annotated)ChatHuggingFaceHuggingFaceEndpointHuggingFaceEmbeddings)ChatGoogleGenerativeAI)ChatGroq)
StateGraph
MessagesStateSTARTEND) add_messages)
SystemMessage HumanMessage
AnyMessage AIMessage)subtract_usage)Tool)tool)TavilySearchResults)WikipediaLoader) ArxivLoader)
BM25Retriever)ToolNodetools_condition)
system_promptcontentreturnc 4 | j d }|D ] }|j j d s#|j j }|sD|d }|j dd j }|j d d }d| c S y)z7Extract arXiv ID from text content and format as a URL.
zarXiv:r vzhttps://arxiv.org/abs/unknown)splitstrip
startswithreplace)r" lineslinepartsarxiv_id_with_prefixarxiv_id
base_arxiv_ids T/Users/eric/Documents/study/hf-agents-course/hfagentscourse-finalassignment/agent.py
get_arxiv_urlr4 s MM$ E ::< " "8 ,JJL&&(E',Qx$/77"EKKM !)s 3A 6
/
?? queryc t d| t | d j }t dt | d| d dj |D cg c]L }d|j
j
d d
d |j
j
d d
d|j dN c} }|s t d d|iS c c}w # t $ r(}t d| ddt | icY d}~S d}~ww xY w)zfSearch Wikipedia for a query and return maximum 2 results.
Args:
query: The search query.z# executing search_wiki with query: r6
load_max_docsFound documents for query ''
---
Empty search results wiki_resultszError in search_wiki: Search error: N)
printr loadlenjoinmetadataget page_content Exceptionstrr6 search_docsdocformatted_search_docses r3 search_wikirV 0 s,
/w
78;%ECHHJ
s;'((>ugQGH - 2 2 '
&C %S\\%5%5h %J$K8TWT`T`TdTdekmoTpSqqvwz xH xH wI IV W&
!
% ) * 566
;
&qc*+ .Q 9::;s1 AC AB>&C >C C4 C/)C4/C4c t d| t d j | }t dt | t d| t dt | d| d d j |D cg c]> }d
|j
d d d
|j
dd d|j
dd d@ c} }|s t d d|iS c c}w # t $ r(}t d| ddt | icY d}~S d}~ww xY w)zcSearch Tavily for a query and return maximum 3 results.
Args:
query: The search query.z" executing search_web with query: ) max_resultsDEBUG: search_docs type: DEBUG: search_docs content: r; r< r= r> r? urlr( rA rB N/ArC r" r& rD rE web_resultszError in search_web: rG N) rH r runtyperJ rK rM rO rP rQ s r3
search_webra I s@
.ug
67:)a8<
,[M:;
s;'((>ugQGH - 2 2 '
&C %SWWUI%>$?xPVX]H^G__dehelelmvxze{d| }J K&
!
% ) * 455
:
%aS)* Ax 899:s1 A*C :AC=C C D # D D D c
0 t d| t | d j }t dt | t d| t dt | d| d d j |D cg c]H }d
t
|j d |j j d d
d|j dd dJ c} }|s t d d|iS c c}w # t $ r(}t d| ddt | icY d}~S d}~ww xY w)zaSearch Arxiv for a query and return maximum 3 result.
Args:
query: The search query.z$ executing search_arxiv with query: rX r9 rZ r[ r; r< r= r> r? rA rB r] rC Ni rD rE
arxiv_resultszError in search_arxiv: rG ) rH r rI r` rJ rK r4 rN rL rM rO rP rQ s r3 search_arxivrd f s^
0
89<!Q?DDF
)${*;)<=>
,[M:;
s;'((>ugQGH - 2 2 '
&C %]33C3C%D$EXcllN^N^_eglNmMnnstw uE uE FK GK uL tM MZ [&
!
% ) * !677
<
's+, >#a&!:;;.AgentStatemessagesN)__name__
__module__ __qualname__r listr r __annotations__ r5 r3
AgentStateri s D ,l :;;r5 rq statec j | d }t d| t |t r|j dddd|_ d|giS )Nrj zDEBUG: LLM result = r ) input_tokens
output_tokens total_tokens)invokerH
isinstancer usage_metadata)rr resultllm_with_toolss r3 assistantz build_graph..assistant sb &&uZ'89
$VH-. fi (V-B-B-J56\]$^F !
r5 r| tools)r
rH
bind_toolsr} r r add_noder add_edger add_conditional_edgesr r compile)llmrq r| builderr{ s @r3 build_graphr s
. C
&' ^^E*N +N+;
<=r sm B B ] ] 9 # A A 0 V V 5 % G @ < 8 8 !
3 3 $ ;s ;tCH~ ; ;0 :c :d38n : :8 < <S#X < <<
. 3 l z5H ME X./H )(
45||Z 23H
/0
j
! " r5