a bg@sddlmZmZmZmZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZmZmZmZddlmZdZGd d d eZGd d d eZGd ddee ZdS))AnyDictListOptionalN)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDictField SecretStrmodel_validator)enforce_stop_tokenszhttps://api.moonshot.cn/v1c@s6eZdZUdZeed<eZeed<e e dddZ dS)_MoonshotClientz0An API client that talks to the Moonshot server.api_keybase_url)requestreturncCsbdd|ji}tj|jd||d}|jsJtd|jd|j| ddd d S) N AuthorizationzBearer z/chat/completions)headersjsonzHTTP z error: choicesrmessagecontent) rZget_secret_valuerequestspostrok ValueError status_codetextr)selfrrresponser$o/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/moonshot.py completions z_MoonshotClient.completionN) __name__ __module__ __qualname____doc__r__annotations__MOONSHOT_SERVICE_URL_BASErstrrr&r$r$r$r%rs  rc@seZdZUdZeed<eZeed<e dddZ e e ed<e dd dZ eed <d Zeed <d Zeed<edddZeedddZeeeefdddZeeeefdddZeddeeeefedddZeeedddZeedd d!ZdS)"MoonshotCommonz$Common parameters for Moonshot LLMs.clientrNr)defaultaliasmoonshot_api_keyzmoonshot-v1-8kmodel model_namei max_tokensg333333? temperatureTr$)populate_by_nameZprotected_namespaces)rcCsddiS)zA map of constructor argument names to secret ids. For example, {"moonshot_api_key": "MOONSHOT_API_KEY"} r2MOONSHOT_API_KEYr$r"r$r$r% lc_secrets7szMoonshotCommon.lc_secretscCs|j|j|jdS)z2Get the default parameters for calling OpenAI API.)r3r5r6)r4r5r6r9r$r$r%_default_params@szMoonshotCommon._default_paramscCsid|ji|jS)Nr3)r4r;r9r$r$r%_invocation_paramsIsz!MoonshotCommon._invocation_paramsbefore)mode)valuesrcCs|S)zBuild extra parameters. Override the superclass method, prevent the model parameter from being overridden. r$clsr?r$r$r% build_extraMszMoonshotCommon.build_extracCs<tt|dd|d<t|dd|vr,|dntd|d<|S)z?Validate that api key and python package exists in environment.r2r8r)rrr/)rr rr,r@r$r$r%validate_environmentVs   z#MoonshotCommon.validate_environmentcCsdS)zReturn type of llm.Zmoonshotr$r9r$r$r% _llm_typeeszMoonshotCommon._llm_type) r'r(r)r*rr+r,rr-r r2rrr4r5intr6floatr model_configpropertydictr:rr;r<r classmethodrBr rCrDr$r$r$r%r.'s*     r.c@s>eZdZdZeddZdeeeeee e edddZ dS) MoonshotafMoonshot large language models. To use, you should have the environment variable ``MOONSHOT_API_KEY`` set with your API key. Referenced from https://platform.moonshot.cn/docs Example: .. code-block:: python from langchain_community.llms.moonshot import Moonshot moonshot = Moonshot(model="moonshot-v1-8k") T)r7N)promptstop run_managerkwargsrcKsB|j}d|dg|d<|||j|}|dur>t||}|S)Nuser)Zrolermessages)r<updater/r&r)r"rLrMrNrOrr!r$r$r%_call}s   zMoonshot._call)NN) r'r(r)r*r rGr-rrrrrSr$r$r$r%rKks  rK)typingrrrrrZlangchain_core.callbacksrZlangchain_core.language_modelsrZlangchain_core.utilsrr r Zpydanticr r r rrZlangchain_community.llms.utilsrr,rr.rKr$r$r$r%s   D