NgnddlZddlmZmZmZmZddlZddlmZddl m Z ddl m Z ej eZGdde ZdS)N)AnyListMappingOptional)CallbackManagerForLLMRun)LLM)enforce_stop_tokensc *eZdZUdZdZeed< dZee ed< dZ e ed< dZ e ed < gZeeed < d Ze ed < d Zeed< edefdZedeeeffdZ ddedeeedeededef dZdS)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_historyreturncdS)Nchat_glm)selfs \/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/chatglm.py _llm_typezChatGLM._llm_type+szc2|jpi}id|jid|iS)zGet the identifying parameters.r r )r r )r _model_kwargss r_identifying_paramszChatGLM._identifying_params/s8)/R  t01 }-  rpromptstop run_managerkwargsc f|jpi}ddi}||j|j|j|jd}||||t d| tj |j ||}n.#tj j $r} td| d} ~ wwxYwt d||jd krtd | |} t!| t"r!d } | | vr | | } n$td | td | n6#tj j$r} td| d|jd} ~ wwxYw|t)| |} |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)rrr 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 rrrrupdateloggerdebugrequestspostr exceptionsRequestException ValueError status_coder% isinstancedictJSONDecodeErrortextr r) rrrr r!rr$payloadr'eparsed_response content_keysr4s r_callz ChatGLM._call8s+,)/R "#56+|.Z    }%%%v 222333 I}T%6gVVVHH"3 I I IGAGGHH H I  4(44555  3 & &@h@@AA A &mmooO/400 Q) ?22*<8DD$%R%R%RSSS !Oo!O!OPPP  "2   /RS//'}//    &tT22D   6*95DL s12BB:#B55B:8AEF)FF)NN)__name__ __module__ __qualname____doc__r str__annotations__r rr2rintrfloatrrrrboolpropertyrrrrrr9rrrr r sp  1L#000#'L(4.'''1Is1K-GT$Z%E50L$' 3X WS#X%6   X %):> IIItCy!I67 I  I IIIIIIrr )loggingtypingrrrrr+langchain_core.callbacksr#langchain_core.language_models.llmsrlangchain_community.llms.utilsr getLoggerr:r)r rrrrJs////////////======333333>>>>>>  8 $ $tttttctttttr