U h$@sdZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZmZdd lmZdd lmZdd lmZeeZGd d d eZdS)z'Wrapper around EdenAI's Generation API.N)AnyDictListLiteralOptional) ClientSession)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM)Fieldroot_validator)get_from_dict_or_envpre_init) get_fields)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<GdddZeeedddZeddeeefeeefdd d!Zeed"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_sequencesc@seZdZdZdS)z EdenAI.ConfigZforbidN)__name__ __module__ __qualname__extrar*r*C/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/edenai.pyConfigHsr,)valuesreturncCst|dd|d<|S)z,Validate that api key exists in environment.rZEDENAI_API_KEY)r )clsr-r*r*r+validate_environmentKs zEdenAI.validate_environmentT)precCsddt|D}|di}t|D]H}||kr*||krNtd|dt|d|d||||<q*||d<|S)z>Build extra kwargs from additional params that were passed in.cSsh|] }|jqSr*)alias).0fieldr*r*r+ Vsz%EdenAI.build_extra..r$zFound z supplied twice.zJ was transferred to model_kwargs. Please confirm that z is what you intended.)rr-getlist ValueErrorloggerwarningpop)r/r-Zall_required_field_namesr) field_namer*r*r+ build_extraSs  zEdenAI.build_extra)r.cCsdS)zReturn type of model.Zedenair*)selfr*r*r+ _llm_typeeszEdenAI._llm_type)outputr.cCs2|jdkr||jdS||jdddSdS)NrZgenerated_textitemsrr)rr)r>r@r*r*r+_format_outputjs zEdenAI._format_outputcCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityrDrCr*r*r+get_user_agentps zEdenAI.get_user_agent)promptstop run_managerkwargsr.cKsd}|jdk r |dk r tdn|jdk r2|j}n|}|jd|jd|j}d|j|d}|j||j|j |j d|j |ddi}d d | D}|j dk r|j|j i|d <t|d } | j||d } | jdkrtd| jnB| jdkr td| jn$| jdkr.td| jd| j| } | |j} | ddkrj| did} t| || }|dk rt||}|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-AgentZ providersrr rr! num_imagesrcSsi|]\}}|dk r||qSNr*r3kvr*r*r+ sz EdenAI._call..settings)headers)urldataEdenAI Server: Error $EdenAI received an invalid payload: 3EdenAI returned an unexpected response with status : statusfailerrormessage)r%r8rrrrrErr rr!r#rArrpost status_code Exceptionrjsonr6rBr)r>rFrGrHrIstopsrXrWpayloadrequestresponserYprovider_responseerr_msgr@r*r*r+_callvsZ            z EdenAI._callc sd}|jdk r |dk r tdn|jdk r2|j}n|}|jd|jd|j}d|j|d}|j||j|j |j d|j |ddi}d d | D}|j dk r|j|j i|d <t4IdH} | j|||d 4IdH} | jd krtd| jnB| jdkr td| jn$| jdkrDtd| jd| j| IdH} | |j} | ddkr| did} t| || }|dk rt||}|W5QIdHRW5QIdHRSQIdHRXW5QIdHRXdS)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. NrJrKrLrMrOrPrcSsi|]\}}|dk r||qSrQr*rRr*r*r+rUsz!EdenAI._acall..rV)rhrWrZr[r\r]r^r_r`rarbrcrd)r%r8rrrrrErr rr!r#rArrrerargrrhr6rBr)r>rFrGrHrIrirXrWrjsessionrlZ response_jsonrmrnr@r*r*r+_acalls^            z EdenAI._acall)NN)NN)(r&r'r(__doc__rstr__annotations__rrrrrrr rfloatr intr!dictr#rrr$r%rr,rr0r r=propertyr?rB staticmethodrEr rorrqr*r*r*r+rsR  "  L r)rrloggingtypingrrrrrZaiohttprZlangchain_core.callbacksrr Z#langchain_core.language_models.llmsr Zlangchain_core.pydantic_v1r r Zlangchain_core.utilsr rZlangchain_core.utils.pydanticrZlangchain_community.llms.utilsrZ&langchain_community.utilities.requestsr getLoggerr&r9rr*r*r*r+s