o Zhn@sjddlZddlmZmZmZmZddlZddlmZddl m Z ddl m Z e eZGddde ZdS)N)AnyListMappingOptional)CallbackManagerForLLMRun)LLM)enforce_stop_tokensc @seZdZUdZdZeed< dZee ed< dZ e ed< dZ e ed < gZeeed < d Ze ed < d Zeed< edefddZedeeeffddZ  ddedeeedeededef ddZdS)ChatGLMa.ChatGLM LLM service. Example: .. code-block:: python from langchain_community.llms import ChatGLM endpoint_url = ( "http://127.0.0.1:8000" ) ChatGLM_llm = ChatGLM( endpoint_url=endpoint_url ) zhttp://127.0.0.1:8000/ endpoint_urlN model_kwargsi N max_tokeng? temperaturehistorygffffff?top_pF with_historyreturncCsdS)NZchat_glm)selfrrW/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/chatglm.py _llm_type+szChatGLM._llm_typecCs |jpi}id|jid|iS)zGet the identifying parameters.r r )r r )r _model_kwargsrrr_identifying_params/s zChatGLM._identifying_paramspromptstop run_managerkwargsc KsZ|jpi}ddi}||j|j|j|jd}||||td|z tj |j ||d}Wntj j yG} zt d| d} ~ wwtd||jd kr\t d |z$|} t| trxd } | | vrq| | } nt d | t d | Wntj jy} z t d| d|jd} ~ ww|durt| |} |jr| d|_| S)awCall out to a ChatGLM LLM inference endpoint. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: The string generated by the model. Example: .. code-block:: python response = chatglm_llm.invoke("Who are you?") z Content-Typezapplication/json)rr r max_lengthrzChatGLM payload: )headersjsonz$Error raised by inference endpoint: NzChatGLM response: zFailed with response: responsezNo content in response : zUnexpected response type: z?Error raised during decoding response from inference endpoint: z . Response: r)r r rr rupdateloggerdebugrequestspostr exceptionsRequestException ValueError status_coder isinstancedictJSONDecodeErrortextrr) rrrrrrrpayloadr eZparsed_responseZ content_keysr-rrr_call8sT        z ChatGLM._call)NN)__name__ __module__ __qualname____doc__r str__annotations__r rr+r intr floatrrrrboolpropertyrrrrrr0rrrrr sB        r )loggingtypingrrrrr$Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_community.llms.utilsr getLoggerr1r"r rrrrs