NgdZddlmZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z ddlmZmZmZddlmZmZmZmZmZdd lmZejeZGd d eZGd d eZGddee ZdS)zWrapper around Minimax APIs.) annotationsN)AnyDictListOptional)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDictField SecretStrmodel_validator)enforce_stop_tokensceZdZUdZded<ded<ded<ded<ed eddZddZdS)_MinimaxEndpointClientz(API client for the Minimax LLM endpoint.strhostgroup_idrapi_keyapi_urlbefore)modevaluesDict[str, Any]returnrcFd|vr|d}|d}|d|}||d<|S)Nrrrz /v1/text/chatcompletion?GroupId=)clsrrrrs \/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/minimax.py set_api_urlz"_MinimaxEndpointClient.set_api_url"sD F " "&>Dj)HIIxIIG 'F9  requestcdd|ji}tj|j||}|jst d|jd|j| dddkrQt d | ddd| dd | d S) N AuthorizationzBearer )headersjsonzHTTP z error: base_resp status_coderzAPI status_msgreply) rget_secret_valuerequestspostrok ValueErrorr+textr))selfr%r(responses r"r0z_MinimaxEndpointClient.post,s"$Odl.K.K.M.M$O$OP=wWMMM{ TRX%9RR8=RRSS S ==??; ' 6 : :Hx}}{3MBHH#==??;7 EHH }}w''r$N)rrrr)r%rrr) __name__ __module__ __qualname____doc____annotations__r classmethodr#r0r r$r"rrs22 IIIMMMLLL_(###[$# ( ( ( ( ( (r$rc$eZdZUdZedZded<dZded< d Zd ed < d Z d ed< dZ d ed< e e Z ded< dZded<dZded<dZded<ed!dZed"dZed"dZed#d ZdS)$ MinimaxCommonz4Common parameters for Minimax large language models.r )protected_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_keyrrrctt|dd|d<t|dd|d<t|ddd|d<t|d|d|d |d <|S) z?Validate that api key and python package exists in environment.rKMINIMAX_API_KEYrJMINIMAX_GROUP_IDrIMINIMAX_API_HOSTzhttps://api.minimax.chat)default)rrrr?)r r r)r!rs r"validate_environmentz"MinimaxCommon.validate_environmentNs%: ):>:2666L####EJAKQED#(5#>#>#>L>>>>V&*****&*****+/O//// X,   X AAAXAXr$r=c eZdZdZ d dd 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]kwargsrrc |j}d|dg|d<|||j|}|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) sender_typer3messages)rUupdater?r0r)r4r`rarcrer%r3s r"_callz Minimax._callsd &/5vFFG v|  ))  'tT22D r$)NN) r`rrarbrcrdrerrr)r6r7r8r9rkr r$r"r_r_{s?  %):> r$r_) r9 __future__rloggingtypingrrrrr/langchain_core.callbacksr#langchain_core.language_models.llmsr langchain_core.utilsr r r pydanticr rrrrlangchain_community.llms.utilsr getLoggerr6loggerrr=r_r r$r"rvs""""""""433333VVVVVVVVVVMMMMMMMMMMMMMM>>>>>>  8 $ $(((((Y(((@>>>>>I>>>B&&&&&mS&&&&&r$