a bg$@sdZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZdd lmZmZmZdd lmZdd lmZeeZGd d d eZdS)z'Wrapper around EdenAI's Generation API.N)AnyDictListLiteralOptional) ClientSession)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM)get_from_dict_or_envpre_init) get_fields) ConfigDictFieldmodel_validator)enforce_stop_tokens)Requestsc@seZdZUdZdZeed<dZeeed<dZ e ded<d Z e d ed <eed <dZ eeed <e dd ddZeeed<e dd dZeeed<dZee ded<e edZeeefed<e edZeeefed<dZeeeed<eddZeeedddZeddeeeefedd d!Z e!ed"d#d$Z"eed%d&d'Z#e$ed"d(d)Z%d/eeeeee&eed*d+d,Z'd0eeeeee(eed*d-d.Z)dS)1EdenAIaEdenAI models. To use, you should have the environment variable ``EDENAI_API_KEY`` set with your API token. You can find your token here: https://app.edenai.run/admin/account/settings `feature` and `subfeature` are required, but any other model parameters can also be passed in with the format params={model_param: value, ...} for api reference check edenai documentation: http://docs.edenai.co. zhttps://api.edenai.run/v2base_urlNedenai_api_keytext)rimagefeatureZ generation subfeatureprovidermodelr)defaultgele temperature)rr max_tokens)Z256x256Z512x512Z 1024x1024 resolution)default_factoryparams model_kwargsstop_sequencesZforbid)extra)valuesreturncCst|dd|d<|S)z,Validate that api key exists in environment.rZEDENAI_API_KEY)r )clsr(r+m/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/edenai.pyvalidate_environmentLszEdenAI.validate_environmentbefore)modecCsddt|D}|di}t|D]H}||vr*||vrNtd|dt|d|d||||<q*||d<|S)z>Build extra kwargs from additional params that were passed in.cSsh|] }|jqSr+)alias).0fieldr+r+r, Xz%EdenAI.build_extra..r%zFound z supplied twice.zJ was transferred to model_kwargs. Please confirm that z is what you intended.)r r(getlist ValueErrorloggerwarningpop)r*r(Zall_required_field_namesr' field_namer+r+r, build_extraTs  zEdenAI.build_extra)r)cCsdS)zReturn type of model.Zedenair+)selfr+r+r, _llm_typegszEdenAI._llm_type)outputr)cCs2|jdkr||jdS||jdddSdS)NrZgenerated_textitemsrr)rr)r=r?r+r+r,_format_outputls zEdenAI._format_outputcCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityrCrBr+r+r,get_user_agentrs zEdenAI.get_user_agent)promptstop run_managerkwargsr)cKsd}|jdur |dur tdn|jdur2|j}n|}|jd|jd|j}d|j|d}|j||j|j |j d|j |ddi}d d | D}|j dur|j|j i|d <t|d } | j||d } | jdkrtd| jnB| jdkrtd| jn$| jdkr2td| jd| j| } | |j} | ddkrn| did} t| || }|durt||}|S)zCall out to EdenAI's text generation endpoint. Args: prompt: The prompt to pass into the model. Returns: json formatted str response. N:stop sequences found in both the input and default params./Bearer  Authorizationz User-Agent providersrr!r r" num_imagesrcSsi|]\}}|dur||qSNr+r1kvr+r+r, r4z EdenAI._call..settings)headers)urldataEdenAI Server: Error $EdenAI received an invalid payload: 3EdenAI returned an unexpected response with status : statusfailerrormessage)r&r7rrrrrDrr!r r"r$r@rrpost status_code Exceptionrjsonr5rAr)r=rErFrGrHstopsrXrWpayloadrequestresponserYprovider_responseerr_msgr?r+r+r,_callxsd            z EdenAI._callc s d}|jdur |dur tdn|jdur2|j}n|}|jd|jd|j}d|j|d}|j||j|j |j d|j |ddi}d d | D}|j dur|j|j i|d <t4IdH8} | j|||d 4IdH} | jd krtd| jnB| jdkr&td| jn$| jdkrJtd| jd| j| IdH} | |j} | ddkr| did} t| || }|durt||}|WdIdHWdIdHS1IdHs0YWdIdHq1IdHs0YdS)aICall EdenAi model to get predictions based on the prompt. Args: prompt: The prompt to pass into the model. stop: A list of stop words (optional). run_manager: A callback manager for async interaction with LLMs. Returns: The string generated by the model. NrIrJrKrLrNrPrcSsi|]\}}|dur||qSrQr+rRr+r+r,rUr4z!EdenAI._acall..rV)rhrWrZr[r\r]r^r_r`rarbrcrd)r&r7rrrrrDrr!r r"r$r@rrrerargrrhr5rAr)r=rErFrGrHrirXrWrjsessionrlZ response_jsonrmrnr?r+r+r,_acallsh            z EdenAI._acall)NN)NN)*__name__ __module__ __qualname____doc__rstr__annotations__rrrrrrrr floatr!intr"dictr$rrr%r&rrZ model_configr r-r classmethodr<propertyr>rA staticmethodrDr rorrqr+r+r+r,rsX    L r) ruloggingtypingrrrrrZaiohttprZlangchain_core.callbacksrr Z#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zlangchain_core.utils.pydanticr ZpydanticrrrZlangchain_community.llms.utilsrZ&langchain_community.utilities.requestsr getLoggerrrr8rr+r+r+r,s