U hY@sdZddlZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZeeZGdd d e ZdS) z)Wrapper around Konko AI's Completion API.N)AnyDictListOptional)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) SecretStrroot_validator) is_openai_v1c@sreZdZUdZdZeed<eed<eed<dZe e ed<dZ e e ed<dZ e e ed <dZe e ed <dZe e ed <dZe e ed <Gd ddZeddeeefeeefdddZd#ee eeeeeefdddZeedddZeedddZeeeefdddZd$ee eee eeeddd Zd%ee eee eeedd!d"Z dS)&KonkozKonko AI models. To use, you'll need an API key. This can be passed in as init param ``konko_api_key`` or set as environment variable ``KONKO_API_KEY``. Konko AI API reference: https://docs.konko.ai/reference/ z#https://api.konko.ai/v1/completionsbase_urlZ konko_api_keymodelN temperaturetop_ptop_k max_tokensrepetition_penaltylogprobsc@seZdZdZdS)z Konko.ConfigZforbidN)__name__ __module__ __qualname__extrarrB/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/konko.pyConfig?srT)pre)valuesreturncCsBz ddl}Wntk r(tdYnXt|ds>td|S)z3Validate that python package exists in environment.rNzRCould not import konko python package. Please install it with `pip install konko`.Z_is_legacy_openaizYou are using an older version of the 'konko' package. Please consider upgrading to access new featuresincluding the completion endpoint.)konko ImportErrorhasattrwarningswarn)clsrrrrrvalidate_environmentBs   zKonko.validate_environment)promptstopkwargsrcKs@|rt|dkr|dn|}|j||d|}dd|DS)Nr)r&r'cSsi|]\}}|dk r||qS)Nr).0kvrrr bsz+Konko.construct_payload..)lendefault_paramsitems)selfr&r'r(Z stop_to_usepayloadrrrconstruct_payloadUszKonko.construct_payload)rcCsdS)zReturn type of model.rrr1rrr _llm_typedszKonko._llm_typecCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityr7r6rrrget_user_agentis zKonko.get_user_agentcCs|j|j|j|j|j|jdS)Nrrrrrrr9r4rrrr/oszKonko.default_params)r&r' run_managerr(rc Ksddl}|j||f|}z(tr0|jjf|}n|jjf|}Wntk r\tdYnXtrr|jdj }n|ddd}|S)zCall out to Konko's text generation endpoint. Args: prompt: The prompt to pass into the model. Returns: The string generated by the model.. rN`konko` has no `Completion` attribute, this is likely due to an old version of the konko package. Try upgrading it with `pip install --upgrade konko`.choicestext) rr3r completionscreate CompletionAttributeError ValueErrorr<r=) r1r&r'r:r(rr2responseoutputrrr_callzs z Konko._callc sddl}|j||f|}z|}|jjf|IdH}n|jjf|IdH}Wntk rpt dYnXtr|j dj } n|ddd} | S)zAsynchronously call out to Konko's text generation endpoint. Args: prompt: The prompt to pass into the model. Returns: The string generated by the model. rNr;r<r=) rr3r Z AsyncKonkor>r?r@ZacreaterArBr<r=) r1r&r'r:r(rr2clientrCrDrrr_acalls z Konko._acall)N)NN)NN)!rrr__doc__r str__annotations__r rrfloatrrintrrrrr rrr%rr3propertyr5 staticmethodr8r/rrErrGrrrrr sX  "     * r )rHloggingr"typingrrrrZlangchain_core.callbacksrrZ#langchain_core.language_models.llmsrZlangchain_core.pydantic_v1r r Z langchain_community.utils.openair getLoggerrloggerr rrrrs