jddlmZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZmZmZmZGdd eZGd d e eZy ) ) annotations)AnyDictIteratorListOptional)CallbackManagerForLLMRun)LLM)GenerationChunk)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDictField SecretStrc$eZdZUdZedZdZded<dZded< dZ ded < d Z d ed < d Z d ed< dZ ded< dZ d ed< dZded< dZded< eeZded< dZded< dZded < dZded!< ed$d"Zed%d#Zy)&VolcEngineMaasBasez%Base class for VolcEngineMaas models.)protected_namespacesNrclientzOptional[SecretStr]volc_engine_maas_akvolc_engine_maas_skz*maas-api.ml-platform-cn-beijing.volces.comz Optional[str]endpointRegionregionzskylark-lite-publicstrmodel model_versiong?zOptional[float]top_pgffffff? temperature)default_factoryDict[str, Any] model_kwargsFbool streaming<z Optional[int]connect_timeout read_timeoutctt|dd}tt|dd}|d}|d |ddk7r|d} ddlm}|||d |d |d  }|j |j |j |j ||d<||d<||d <|S#t$r tdwxYw)NrVOLC_ACCESSKEYrVOLC_SECRETKEYrr) MaasServicerr(r))connection_timeoutsocket_timeoutrzMvolcengine package not found, please install it with `pip install volcengine`)r r volcengine.maasr.set_akget_secret_valueset_sk ImportError)clsvaluesrrrr.maass r/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/llms/volcengine_maas.pyvalidate_environmentz'VolcEngineMaasBase.validate_environment9s 3 )>@P Q 4 )>@P Q *% *  )fZ.@B.Fj)H  3x #)*;#<%n5 D KK+<<> ? KK+<<> ?,?F( ),?F( )#F8    +  s A(B//CcT|j|jd}i||jS)z:Get the default parameters for calling VolcEngineMaas API.)r r!)r r!r$)self normal_paramss r9_default_paramsz"VolcEngineMaasBase._default_paramsZs3ZZ++ 6-54#4#455)r7rreturnr)r@r#)__name__ __module__ __qualname____doc__r model_configr__annotations__rrrrrrr r!rdictr$r&r(r)rr:propertyr>rr?r9rr s/26LFC/3,3$/3,3$JHmJ-$FM$+&E3&>$(M='R!E? D#'K'Q#(#>L.>GIt(%'O]'R"$L-$@66r?rcveZdZdZeddZ ddZ d d dZ d d dZy) VolcEngineMaasLLMa0volc engine maas hosts a plethora of models. You can utilize these models through this class. To use, you should have the ``volcengine`` python package installed. and set access key and secret key by environment variable or direct pass those to this class. access key, secret key are required parameters which you could get help https://www.volcengine.com/docs/6291/65568 In order to use them, it is necessary to install the 'volcengine' Python package. The access key and secret key must be set either via environment variables or passed directly to this class. access key and secret key are mandatory parameters for which assistance can be sought at https://www.volcengine.com/docs/6291/65568. Example: .. code-block:: python from langchain_community.llms import VolcEngineMaasLLM model = VolcEngineMaasLLM(model="skylark-lite-public", volc_engine_maas_ak="your_ak", volc_engine_maas_sk="your_sk") cy)zReturn type of llm.zvolc-engine-maas-llmr)r<s r9 _llm_typezVolcEngineMaasLLM._llm_type~s&r?c dd|jii}|j|j|dd<i|d|dgi|j|dS)Nrnameversionuser)rolecontent)messages parameters)rrr>)r<promptkwargs model_reqs r9_convert_prompt_msg_paramsz,VolcEngineMaasLLM._convert_prompt_msg_paramsst      ),0,>,>Ig y )  "(V<=>J%fdKJ6Jejj( K 000B6B;;##F+||Hb)--i<@@BOOr?c+6K|j|fi|}|jj|D]d}|st|j dij dij dd}|r|j |j ||fyw)NrZr[rRr-)r])rc)rXr stream_chatr r_on_llm_new_tokenr])r<rUr`rarVrdresrcs r9r\zVolcEngineMaasLLM._streams100B6B;;**62C'2.229bAEEiQST0050I 3s 4BA"B)r@r)rUrrVrr@rG)NN) rUrr`Optional[List[str]]ra"Optional[CallbackManagerForLLMRun]rVrr@r) rUrr`rkrarlrVrr@zIterator[GenerationChunk]) rArBrCrDrHrLrXrfr\rr?r9rJrJes0&&      ,%):> PP"P8 P  P P(%):> "8    # r?rJN) __future__rtypingrrrrrlangchain_core.callbacksr #langchain_core.language_models.llmsr langchain_core.outputsr langchain_core.utilsr r rpydanticrrrrrrJrr?r9rtsC"66=32VV<<V6V6rQ/Qr?