0(hI S SK Jr S SKJrJrJrJrJrJr S SK J
r
J r J r S SK
Jr S SKJr S SKJrJrJrJr S SKJr S SKJrJrJrJ r S S K J r S S
K J r S S K!J"r" S S K#J$r$ S S
K%J&r& S SK%J'r' S SK(J)r) S SK*J+r+J,r, S SK-J.r. \" 5 \"S\/S\/4S j5 r0\"S\/S\/4S j5 r1\"S\/S\/4S j5 r2\" \.S9r3\1\0\2/r4S r5\6S:X a@ Sr7\5" 5 r8\" \7S9/r9\8Ru S\905 r9\9S H r;\;Ry 5 M g g ) ) 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_promptqueryreturnc [ U SS9R 5 nSR U Vs/ s H@ nSUR S SUR R SS5 S UR
S
3PMB sn5 nS U0$ s snf ) z^Search Wikipedia for a query and return maximum 2 results.
Args:
query: The search query. r"
load_max_docs
---
wiki_results)r loadjoinmetadataget page_contentr" search_docsdocformatted_search_docss T/Users/eric/Documents/study/hf-agents-course/hfagentscourse-finalassignment/agent.py search_wikir; s "Q?DDFK ) . . #
" ! h!7 8 AQAQRXZ\A]@^^cdgdtdtcu vC
D"
1 22
s AA:c [ SU 35 [ SS9R U 5 n[ S[ U5 SU S35 SR U Vs/ s HM nSUR
R
S S
5 S UR
R
S S
5 SUR S3PMO sn5 nU( d [ S5 SU0$ s snf ! [ a( n[ SU 35 SS[ U5 30s SnA$ SnAff = f)z[Search Tavily for a query and return maximum 3 results.
Args:
query: The search query.z" executing search_web with query: ) max_resultszFound z documents for query ''r( r) r* unknownr+ r, N/Ar. r/ zEmpty search results web_resultszError in search_web: zSearch error: N)
printr runlenr2 r3 r4 r5 Exceptionstr)r" r7 r8 r9 es r:
search_webrI . s,
.ug
67:)a8<ugQGH - 2 2 '
&C %S\\%5%5h %J$K8TWT`T`TdTdekmrTsSttyz} {K {K zL LY Z&
!
% ) * 455
:
%aS)* Ax 899:s1 AC AC& C C
C: C5/C:5C:c
[ U SS9R 5 nSR U Vs/ s HC nSUR S SUR R SS5 S UR
S
S S 3PME sn5 nS
U0$ s snf )zYSearch Arxiv for a query and return maximum 3 result.
Args:
query: The search query.r= r& r( r) r* r+ r, rA r. Ni r/
arxiv_results)r r1 r2 r3 r4 r5 r6 s r: search_arxivrL H s E;@@BK ) . . #
" ! h!7 8 AQAQRXZ_A`@aafgjgwgwx}y}g~f @M
N"
2 33
s A
A=)contentc ^ [ SSS9n [ SU 35 U R [ 5 m[ ST 35 " S S[ 5 nSU4U4S jjn[ U5 nUR
S
U5 UR
S [ [ 5 5 UR [ S
5 UR S
[ S S [ [ 05 UR S S
5 UR 5 $ ) Nz gemini-2.5-flash-preview-04-17r )model temperaturezDEBUG: llm object = z DEBUG: llm_with_tools object = c 0 \ rS rSr% \\\ \4 \S' Sr g) build_graph..AgentStatex messages N)
__name__
__module__ __qualname____firstlineno__r listr r __annotations____static_attributes__rU r:
AgentStaterR x s D ,l :;;r] r^ statec > TR U S 5 n[ SU 35 [ U[ 5 ( a UR c SSSS.Ul SU/0$ )NrT zDEBUG: LLM result = r ) input_tokens
output_tokens total_tokens)invokerC
isinstancer usage_metadata)r_ resultllm_with_toolss r: assistant build_graph..assistant{ se &&uZ'89
$VH-. fi ( (V-B-B-J56\]$^F !
r] ri tools)r
rC
bind_toolsrk r r add_noder add_edger add_conditional_edgesr r compile)llmr^ ri builderrh s @r: build_graphrs g s
. C
&' ^^E*N +N+;
<=r s) B B ] ] 9 # A A 0 V V 5 % G @ < 8 8 !
3s 3s 3 3 :c :c : :2 4 4 4 4&
. 7 t z~H ME X./H||Z 23H
j
! " r]