a bg@sddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZmZddlmZGdd d e Zd S) )partial)AnyDictListMappingOptionalSet)CallbackManagerForLLMRun)LLM)pre_init) ConfigDictField)enforce_stop_tokensc@sXeZdZUdZeed<edddZeeed<edddZ e ed<edd dZ e ed <ed d dZ e ed <ed d dZ eed <ed ddZeed<ed ddZeed<ed ddZeed<ed ddZeed<edddZee ed<dZee ed<dZeeed<dZeeed<dZee ed<d Zeeed<gZeeeed<dZee ed<d Zeeed!<ed"d#dZe ed#<d Zeed$<d Z eed%<ed&d'dZ!eeed'<dZ"e#ed(<e$d)d*Z%e&e'ed+d,d-Z(e)ee#fd+d.d/Z*e+e)e)d0d1d2Z,e-e.ee#fd+d3d4Z/e-ed+d5d6Z0d:eeeeee1e#ed7d8d9Z2dS);GPT4AllaGPT4All language models. To use, you should have the ``gpt4all`` python package installed, the pre-trained model file, and the model's config information. Example: .. code-block:: python from langchain_community.llms import GPT4All model = GPT4All(model="./models/gpt4all-model.bin", n_threads=8) # Simplest invocation response = model.invoke("Once upon a time, ") modelNbackend)alias max_tokensn_partsrseedFf16_kv logits_all vocab_only use_mlock embedding n_threads n_predictgffffff?tempg?top_p(top_kechostop@ repeat_last_ngzG?repeat_penaltyn_batch streamingallow_downloadcpudeviceclientZforbid)extra)returncCshdS)N> r(r,r r+r$r)r!r"rr3r3r3n/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/gpt4all.py_model_param_namesgszGPT4All._model_param_namesc Cs*|j|j|j|j|j|j|j|j|jd S)N rr r$r"r!r+r)r(r,r6selfr3r3r4_default_paramsuszGPT4All._default_params)valuesr2cCszddlm}Wnty*tdYn0|d}|d\}}}||7}|||pVd|d|d|d d |d <|d dur|d j|d z|d j|d<Wn"ty|d jj|d<Yn0|S) z;Validate that the python package exists in the environment.r)rzVCould not import gpt4all python package. Please install it with `pip install gpt4all`.r/Nrr-r/) model_path model_typer-r/r0r)gpt4allr ImportError rpartitionrZset_thread_countr=AttributeError)clsr:Z GPT4AllModel full_pathr< delimiterZ model_namer3r3r4validate_environments.     zGPT4All.validate_environmentcs*djifddjDS)zGet the identifying parameters.rcs"i|]\}}|vr||qSr3)r5).0kvr7r3r4 sz/GPT4All._identifying_params..)rr9__dict__itemsr7r3r7r4_identifying_paramss zGPT4All._identifying_paramscCsdS)zReturn the type of llm.r>r3r7r3r3r4 _llm_typeszGPT4All._llm_type)promptr& run_managerkwargsr2c Kspd}|rt|j|jd}d}i||}|jj|fi|D]}|rP||||7}q@|durlt||}|S)aCall out to GPT4All's generate method. Args: prompt: The prompt to pass into the model. stop: A list of strings to stop generation when encountered. Returns: The string generated by the model. Example: .. code-block:: python prompt = "Once upon a time, " response = model.invoke(prompt, n_predict=55) N)verbose)rZon_llm_new_tokenrQr9r0generater) r8rNr&rOrPZ text_callbacktextparamstokenr3r3r4_calls  z GPT4All._call)NN)3__name__ __module__ __qualname____doc__str__annotations__r rrrintrrrboolrrrrrr r!floatr"r$r%r&rr(r)r+r,r-r/r0rr Z model_config staticmethodrr5rr9r rEpropertyrrLrMr rWr3r3r3r4r s\      !  rN) functoolsrtypingrrrrrrZlangchain_core.callbacksr Z#langchain_core.language_models.llmsr Zlangchain_core.utilsr Zpydanticr r Zlangchain_community.llms.utilsrrr3r3r3r4s