iR(ddlmZmZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZmZmZmZdd lmZdd lmZdd lmZdd lm Z Gd deZ!GddeZ"GddeZ#GddeZ$GddZ%y)) BaseModelField) TypedDict) AnnotatedListDictLiteralOptionalTupleUnion)settings)Tool) ChatOpenAI)ToolNode) StateGraphSTARTEND add_messages)CompiledStateGraph)AIMessageChunk)parse_partial_json) SystemMessagec"eZdZUeeefed<y)StatemessagesN)__name__ __module__ __qualname__rlistr__annotations__2/var/www/html/hotal-chatbot-be/app/rag/__init__.pyrrsl*++r"rc @eZdZdZddeeefdeedeeeffdZ y) RetrieverToolz;Custom Tool Just For Override _to_args_and_kwargs function.N tool_input tool_call_idreturnc4t|tr|fifSd|fS)Nr!) isinstancestr)selfr&r's r#_to_args_and_kwargsz!RetrieverTool._to_args_and_kwargss# j# &="$ $z> !r"N) rrr__doc__r r+rr r r-r!r"r#r%r%s:E"eCI.>"hWZm"glmrtxmxgy"r"r%c eZdZUedZeed<edZeed<edZe ed<edZ e ed <dedede d e d d f d Z dd Z y )SearchRoomsSchemazThe start date of the stay description from_datezThe end date of the stayto_datezThe number of adultsadultszThe number of childrenchildrenr(Nc<||_||_||_||_yr.)r4r5r6r7)r,r4r5r6r7s r#__init__zSearchRoomsSchema.__init__$s"    r"c"||j|<yr.)__dict__)r,namevalues r# __setattr__zSearchRoomsSchema.__setattr__*s# dr")r(N) rrrrr4r+r r5r6intr7r9r>r!r"r#r1r1sp'CDIsD%?@GS@$:;FC;&>?Hc?!#!!S!C!TX! $r"r1cLeZdZUedZeed<edZeeed<y)ClassifyResponsez6Final answer to the user's question in markdown formatr2answerzIGive 2-3 related questions from user's perspective to the user's questionrelated_questionsN) rrrrrBr+r rCrr!r"r#rArA-s0$\]FC]#(6A$BtCyBr"rAcxeZdZdZddefdZdeefdZde de dfd Z dd Z de fd Z ifdefd Zd efdZy) HotelAgentcDtdtj|_y)Nzgpt-5.2)modelapi_key)rr OPENAI_API_KEYllmr,s r#r9zHotelAgent.__init__3sIx7N7NOr"Nqueryc Kyw)NzRoom search completedr!)r,rL callbackskwargss r#data_retrievalzHotelAgent.data_retrieval6s &sr(c\Ktdd|j|jtgSw)NrPalRetrieves detailed and up-to-date information about Plaza Beach Resorts, including room availability for specified dates, occupancy limits (adults and children), room types, amenities, and property-specific details. Use this tool only when the requested information is not already available in the system prompt. Do not use it for greetings or unrelated questions.)r<r3func coroutine args_schema)r%rPr1rKs r# get_toolszHotelAgent.get_tools9s8 %` ((---   s*,statecontinueendcfK|d}|d}|j}|sy|djd}yw)NrrYrr<rX) tool_callsget)r,rVr last_messager\ tool_names r#should_continuezHotelAgent.should_continueJsA$| !,, qM%%f- s/1cK|jd{}tt}|jd|j|jdt ||j td|jd|jdtd|j dd|j dt|jS7w)z Builds and compiles the state graph defining agent behavior and tool usage. Returns: CompiledStateGraph: Compiled graph ready for execution. N) state_schemaagenttoolsrdrW) rUrradd_nodercradd_edgeradd_conditional_edgesr`rcompile)r,rdbuilders r# build_graphzHotelAgent.build_graphWsnn&&%0,("78(%%   "   '*#&  ''sCCB?CcFK|jd{}d|j_|jj|}|j d{}|j |d|j |dditd{}d|giS77A7 w)NTrerrun_nameAgent)configresponse_format)rUrJuse_responses_api bind_toolsget_agent_promptextendainvokerA)r,rVrdllm_with_toolsrresponses r#rczHotelAgent.agenttsnn&&%)",,5,9..00j)*'// w'.0 XJ''' 1s4B!BAB!B5B!B B!B!B!cnKd}t|g}|j|jdg|Sw)Na You are the official AI assistant for Plaza Beach Resorts (St. Pete Beach, Florida), featuring three hotels: Plaza Beach Hotel (beachfront), Bayview Plaza Waterfront Resort & Marina (waterfront, pet-friendly, couples-friendly), and Bay Palms Waterfront Resort & Marina (family-friendly suites with kitchens). Your role: - Provide helpful, friendly, and professional information only about these three hotels. - You are only here for guide users about the hotels, you can't perform any actions like booking, cancellation, payment, availability check or reservation changes etc. - Politely refuse unrelated questions. - Don't tell user about the tools you have or what you can't do(until user ask about it), you are only here to answer questions about the hotels. - Use the data retrieval tool only if needed and never mention it. - Never guess. If information is unavailable, say so politely. - Don't give big examples in your response, just answer the question. We have related questions for user to ask you. - Keep responses clear, concise, and welcoming. Important: Your response should be in markdown formatting. )content chat_history)rrtr])r,inputssystem_templaters r#rszHotelAgent.get_agent_prompts< " / 2   >267s35r{cK|jd{}d}d}d}|j|ddid23d{}|\}}|st|ts"t |ds/|j s:wui?X]b]j]jorsxtAtApB a(,,V4% 1,? +L0H ,X 6s6{| DIi'F B $,i@@$ -)  %f$ 66sNC:C, C:C0C.C0C:C: C:& C:3C: A$C:.C00 C:r.)r(r)rrrr9r+rPrr%rUrr r`rkrcrrsdictrr!r"r#rErE1slP'' m!4 " 5 W=N5O !:(("-/427D7r"rEN)&pydanticrrtyping_extensionsrtypingrrrr r r r app.configr langchain_core.toolsr!langchain_openai.chat_models.baserlanggraph.prebuiltrlanggraph.graphrrrrlanggraph.graph.staterlangchain_core.messagesr"langchain_core.output_parsers.jsonrrrr%r1rArEr!r"r#rsp%'III%8'@@42A1,I,"D" $ $ByBF7F7r"