*`i4ddlZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z ddlmZddlmZmZdd lmZdd lmZGd d eeZGd deZGddeZGddeZGddeZGddeZGddeZGddeeZGddeZ GddeZ!Gdd eZ"Gd!d"eZ#ee"ezeze!ze ze#ze d#$fZ$ee"ezeze!ze ze d#$fZ%d%e&eee&eefzfd&e$fd'Z'dS)(N)Enum)Path)Literal)urlparse) ConfigDictFieldValidationErrorfield_validator) Annotated)EXPECTED_FORMAT_VALUESAudio) MistralBase)SerializableImagec*eZdZdZdZdZdZdZdZdZ dS) ChunkTypesaEnum for the types of chunks that can be sent to the model. Attributes: text: A text chunk. image: An image chunk. image_url: An image url chunk. input_audio: An input audio chunk. audio_url: An audio url chunk. Examples: >>> from mistral_common.protocol.instruct.chunk import ChunkTypes >>> chunk_type = ChunkTypes.text textimage image_url input_audio audio_urlthinkingN) __name__ __module__ __qualname____doc__rrrrrrz/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/protocol/instruct/chunk.pyrrs8   D EIKIHHHrrceZdZUdZeejejejej ej ej fe d<de eee eefzffdZede eee eefzfddfdZdS)BaseContentChunkzBase class for all content chunks. Content chunks are used to send different types of content to the model. Attributes: type: The type of the chunk. typereturncJtdt|j)z_Converts the chunk to the OpenAI format. Should be implemented by subclasses. z%to_openai method not implemented for )NotImplementedErrorr!rselfs r to_openaizBaseContentChunk.to_openai8s$ ""_$t**J]"_"_```r openai_chunkc0td|j)zgConverts the OpenAI chunk to the Mistral format. Should be implemented by subclasses. z'from_openai method not implemented for )r$rclsr(s r from_openaizBaseContentChunk.from_openai?s ""ZCL"Z"Z[[[rN)rrrrrrrrrrrr__annotations__dictstrr' classmethodr,rrrr r &s    a4S4S>%9 9:aaaa\tCtCH~1E,E'F\K]\\\[\\\rr ceZdZUdZejZeejed<e ed<e dZ de e e e e e fzffdZede e e e e e fzfddfd Zd S) ImageChunkzImage chunk. Attributes: image: The image to be sent to the model. Examples: >>> from PIL import Image >>> image_chunk = ImageChunk(image=Image.new('RGB', (200, 200), color='blue')) r!rTarbitrary_types_allowedr"cP|dhddid}dd|idS)(Converts the chunk to the OpenAI format.radd_format_prefixT)includecontextrurlr!r model_dump)r& base64_images rr'zImageChunk.to_openaiWs9yCVX\B]^^_fg #5,2GHHHrr(cV|ddks J||d}t|trd|vs J|tjd|dr$|ddd|d<|d|diS)0Converts the OpenAI chunk to the Mistral format.r!rr:z^data:image/\w+;base64,,r)get isinstancer.rematchsplitmodel_validate)r+r(image_url_dicts rr,zImageChunk.from_openai\s'';666 666%k2.$//[E^4K4K4K^4K4KK 8.u0E F F H$25$9$?$?$D$DQ$GN5 !!!7N5,A"BCCCrN)rrrrrrr!rr-rr model_configr.r/r'r0r,rrrr2r2Hs'1&6D'*" #666 :d;;;LI4S4S>%9 9:IIII  DtCtCH~1E,E'F D< D D D[ D D Drr2c2eZdZUdZeed<dZedzed<dS)ImageURLzImage URL or a base64 encoded image. Attributes: url: The URL of the image. detail: The detail of the image. Examples: >>> image_url = ImageURL(url="https://example.com/image.png") r:Ndetail)rrrrr/r-rMrrrrLrLjs; HHHFC$JrrLceZdZUdZejZeejed<e e zed<e dZ de fdZ dee e ee e fzffdZed ee e ee e fzfddfd Zd S) ImageURLChunkzImage URL chunk. Attributes: image_url: The URL of the image or a base64 encoded image to be sent to the model. Examples: >>> image_url_chunk = ImageURLChunk(image_url="data:image/png;base64,iVBORw0") r!rTr3r"c\t|jtr |jjS|jSN)rDrrLr:r%s rget_urlzImageURLChunk.get_urls) dnh / / &>% %~rcd|i}t|jtr|jj|jj|d<d|d}|S)r6r:NrMrr;)rRrDrrLrM)r&rIout_dicts rr'zImageURLChunk.to_openais\0 dnh / / =DN4I4U'+~'D'*& '>>>#~:d;;;L 4S4S>%9 9:    LtCtCH~1E,E'FL?LLL[LLLrrOceZdZUdZeezed<eed<ededdfdZ e ddedefdZ d S) RawAudioaKBase64 encoded audio data. This class represents raw audio data encoded in base64 format. Attributes: data: The base64 encoded audio data, which can be a string or bytes. format: The format of the audio data. Examples: >>> audio = RawAudio(data="base64_encoded_audio_data", format="mp3") dataformataudior"cV|j}||d}|||S)zCreates a RawAudio instance from an Audio object. Args: audio: An Audio object containing audio data, format, and duration. Returns: An AudioChunk instance initialized with the audio data. FrYrZ)rZ to_base64)r+r[rZrYs r from_audiozRawAudio.from_audios3vu--sV,,,,rvcN|tvrtdtd|d|S)Nz`format` should be one of z. Got: `)r r r+r`s rshould_not_be_emptyzRawAudio.should_not_be_emptys7 * * *!"b?U"b"b^_"b"b"bcc crN) rrrrr/bytesr-r0r r_r rdrrrrXrXs   + KKK -u - - - -[ -_XCCrrXceZdZUdZeed<dS)AudioURLzHAudio URL. Attributes: url: The URL of the audio file. r:N)rrrrr/r-rrrrgrgs% HHHHHrrgc"eZdZdZdZdZdZdZdS) AudioURLTypezEnum for the types of audio URLs. Attributes: url: A URL. base64: A base64 encoded audio. Can be prefixed with `data:audio/;base64,`. file: A file path. file_uri: A file URI (eg. `file:///path/to/file`). r:base64filefile_uriN)rrrrr:rjrkrlrrrriris. C F DHHHrriceZdZUdZejZeejed<e e zed<e de fdZ de fdZdee e ee e fzffdZedee e ee e fzfddfd Zd S) AudioURLChunkzAudio URL chunk. Attributes: type: The type of the chunk, which is always `ChunkTypes.audio_url`. audio_url: The URL of the audio file. r!rr"c\t|jtr |jjS|jSrQ)rDrrgr:r%s rr:zAudioURLChunk.urls) dnh / / &>% %~rcHt|jj}|dvr tjS|dkr tjS|dkr tjS t |j}|}n#t$rd}YnwxYw|r tj StjS)aReturns the type of the audio URL. Note: URLs should be either: - a valid URL (http:// or https://) - a valid file path (e.g. /path/to/file) - a valid file URI (e.g. file:///path/to/file) - a base64 encoded audio. It is assumed to be base64 encoded if it is not a valid URL or file path. Returns: The type of the audio URL. >httphttpsrYrkF) rr:schemerirjrlrexistsOSErrorrk)r& url_schemeurl_path exist_paths r get_url_typezAudioURLChunk.get_url_typesdh''. * * *# # 6 ! !& & 6 ! !( ( DH~~H!**JJ   JJJ   %$ $""s(A88 BBct|jtr|S|jd|jidS)r6r:)r!r)rDrrgr=r!r%s rr'zAudioURLChunk.to_openais> dnh / / M??$$ $ IUDN4KLL Lrr(c,||Sr@rVr*s rr,zAudioURLChunk.from_openai!!,///rN)rrrrrrr!rr-r/rgpropertyr:riryr.r'r0r,rrrrnrns +5*>D'*& '>>>X~ SX #l####@M4S4S>%9 9:MMMM0tCtCH~1E,E'F0?000[000rrnceZdZUdZejZeejed<e ed<e dde de fdZ e de ddfdZdeeeeeefzffd Ze d eeeeeefzfddfd Zd S) AudioChunkaAudio chunk containing raw audio data. This class represents a chunk of audio data that can be used as input. Attributes: type: The type of the chunk, which is always ChunkTypes.input_audio. input_audio: The RawAudio object containing the audio data. Examples: >>> audio_chunk = AudioChunk(input_audio=RawAudio(data="base64_encoded_audio_data", format="mp3")) r!rr`r"c^|jstd|d|S)Nz'`InputAudio` should not be empty. Got: rb)rYstripr rcs rrdzAudioChunk.should_not_be_empty.s4v||~~ R!"PA"P"P"PQQ Qrr[cJ|t|S)zCreates an AudioChunk instance from an Audio object. Args: audio: An Audio object containing audio data. Returns: An AudioChunk instance initialized with the audio data. )r)rXr_)r+r[s rr_zAudioChunk.from_audio5s%sx22599::::rct|jjtr|jjdn |jj}|jt ||jjdS)zConverts the chunk to the OpenAI format. Returns: A dictionary representing the audio chunk in the OpenAI format. zutf-8r])r!r) rDrrYredecoder!rXrZr=)r&contents rr'zAudioChunk.to_openaiAs{6@@P@UW\5]5] xD  ! ( ( 1 1 1cgcscx I#9I9PQQQ\\^^   rr(c,||S)aConverts the OpenAI chunk to the Mistral format. Args: openai_chunk: A dictionary representing the audio chunk in the OpenAI format. Returns: An AudioChunk instance initialized with the data from the OpenAI chunk. rVr*s rr,zAudioChunk.from_openaiOs!!,///rN)rrrrrrr!rr-rXr rdr0r r_r.r/r'r,rrrrrs%  -7,BD'*( )BBB_]##H$#  ;u ; ; ; ;[ ;  4S4S>%9 9:      0tCtCH~1E,E'F 0< 0 0 0[ 0 0 0rrceZdZUdZejZeejed<e ed<de e e e e e fzffdZ e de e e e e e fzfddfdZ dS) TextChunkzText chunk. Attributes: text: The text to be sent to the model. Examples: >>> text_chunk = TextChunk(text="Hello, how can I help you?") r!rr"c*|Sr6r<r%s rr'zTextChunk.to_openaii   rr(c,||Sr|rVr*s rr,zTextChunk.from_openaimr}rN)rrrrrrr!rr-r/r.r'r0r,rrrrr\s&0_D'*/ "444 III!4S4S>%9 9:!!!!0tCtCH~1E,E'F0;000[000rrceZdZUdZejZeejed<e ed<e ddZ e ed<de e e e e e fzffd Zed e e e e e e fzfddfd Zd S) ThinkChunkzThinking chunk. Attributes: type: The type of the chunk, which is always ChunkTypes.thinking. thinking: The list of text chunks of the thinking. closed: Whether the thinking chunk is closed or not. r!rTz,Whether the thinking chunk is closed or not.)default descriptionclosedr"c*|Srr<r%s rr'zThinkChunk.to_openairrr(c,||Sr|rVr*s rr,zThinkChunk.from_openair}rN)rrrrrrr!rr-r/rrboolr.r'r0r,rrrrrss*4)%9 9:!!!!0tCtCH~1E,E'F0<000[000rrr!) discriminatoropenai_content_chunksr"c|d}|tdt|}|tjkrt|S|tjkrt|S|tjkrt|S|tj krt|S|tj krt|S|tjkrt|Std|)Nr!z%Content chunk must have a type field.zUnknown content chunk type: )rC ValueErrorrrrr,rrOrr2rrrrnrr)rcontent_type_str content_types r_convert_openai_content_chunksrs!,0088@AAA.//Lz&&$$%:;;; - - -(()>??? ) ) )%%&;<<< / / /%%&;<<< - - -(()>??? , , ,%%&;<<<F FFGGGr)(rEenumrpathlibrtypingr urllib.parserpydanticrrr r typing_extensionsr mistral_common.audior r mistral_common.basermistral_common.imagerr/rr r2rLrOrXrgrirnrrr ContentChunkUserContentChunkr.rrrrrs !!!!!!HHHHHHHHHHHH''''''>>>>>>>>++++++222222d.\\\\\{\\\DDDDDD!DDDD     {   #L#L#L#L#L$#L#L#LL$$$$${$$$N     {        3    ;0;0;0;0;0$;0;0;0|;0;0;0;0;0!;0;0;0|00000 000.00000!000. ]*Z7-G*TV[V[jpVqVqVqq  ]*Z7-G]cIdIdIdd H$sC$sTWx.DX?X:YH^jHHHHHHr