o Zhr@sdZddlZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZmZmZddlmZeeZGdd d e ZdS) z)Wrapper around Konko AI's Completion API.N)AnyDictListOptional)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) ConfigDict SecretStrmodel_validator) is_openai_v1c @seZdZUdZdZeed< eed< eed< dZe e ed< dZ e e ed< dZ e e ed < dZe e ed < dZe e ed < dZe e ed < ed dZeddedeeefdefddZ d%dede eededeeeffddZedefddZedefddZedeeeffddZ  d&dede eed e ededef d!d"Z  d&dede eed e e!dedef d#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_penaltylogprobsZforbid)extrabefore)modevaluesreturncCs<zddl}Wn tytdwt|dstd|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)clsrrr!U/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/konko.pyvalidate_environmentCs   zKonko.validate_environmentpromptstopkwargscKsF|r t|dkr |dn|}i|j||d|}dd|DS)Nr)r$r%cSsi|] \}}|dur||qSNr!).0kvr!r!r" dsz+Konko.construct_payload..)lendefault_paramsitems)selfr$r%r&Z stop_to_usepayloadr!r!r"construct_payloadWszKonko.construct_payloadcCsdS)zReturn type of model.rr!r0r!r!r" _llm_typefszKonko._llm_typecCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityr6r5r!r!r"get_user_agentks  zKonko.get_user_agentcCs|j|j|j|j|j|jdS)Nrrrrrrr8r3r!r!r"r.qszKonko.default_params run_managerc Ksddl}|j||fi|}ztr|jjdi|}n |jjdi|}Wn ty1tdwtr=|jdj }|S|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`.choicestextr!) rr2r completionscreate CompletionAttributeError ValueErrorr;r<) r0r$r%r9r&rr1responseoutputr!r!r"_call|s"  z Konko._callc sddl}|j||fi|}z"tr$|}|jjdi|IdH}n |jjdi|IdH}Wn ty<t dwtrH|j dj } | S|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<r!) rr2r Z AsyncKonkor=r>r?Zacreater@rAr;r<) r0r$r%r9r&rr1clientrBrCr!r!r"_acalls&  z Konko._acallr()NN)#__name__ __module__ __qualname____doc__rstr__annotations__r rrfloatrrintrrrr Z model_configr classmethodrrr#rr2propertyr4 staticmethodr7r.rrDrrFr!r!r!r"r s        * r )rJloggingrtypingrrrrZlangchain_core.callbacksrrZ#langchain_core.language_models.llmsrZpydanticr r r Z langchain_community.utils.openair getLoggerrGloggerr r!r!r!r"s