J4fY z d dl Z d dlZd dlmZmZmZ d dlmZmZm Z d dl
m
Z
m Z d dl m
Z
d dlmZ d dlmZ d dlmZmZmZ d d lmZmZmZmZ d d
lm Z d d l m Z d d l m Z d
dl!m"Z" d dl#m$Z$ e$j% Z& G d de Z' G d de Z(dZ) ddeee defdZ* G d d Z+dS ) N) BaseModelExtraField)ListOptionalUnion)datetime timedelta)CallbackManager)!create_structured_output_runnable)Runnable)ChatPromptTemplateSystemMessagePromptTemplateHumanMessagePromptTemplate)
SystemMessage AIMessage HumanMessage BaseMessage)BaseMessagePromptTemplate)settings)
ChatAdditonal )
MultiArgsTool) CustomLoggerc e Zd ZU edd Zeed< edd Zeed< edd Ze ed < ed
d Z
e ed < ed
d Z e e
ed< dS )SearchRoomToolSchemazStart date of bookingN) descriptiondefault from_datezEnd date of bookingto_datezCount of adult adultzCount of childr childzQUser Provided Keywords on which we find rooms, like 'room_name', 'amenities', etc room_filters)__name__
__module__ __qualname__r r r __annotations__r r" intr# r$ liststr :/var/www/html/ls-bot-apis/app/agents/tools/search_rooms.pyr r s %,CTRRRIxRRR *?NNNGXNNN#3Q???E3???#3Q???E3??? #e 1D NR S S SL$s) S S S S Sr- r c z e Zd ZU ed Zeed< ed Zee ed< G d d Z dS ) SearchRoomSchemaz-Answer to user question with details of rooms)r answerzGenerate very short responses of user consisting of 2-3 words only. Examples could include 'Book now', 'Show more rooms', or 'Need help'followUpQuestionsc e Zd Zej ZdZdS )SearchRoomSchema.ConfigTN)r% r& r' r allowextraarbitrary_types_allowedr, r- r. Configr4 ! s "& r- r8 N)
r% r& r' r r1 r+ r( r2 r r8 r, r- r. r0 r0 s %$STTTFCTTT"'% 5 #@ #@ #@d3i @ @ @' ' ' ' ' ' ' ' ' 'r- r0 a You are a helpful AI assistant trained to answer questions about "Bay View Hotel" -> Available Rooms Details. Use the following pieces of context to answer the question. If you don't find the answer in the context, politely explain that you don't know and suggest rephrasing the question, don't try to make up an answer. Answer in the language of the user question.
Today's in date DD/MM/YYYY for your Knowledge: {date}
Include any relevant images or URLs, but only if they are found in the context. Don't make answer too much large for user gave you view in concise way (Follow law of least action).
{context}
Format your helpful response as "markdown".extra_prompt_messagesreturnc | pg }g |t j t t j d }t | S )Nz
{question})messages)r
from_templateSEARCH_ROOM_PROMPTr r )r9 _promptsr< s r. create_search_tool_promptr@ 0 s] %*HE
E
' 56H I IE
' 4\ B BE
H
x 0 0 00r- c ~ e Zd Zd
dededeeeee f ddfdZ
d Z dd e fd
Z
ddefd Zed ZdS )SearchRoomToolNquestionllmhistoryr: c || _ || _ || _ t j | _ t j | _ t j | _ d S N)
rD rC chat_hisoryr
LS_API_URL hotel_urlHOTEL_CURRENT_VERSIONhotel_current_verion SUB_HOTEL_ID sub_hotel_id)selfrC rD rE s r. __init__zSearchRoomTool.__init__> s>
"!,$,$B! $ 1r- c > t dd| j t d S )N*search_or_check_hotel_rooms_availabilitieszUse this tool to Search or check hotels rooms availability with price. Only use this tool if all parameters are present in user query, if not then ask the user for requierd parameters to use this tool.T)namer func args_schema
return_direct)r
_search_roomsr )rO s r. __call__zSearchRoomTool.__call__F s. = d#,
r- r callbacksc T t | j t t | j } | j ||||fi |}t
| r| nd} | | j | t j d dd|i }
d |
_
t
| rf|r|r|rdnd } t d
d g|| rz-SearchRoomTool._get_rooms..t s0 o o ot^EZEZ[_amEnEn od o o or- datac
v g | ]6}|d |d |d |d |d |d |d d7S )room_display_namethumbnailImage
bundleRoom occupancybedroom_type_code room_type_id)roomName roomImager r r r r r, r{ r~ s r. r} z-SearchRoomTool._get_rooms..v sk !%%8 9!%&6!7"&|"4!%k!2 ;&*+;&<$( $8 r- z No Room Found While Filteringc g | ]
}|d S )r r, r s r. r} z-SearchRoomTool._get_rooms.. s FFF$n-FFFr- z)/frontend/Searchresults/getRoomsRatesInfoz%Y-%m-%dr )daystrue )adultsr# room_stay_from room_stay_tocalled_for_suitesroomIdspromo)jsonru z(Error Occur in While Room Rates Fetchingc g | ]
}|d S )rater, r s r. r} z-SearchRoomTool._get_rooms.. s VVV$f VVVr- resultsr zFailed to upload file: )requestsgetrJ rL r loadscontentdecode status_code Exceptionrh post
isinstancer rl rk r
valuesziploggerrx r+ )rO r r r" r# r$ rn response
rooms_datarp room_idsrooms_rate_data room_rateroomr es ` r. rg zSearchRoomTool._get_roomsg s 9 |!AA*D,EF H H$4$;$;K$H$HIIJ#s**j.Bg.M.M DEEEJ
p o o o oz&/A o o o
'v . J z??
A ?@@@FF*FFFH }!LL # "HRS\^fHgHg 'Qi&8&8&D&D&Dmumym{m{ nE nE FP nQ nQDNwX`DaDa %aG$4$4Z$@$@$@hphthvhv zC IJ zK zK zK iK hU hU V` ha ha)/ ' +D,EF H #s**j.Bg.M.M JKKK "j)9)@)@)M)M N NOVV/)2L2S2S2U2UVVVI !*i 8 8
$
$
d #V LL;3q66;; < < <IIIIII s H8H< <
I<+I71I<7I<c : t fd|D S )Nc 3 t K | ]2}| d v V 3dS )r N)lower)r{ entityr| s r. z(SearchRoomTool.is_has.. sB ^^66<<>>T*=%>%D%D%F%FF^^^^^^r- )any)r| entitiess ` r. rz zSearchRoomTool.is_has s' ^^^^U]^^^^^^r- rG )NNNr r )NNr r N)r% r& r' r+ r
r r r r r rP rX r rW r* rg staticmethodrz r, r- r. rB rB = s 2 2 28 2d5 W`boIoCp>q 2 ~B 2 2 2 2
0: :mq : : : :z _ _ \_ _ _r- rB rG ),r r pydanticr r r typingr r r r r
langchain.callbacks.managerr !langchain.chains.openai_functionsr langchain_core.runnablesr
langchain.promptsr r r langchain.schemar r r r langchain.prompts.chatr app.configsr app.schemas.chatr r r app.utils.loggerr
get_loggerr r r0 r> r@ rB r, r- r. r s/ , , , , , , , , , , ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( 7 7 7 7 7 7 O O O O O O - - - - - - i i i i i i i i i i P P P P P P P P P P P P < < < < < < * * * * * * ) ) ) ) ) ) " "S S S S S9 S S S' ' ' ' 'y ' ' ' / :>
1
1#D$56
1
1
1
1
1i_ i_ i_ i_ i_ i_ i_ i_ i_ i_r-