|j3 ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZddlmZmZmZmZdd lmZmZmZGd d eZGd d eZGddeeZGddeeZGddeeZGddeeZ GddeZ!GddeZ"GddeZ#GddeeZ$GddeZ%Gd d!eZ&gd"d"fd#e'd$ed"zd%ed"zd&efd'Z(gd"fd(ed)ed#e'd*ed&e)f d+Z*Gd,d-Z+y").)Enum)AnyLiteralOptional) BaseModelField)settings)retrieve) ChatOpenAI)parse_partial_json)ChatPromptTemplateHumanMessagePromptTemplateSystemMessagePromptTemplateAIMessagePromptTemplate)BASE_SYSTEM_TEMPLATEget_hotel_details SYSTEM_TEMPLATE_FOR_CONTEXT_DATAc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) ToolEnummodify_reservation_toolsearch_room_tool add_to_cartadd_billing_detailssend_email_toolclear_cart_toolnote_special_requestget_reservation_detailsno_toolN) __name__ __module__ __qualname__modifyReservationsearchbookingbilling send_email clear_cartspecial_requestrdefault4/var/www/html/ls-ibe-bot-apis/app/agents/__init__.pyrr s11 FG#G"J"J,O7Gr+rc(eZdZUedZeed<edZeed<eddd Ze ed <ed d d Z e ed <egdZ e eed<egdZ e eed<eddZeed<eddZeed<eddZeed<y) AgentParamszCheck-in date in MM-DD-YYYY format. Always populate this field. Use the user's check-in date if provided; otherwise use the default CHECK-IN value from the prompt. descriptioncheck_inzCheck-out date in MM-DD-YYYY format. Always populate this field. Use the user's check-out date if provided; otherwise use the default CHECK-OUT value from the prompt. check_outzNumber of adults for the room search or booking. Always populate this field. Use the user's value if provided; otherwise default to 2.)r)ger0 adultCountrzNumber of children for the room search or booking. Always populate this field. Use the user's value if provided; otherwise default to 0. childCountz9Requested room names if the user mentions one explicitly.r)r0 room_namezList of exact room_type_ids for booking or add-to-cart requests. Use only when the requested rooms is known from context or prior results. Never guess or fabricate this value. room_type_idNaReservation confirmation number for cancellation-email requests. Itenrary is also accepted here. Reservation confirmation number for special request notes if reservation is already confirmed. Itenrary is also accepted here.Reservation confirmation number for getting reservation details when calling get_reservation_details tool. Itenrary is also accepted here.Use a value from the user query or conversation history when available.confirmation_numberzEmail address of the user for special request notes. Email address of the user for getting reservation details when calling get_reservation_details tool. Use a value from the user query or conversation history when available.emailzSpecial request notes from the user for the reservation. Don't copy exact text from user query, Explain the request in your own words and summarize it in your own words.r()rr r!rr1str__annotations__r2r6intr7r9listr:r;r<r(r*r+r,r.r.s NHc  PIs   d J  d J!OItCy$ 3L$s) % V  VE3! OSr+r.ceZdZdZdZy) SortDirection ascending descendingN)rr r!rCrDr*r+r,rBrBYs IJr+rBceZdZdZdZy) SortFieldrate occupancyNrr r!rGrHr*r+r,rFrF] DIr+rFceZdZdZdZy) FilterFieldrGrHNrIr*r+r,rLrLarJr+rLc$eZdZdZdZdZdZdZdZy)FilterOperatorz==z!=>=z<=N) rr r!eqnegtltgtelter*r+r,rNrNes B B B B C Cr+rNcbeZdZUedZeed<edZeed<edZ e ed<y) FilterConditionz/Field name to filter on (e.g., rate, occupancy)r/fieldzComparison operatoroperatorzValue to compare againstvalueN) rr r!rrYrLr>rZrNr[rr*r+r,rXrXms2+\]E;]$1FGHnG#=>E3>r+rXcFeZdZUedZeed<edZeed<y) SortConfigzField to sort byr/rYzSort direction directionN) rr r!rrYrFr>r^rBr*r+r,r]r]ss");r@rdrXr*r+r,r`r`ws<#+D*t )./)J_%r+r`c eZdZdZdZdZdZdZy)BillingFormStepaddons insurancecustomer_details_formpromo_code_formpayment_details_formN)rr r!rgrhrirjrkr*r+r,rfrfs FI3'O1r+rfc0eZdZUedZeed<eddZee ed<edZ e ed <ed Z e ed <edd Zeed <eddZeed<eddZeeded<eedZee ed<eddZeed<y) AgentSchemaaTool to execute for the user's request. Use add_to_cart for booking or add-to-cart actions, search_room_tool for room search or reopening the room slider, send_email_tool for cancellation email requests, note_special_request for special request notes, and no_tool when only a conversational answer is needed. The no_tool path may still include room_type_ids so the frontend can open the slider from general room matches.r/toolNzMif calling the billing tool, the step of the billing form to show to the userr8billing_form_stepzvFriendly user-facing response. Keep it positive and concise. If a tool will be used, start naturally with 'Let me...'.answerz8Structured parameters extracted from the user's request.paramsz#Filters to apply to the room searchfilterszVNumber of rooms to return in the frontend slider. Use -1 to return all matching rooms. room_countaOrdered list of hotel IDs whose rooms should be returned. Rules: 1. Default to ['124'] and show Bayview Plaza rooms first. 2. If the user asks for more options or other hotels, use ['123', '124', '125']. 3. If the user explicitly requests a specific hotel, include only the requested hotel IDs. 4. Do not change hotels unless the user asks or signals dissatisfaction. 5. room_type_ids must stay scoped to these hotel_ids.cdgS)N124r*r*r+r,zAgentSchema.swr+)r0rc)123rv125 hotel_idsaSpecific room_type_id values to render on the frontend when search_room_tool is not being used. If the answer lists or recommends room types from GENERAL-ROOM-DETAILS, always include the matching room_type_ids here. This is required for frontend slider rendering. Only include room_type_ids from the currently selected hotel_ids. By default that means only hotel 124 unless the user explicitly asks for another hotel or broader hotel options. Leave empty only when no specific room types are being referenced.rb room_type_idsraTotal number of allowed pets (dogs and cats) in the room. Use 0 for no pets. Always include this field if the user explicitly asks for pets or mentions they are bringing pets. Frontend will use this field to automatically add pets add-on to the cart if user add a room to cart in the same chat. Wait for clarification from the user if you are not sure about the number of pets. If user mention any pet so in every conversation you have keep track of the number of pets and update this field accordingly becasue this is the only way to know the number of pets in the room until chat is over. Make it back to 0 if user mention no pets or no more pets or switch to hotel which doesn't allow pets. total_pets)rr r!rrnrr>rorrfrpr=rqr.rrr`rtr?rzr@rr{r|r*r+r,rmrms ~ D( 49c4x0 HFC  NFK9GWlJ5: @'5(ItG/01( % Q M49 L Jr+rmcJeZdZUeddZeed<eddZeed<y) ContextDatazdThe standalone query to retrieve the most relevant context for answering the user's latest question.)r0r)queryzIWhether to skip vector database retrieval for the user's latest question.FskipN) rr r!rrr=r>rboolr*r+r,r~r~s5$JTVWE3W#nx}~D$~r+r~Nhistory tool_response tool_callreturncHts tdtjtg|t jd}|rJ|rH|j t jd|j t jdtj|S)NzInvalid BASE_SYSTEM_TEMPLATEz {question}z {tool_call}zTool response: {tool_response}) r ValueErrorr from_templaterappendrr from_messages)rrrmessagess r,get_chat_promptrs 788 $112FG  #00>H /==mLM/==>^_`  + +H 55r+question sub_hotel_id callbackscjKttjd}tjt g}t j||jtz}|j||dd|dd{}|jryt|j|d{S7/7w)Nz gpt-5.4-mini)api_keymodel)rrz Context Data)run_namer)configr)r r OPENAI_API_KEYrrrr rwith_structured_outputr~ainvokerr r)rrrrllmrchainresults r,get_context_datars X44N KC $112RSH  , ,X 6#:T:TU`:a aE % 'x.Xn|LUbV !WWF {{&,, 5 55 W 6s$A=B3?B/*B3*B1+B31B3c zeZdZdejdfdededededzfdZigdddfded ed edzd edzd e f d Z y) HotelAgentzgpt-5.1Nrrrconversation_idcJ||_||_||_||_d|_y)N)rrrrr)selfrrrrs r,__init__zHotelAgent.__init__s'(  +:*. r+inputrrrrcKt|j|jd}tj }|dddj dd|j d|ddd<|dd dj d |j |dd d<|j|}t|jd |j ||d{} | |d <t||| |z} |r |r ||d<||d<| j|d|jr |jnd|j r |j ndg|d} d} d} | 23d{}|r|jddk(r|jdr|jdjdrt|jdjddrf|jdjdj}|s| |z } t| }|%d|vr!|dt!| d}| |z } |dk(r|| } |s|jddk(s|jdijd}|tj#||_97786yw)NT)rr streaming propertiesrzr0z['124']z['z']r{rvr hotel_context)rrrrrz Main Agentzunknown-conversationz unknown-hotel)rtagsr)rrreventon_chat_model_streamdatachunkcontentrp on_chain_endoutput)r rrrmmodel_json_schemareplacerrrgetrastream_eventsrhasattrrr lenmodel_validater)rrrrrrrschemallm_with_schemacontextr stream_result final_answerrprtoken escaped_text new_tokenrs r,streamzHotelAgent.streamsTZZ4P..0;A,;OP[;\]j;k;s;st}BDEIEVEVDWWY@Z<[|[)-8?El?STc?der?s?{?{}BDHDUDU@V|_-m<44V<(:)>@Q@QSZ\eff!(o}Xabett Y%2E/ "!*E+ ,,(151E1E--Kax|yJyJcgctctP_`&-  ( E%7+/EE99V$6):)>)>w)Guyy044W=yI % & 1 5 5g > F F$$$- '9,'G '3L8P(4X(>s6{|(LI"i/F(B ("+O!'7+~=62.228<%"-"<"rsH))%)*8AMMii t @)@DCT#tS$?i? CCi2c428)8t)$&4ae 6D 6cDj 6TWZ^T^ 6j| 6NPbf 6S 6 6d 6\_ 6ko 6;E;Er+