a bgn@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<eedddZeeeefdddZdeeeeeeeedddZdS)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_history)returncCsdS)NZchat_glm)selfrrn/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/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_params)promptstop run_managerkwargsrc Ksz|jpi}ddi}||j|j|j|jd}||||td|ztj |j ||d}Wn6tj j y} zt d| WYd} ~ n d} ~ 00td||jd krt d |zJ|} t| trd } | | vr| | } nt d | nt d | Wn@tj jyN} z"t d| d|jWYd} ~ n d} ~ 00|durdt| |} |jrv| 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_call8sJ   $      z ChatGLM._call)NN)__name__ __module__ __qualname____doc__r str__annotations__r rr+r intr floatrrrrboolpropertyrrrrrr0rrrrr s*        r )loggingtypingrrrrr$Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_community.llms.utilsr getLoggerr1r"r rrrrs