NgrdZddlZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZmZmZddlmZejeZGdd e ZdS) z)Wrapper around Konko AI's Completion API.N)AnyDictListOptional)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) ConfigDict SecretStrmodel_validator) is_openai_v1c eZdZUdZdZeed< eed< eed< dZe e ed< dZ e e ed< dZ e e ed < dZe e ed < dZe e ed < dZe e ed < ed ZededeeefdefdZ ddede eededeeeffdZedefdZedefdZedeeeffdZ ddede eede ededef dZ ddede eede e!dedef dZ"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_url konko_api_keymodelN temperaturetop_ptop_k max_tokensrepetition_penaltylogprobsforbid)extrabefore)modevaluesreturnc ddl}n#t$rtdwxYwt|dstjd|S)z3Validate that python package exists in environment.rNzRCould not import konko python package. Please install it with `pip install konko`._is_legacy_openaizYou are using an older version of the 'konko' package. Please consider upgrading to access new featuresincluding the completion endpoint.)konko ImportErrorhasattrwarningswarn)clsrr!s Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/konko.pyvalidate_environmentzKonko.validate_environmentCsy  LLLL   >   u122  M5     s!promptstopkwargsc |rt|dkr|dn|}i|j||d|}d|DS)Nr)r)r*ci|] \}}||| SN).0kvs r' z+Konko.construct_payload..dsBBBAAM1MMM)lendefault_paramsitems)selfr)r*r+ stop_to_usepayloads r'construct_payloadzKonko.construct_payloadWst "&B#d))q..d1ggd # !# # # #  #  CBBBBBr5cdS)zReturn type of model.r!r0r9s r' _llm_typezKonko._llm_typefs wr5cddlm}d|S)Nr __version__z langchain/)langchain_communityrBrAs r'get_user_agentzKonko.get_user_agentks"333333)K)))r5cP|j|j|j|j|j|jdS)NrrrrrrrFr>s r'r7zKonko.default_paramsqs3Z+ZZ/"&"9    r5 run_managerc :ddl}|j||fi|} tr|jjdi|}n|jjdi|}n#t $rtdwxYwtr|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`.choicestextr0) r!r<r completionscreate CompletionAttributeError ValueErrorrJrK) r9r)r*rGr+r!r;responseoutputs r'_callz Konko._call|s  ($(@@@@ ~~ >35,3>>g>>25+2==W==   6   >> 4%a(-FFi(+F3F s 3A A#c~Kddl}|j||fi|} tr-|}|jjdi|d{V}n|jjdi|d{V}n#t$rtdwxYwtr|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. rNrIrJrKr0) r!r<r AsyncKonkorLrMrNacreaterOrPrJrK) r9r)r*rGr+r!r;clientrQrRs r'_acallz Konko._acalls  ($(@@@@ ~~ E))++!:!3!:!E!EW!E!EEEEEEE!9!1!9!D!DG!D!DDDDDDD   6   >> 4%a(-FFi(+F3F s AA++Br/)NN)#__name__ __module__ __qualname____doc__rstr__annotations__r rrfloatrrintrrrr model_configr classmethodrrr(rr<propertyr? staticmethodrDr7rrSrrXr0r5r'rrs:Hc999! JJJ$(K%'''!E8E?!!!  E8C= !%J $$$3*....#Hhsm""":L_(###$sCx.S[$#*%) C C CtCy! C C c3h C C C C3X*C***\*  S#X   X %):> %%%tCy!%67 %  % %%%%T%)?C &&&tCy!&;< &  & &&&&&&r5r)r\loggingr$typingrrrrlangchain_core.callbacksrr#langchain_core.language_models.llmsr pydanticr r r langchain_community.utils.openair getLoggerrYloggerrr0r5r'rms//,,,,,,,,,,,,433333;;;;;;;;;;999999  8 $ $vvvvvCvvvvvr5