o Zh$@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< eddZededefddZedd edeeefdefd!d"Z e!defd#d$Z"d%edefd&d'Z#e$defd(d)Z%  d2d*ed+eeed,ee&d-edef d.d/Z'  d2d*ed+eeed,ee(d-edef d0d1Z)dS)3EdenAIaEdenAI 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)extravaluesreturncCst|dd|d<|S)z,Validate that api key exists in environment.rZEDENAI_API_KEY)r )clsr(r+V/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/edenai.pyvalidate_environmentLszEdenAI.validate_environmentbefore)modecCsddt|D}|di}t|D]$}||vr9||vr'td|dt|d|d||||<q||d<|S)z>Build extra kwargs from additional params that were passed in.cSsh|]}|jqSr+)alias).0fieldr+r+r, Xsz%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_extracCsdS)zReturn type of model.Zedenair+)selfr+r+r, _llm_typegszEdenAI._llm_typeoutputcCs.|jdkr ||jdS||jdddS)NrZgenerated_textitemsrr)rr)r<r>r+r+r,_format_outputls zEdenAI._format_outputcCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityrBrAr+r+r,get_user_agentrs  zEdenAI.get_user_agentpromptstop run_managerkwargscKsd}|jdur|durtd|jdur|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 } | jdkrvtd| j| jdkrtd| j| jdkrtd| jd| j| } | |j} | ddkr| 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_imagesrcSi|] \}}|dur||qSNr+r1kvr+r+r, z EdenAI._call..settings)headers)urldataEdenAI Server: Error $EdenAI received an invalid payload: 3EdenAI returned an unexpected response with status : statusfailerrormessage)r&r6rrrrrCrr!r r"r$r?rrpost status_code Exceptionrjsonr4r@r)r<rDrErFrGstopsrYrXpayloadrequestresponserZprovider_responseerr_msgr>r+r+r,_callxsd          z EdenAI._callc s d}|jdur|durtd|jdur|j}n|}|jd|jd|j}d|j|d}|j||j|j |j d|j |ddi}d d | D}|j dur^|j|j i|d <t4IdH} | j|||d 4IdHp} | jd krtd| j| jdkrtd| j| jdkrtd| jd| j| IdH} | |j} | ddkr| did} t| || }|durt||}|WdIdHWdIdHS1IdHswYWdIdHdS1IdHswYdS)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. NrHrIrJrKrMrOrcSrPrQr+rRr+r+r,rUrVz!EdenAI._acall..rW)rirXr[r\r]r^r_r`rarbrcrdre)r&r6rrrrrCrr!r r"r$r?rrrfrbrhrrir4r@r)r<rDrErFrGrjrYrXrksessionrmZ response_jsonrnror>r+r+r,_acallsr          .z EdenAI._acall)NN)*__name__ __module__ __qualname____doc__rstr__annotations__rrrrrrrr floatr!intr"dictr$rrr%r&rrZ model_configr r-r classmethodr;propertyr=r@ staticmethodrCr rprrrr+r+r+r,rsx    L r) rvloggingtypingrrrrrZaiohttprZlangchain_core.callbacksrr Z#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zlangchain_core.utils.pydanticr ZpydanticrrrZlangchain_community.llms.utilsrZ&langchain_community.utilities.requestsr getLoggerrsr7rr+r+r+r,s