o ưi@sHddlZddlmZddlZddlmZddlmZGdddeZdS)N)Optional)OpenAITextCompletionConfig)GenericStreamingChunkcs$eZdZUdZdZeeed<dZee ed<dZ ee ed<dZ ee ed<dZ ee ed<dZee ed<        ddeedee d eedee dee dee dee d eed dfd d ZefddZdefddZdededede d ef ddZded efddZZS)CodestralTextCompletionConfigzO Reference: https://docs.mistral.ai/api/#operation/createFIMCompletion Nsuffix temperature max_tokens min_tokensstream random_seedtop_pstopreturnc Cs>t} | D]\} } | dkr| durt|j| | q dS)Nself)localscopyitemssetattr __class__) rrrr rr r r r Zlocals_keyvaluerg/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/codestral/completion/transformation.py__init__s z&CodestralTextCompletionConfig.__init__cs tS)N)super get_config)clsrrrr%s z(CodestralTextCompletionConfig.get_configmodelcCsgdS)N)rrr rmax_completion_tokensr seedr r)rrrrrget_supported_openai_params)sz9CodestralTextCompletionConfig.get_supported_openai_paramsnon_default_paramsoptional_params drop_paramscCs|D]L\}}|dkr||d<|dkr||d<|dkr ||d<|dks(|dkr,||d<|dkr8|dur8||d<|dkr@||d<|d krH||d <|d krP||d <q|S) Nrrr rrr Tr r r r )r)rr"r#rr$paramrrrrmap_openai_params5s&z/CodestralTextCompletionConfig.map_openai_params chunk_datac Csd}d}d}d}tj|pd}|}t|dks|dkr$d||dSzt|}Wntjy;d||dYSwtjd i|}| dgpKg}t|dkrXd||dS|d} | di dd}| d duryd }| d }| d }t |||||d S)NFrz[DONE])text is_finished finish_reasonchoicesdeltacontentr+Tlogprobs)r)original_chunkr*r+r/r) litellmZCustomStreamWrapperZ_strip_sse_data_from_chunkstriplenjsonloadsJSONDecodeErrorZModelResponseStreamgetr) rr'r)r*r+r/Zchunk_data_dictr0Z_choicesZ_choicerrr _chunk_parserPsP    z+CodestralTextCompletionConfig._chunk_parser)NNNNNNNN)__name__ __module__ __qualname____doc__rrstr__annotations__rintrr r boolr floatr classmethodrr!dictr&rr8 __classcell__rrrrr sd     r) r4typingrr1Z-litellm.llms.openai.completion.transformationrZlitellm.types.llms.databricksrrrrrrs