昺h^dZddlmZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z ddlmZmZmZmZddlmZmZdd lmZejeZGd d eZGd d eZGddee ZdS)zWrapper around Minimax APIs.) annotationsN)AnyDictListOptional)CallbackManagerForLLMRun)LLM) BaseModelField SecretStrroot_validator)convert_to_secret_strget_from_dict_or_env)enforce_stop_tokenscreZdZUdZded<ded<ded<ded<edd dd ZddZdS)_MinimaxEndpointClientz3An API client that talks to a Minimax llm endpoint.strhostgroup_idr api_keyapi_urlT)pre allow_reusevaluesDict[str, Any]returncFd|vr|d}|d}|d|}||d<|S)Nrrrz /v1/text/chatcompletion?GroupId=)clsrrrrs h/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/llms/minimax.py set_api_urlz"_MinimaxEndpointClient.set_api_url!sB F " "&>Dj)HIIxIIG 'F9  requestrcdd|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 r.z_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 r!r.rr"r rrs== IIIMMMLLL^$///0/ ( ( ( ( ( (r"rceZdZUdZded<dZded< dZded < d Zd ed < d Zd ed< e e Z ded< dZ ded<dZ ded<dZded<eddZed dZed dZed!dZdS)" 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_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.rGMINIMAX_API_KEYrFMINIMAX_GROUP_IDrEMINIMAX_API_HOSTzhttps://api.minimax.chat)default)rrrr;)rrr)rrs r validate_environmentz"MinimaxCommon.validate_environmentJs%: ):>####EJAKQED#(5#>#>#>L>>>>V&*****&*****+/O////^,   X AAAXAXr"r:c eZdZdZ d dd ZdS)MinimaxaWrapper around Minimax 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_typer1messages)rQupdater;r.r)r2r[r\r^r`r#r1s r _callz Minimax._callsd &/5vFFG v|  ))  'tT22D r")NN) r[rr\r]r^r_r`rrr)r4r5r6r7rfrr"r rZrZws?  %):> r"rZ)r7 __future__rloggingtypingrrrrr-langchain_core.callbacksr#langchain_core.language_models.llmsr langchain_core.pydantic_v1r r r r langchain_core.utilsrrlangchain_community.llms.utilsr getLoggerr4loggerrr:rZrr"r rqs""""""""433333RRRRRRRRRRRRLLLLLLLL>>>>>>  8 $ $(((((Y(((><<<<<I<<<~%%%%%mS%%%%%r"