a bg@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@s`eZdZUdZded<ded<ded<ded<edd ed d d d dZd d dddZdS)_MinimaxEndpointClientz(API client for the Minimax LLM endpoint.strhostgroup_idrapi_keyapi_urlbefore)modeDict[str, Any]rvaluesreturncCs2d|vr.|d}|d}|d|}||d<|S)Nrrrz /v1/text/chatcompletion?GroupId=)clsrrrrrrn/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/minimax.py set_api_url"s z"_MinimaxEndpointClient.set_api_url)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$responserrr r(,sz_MinimaxEndpointClient.postN) __name__ __module__ __qualname____doc____annotations__r classmethodr!r(rrrr rs rc@seZdZUdZeddZded<dZded<d Zd ed <d Z d ed<dZ d ed<e e dZ ded<dZded<dZded<dZded<edddddZedddd Zeddd!d"Zeddd#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_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.rAZMINIMAX_API_KEYr@ZMINIMAX_GROUP_IDr?ZMINIMAX_API_HOSTzhttps://api.minimax.chat)default)rrrr5)r r r)rrrrr validate_environmentNs$   z"MinimaxCommon.validate_environment)rcCs|j|j|j|jd|jS)z2Get the default parameters for calling OpenAI API.)r6Ztokens_to_generater;r<)r6r9r;r<r>r,rrr _default_paramseszMinimaxCommon._default_paramscCsid|ji|jS)zGet the identifying parameters.r6)r6rErDrrr _identifying_paramspsz!MinimaxCommon._identifying_paramscCsdS)zReturn type of llm.ZminimaxrrDrrr _llm_typeuszMinimaxCommon._llm_type)r.r/r0r1r Z model_configr2r6r9r;r<rdictr>r?r@rAr rCpropertyrErFrGrrrr r4:s&          r4c@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|}|dur>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)rEupdater5r(r)r,rKrLrMrNr"r+rrr _calls   z Minimax._call)NN)r.r/r0r1rRrrrr rJ{srJ) r1 __future__rloggingtypingrrrrr'Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r r Zpydanticr r rrrZlangchain_community.llms.utilsr getLoggerr.loggerrr4rJrrrr s      A