U hP@sdZddlmZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z ddlmZmZmZmZddlmZmZmZdd lmZeeZGd d d eZGd d d eZGdddee ZdS)zWrapper around Minimax APIs.) annotationsN)AnyDictListOptional)CallbackManagerForLLMRun)LLM) BaseModelField SecretStrroot_validator)convert_to_secret_strget_from_dict_or_envpre_init)enforce_stop_tokensc@s\eZdZUdZded<ded<ded<ded<edd d d d d d ZdddddZdS)_MinimaxEndpointClientz(API client for the Minimax LLM endpoint.strhostgroup_idr api_keyapi_urlT)preDict[str, Any]valuesreturncCs2d|kr.|d}|d}|d|}||d<|S)Nrrrz /v1/text/chatcompletion?GroupId=)clsrrrrrrD/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/minimax.py set_api_url"s z"_MinimaxEndpointClient.set_api_urlr)requestrcCsdd|ji}tj|j||d}|jsDtd|jd|j| dddkrtd | 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)__name__ __module__ __qualname____doc____annotations__r rr&rrrrrs rc@seZdZUdZded<dZded<dZded <d Zd ed <d Zd ed<e e dZ ded<dZ ded<dZ ded<dZded<edddddZeddddZedddd Zeddd!d"ZdS)# MinimaxCommonz4Common parameters for Minimax large language models.r_clientz abab5.5-chatrmodelint max_tokensgffffff?float temperaturegffffff?top_p)default_factoryr model_kwargsNz Optional[str]minimax_api_hostminimax_group_idzOptional[SecretStr]minimax_api_keyrrcCsZtt|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)rrrr2)r rr)rrrrrvalidate_environmentKs(   z"MinimaxCommon.validate_environment)rcCs|j|j|j|jd|jS)z2Get the default parameters for calling OpenAI API.)r3Ztokens_to_generater8r9)r3r6r8r9r;r*rrr_default_paramsbszMinimaxCommon._default_paramscCsd|ji|jS)zGet the identifying parameters.r3)r3rBrArrr_identifying_paramsmsz!MinimaxCommon._identifying_paramscCsdS)zReturn type of llm.ZminimaxrrArrr _llm_typerszMinimaxCommon._llm_type)r,r-r.r/r0r3r6r8r9r dictr;r<r=r>rr@propertyrBrCrDrrrrr19s$         r1c@s(eZdZdZd dddddddd 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") NrzOptional[List[str]]z"Optional[CallbackManagerForLLMRun]r)promptstop run_managerkwargsrcKsB|j}d|dg|d<|||j|}|dk r>t||}|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)rBupdater2r&r)r*rHrIrJrKr r)rrr_calls   z Minimax._call)NN)r,r-r.r/rOrrrrrGxsrG)r/ __future__rloggingtypingrrrrr%Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.pydantic_v1r r r r Zlangchain_core.utilsr rrZlangchain_community.llms.utilsr getLoggerr,loggerrr1rGrrrrs     ?