a bgr@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@sjeZdZUdZdZeed<eed<eed<dZe e ed<dZ e e ed<dZ e e ed <dZe e ed <dZe e ed <dZe e ed <ed dZeddeeeefedddZd#ee eeeeeefdddZeedddZeedddZeeeefdddZd$ee eee eeeddd Z d%ee eee e!eedd!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)mode)valuesreturncCs@z ddl}Wnty&tdYn0t|dsdz+Konko.construct_payload..)lendefault_paramsitems)selfr$r%r&Z stop_to_usepayloadr!r!r"construct_payloadWszKonko.construct_payload)rcCsdS)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)r$r% run_managerr&rc Ksddl}|j||fi|}z0tr8|jjfi|}n|jjfi|}WntyftdYn0tr||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) rr2r completionscreate CompletionAttributeError ValueErrorr;r<) r0r$r%r9r&rr1responseoutputr!r!r"_call|s  z Konko._callc sddl}|j||fi|}zDtrF|}|jjfi|IdH}n|jjfi|IdH}Wntyzt dYn0tr|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<) rr2r Z AsyncKonkor=r>r?Zacreater@rAr;r<) r0r$r%r9r&rr1clientrBrCr!r!r"_acalls  z Konko._acall)N)NN)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