o Zh@sdZddlmZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z ddlmZmZmZddlmZmZmZmZmZdd lmZeeZGd d d eZGd d d eZGdddee ZdS)zWrapper around Minimax APIs.) annotationsN)AnyDictListOptional)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDictField SecretStrmodel_validator)enforce_stop_tokensc@sTeZdZUdZded<ded<ded<ded<edd edddZdddZdS)_MinimaxEndpointClientz(API client for the Minimax LLM endpoint.strhostgroup_idrapi_keyapi_urlbefore)modevaluesDict[str, Any]returnrcCs2d|vr|d}|d}|d|}||d<|S)Nrrrz /v1/text/chatcompletion?GroupId=)clsrrrrrrW/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/minimax.py set_api_url"s z"_MinimaxEndpointClient.set_api_urlrequestcCsdd|ji}tj|j||d}|js"td|jd|j| dddkrBtd | ddd| dd | d S) N AuthorizationzBearer )headersjsonzHTTP z error: Z base_resp status_coderzAPI Z status_msgZreply) rZget_secret_valuerequestspostrok ValueErrorr%textr$)selfr!r#responserrrr',s z_MinimaxEndpointClient.postN)rrrr)r!rrr) __name__ __module__ __qualname____doc____annotations__r classmethodr r'rrrrrs rc@seZdZUdZeddZded<dZded< d Zd ed < d Z d ed< dZ d ed< e e dZ ded< dZded<dZded<dZded<ed%ddZed&dd Zed&d!d"Zed'd#d$ZdS)( MinimaxCommonz4Common parameters for Minimax large language models.r)Zprotected_namespacesr_clientz abab5.5-chatrmodelint max_tokensgffffff?float temperaturegffffff?top_p)default_factoryr model_kwargsNz Optional[str]minimax_api_hostminimax_group_idzOptional[SecretStr]minimax_api_keyrrrcCsZtt|dd|d<t|dd|d<t|dddd|d<t|d|d|dd |d <|S) z?Validate that api key and python package exists in environment.r@ZMINIMAX_API_KEYr?ZMINIMAX_GROUP_IDr>ZMINIMAX_API_HOSTzhttps://api.minimax.chat)default)rrrr4)r r r)rrrrrvalidate_environmentNs$   z"MinimaxCommon.validate_environmentcCs|j|j|j|jd|jS)z2Get the default parameters for calling OpenAI API.)r5Ztokens_to_generater:r;)r5r8r:r;r=r+rrr_default_paramseszMinimaxCommon._default_paramscCsid|ji|jS)zGet the identifying parameters.r5)r5rDrCrrr_identifying_paramspsz!MinimaxCommon._identifying_paramscCsdS)zReturn type of llm.ZminimaxrrCrrr _llm_typeuszMinimaxCommon._llm_type)rrrr)rr)rr)r-r.r/r0r Z model_configr1r5r8r:r;rdictr=r>r?r@r rBpropertyrDrErFrrrrr3:s2            r3c@s eZdZdZ  ddd d ZdS)MinimaxaMinimax large language models. To use, you should have the environment variable ``MINIMAX_API_KEY`` and ``MINIMAX_GROUP_ID`` set with your API key, or pass them as a named parameter to the constructor. Example: . code-block:: python from langchain_community.llms.minimax import Minimax minimax = Minimax(model="", minimax_api_key="my-api-key", minimax_group_id="my-group-id") NpromptrstopOptional[List[str]] run_manager"Optional[CallbackManagerForLLMRun]kwargsrrcKsB|j}d|dg|d<|||j|}|durt||}|S)a)Call out to Minimax's completion endpoint to chat Args: prompt: The prompt to pass into the model. Returns: The string generated by the model. Example: .. code-block:: python response = minimax("Tell me a joke.") USER)Z sender_typer*messagesN)rDupdater4r'r)r+rJrKrMrOr!r*rrr_calls   z Minimax._call)NN) rJrrKrLrMrNrOrrr)r-r.r/r0rSrrrrrI{s rI) r0 __future__rloggingtypingrrrrr&Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r r Zpydanticr r rrrZlangchain_community.llms.utilsr getLoggerr-loggerrr3rIrrrrs      A