7|hrdZddlZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZmZmZddlmZej&eZGdd e Zy) 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"y) 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}t|dstjd|S#t$r tdwxYw)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 ]/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/llms/konko.pyvalidate_environmentzKonko.validate_environmentCsR  u12 MM5    >  s)>promptstopkwargsc |rt|dk(r|dn|}i|j||d|}|jDcic] \}}| || c}}Scc}}w)Nr)r)r*)lendefault_paramsitems)selfr)r*r+ stop_to_usepayloadkvs r'construct_payloadzKonko.construct_payloadWsm "&#d)q.d1gd # !!# #  #  ")BAAM1BBBs A Acy)zReturn type of model.r!r1s r' _llm_typezKonko._llm_typefscddlm}d|S)Nr __version__z langchain/)langchain_communityr>r=s r'get_user_agentzKonko.get_user_agentks3K=))r;c|j|j|j|j|j|j dS)NrrrrrrrBr9s r'r/zKonko.default_paramsqs<ZZ++ZZZZ//"&"9"9   r; run_managerc Nddl}|j||fi|} tr|jjdi|}n|j jdi|}tr|jdj}|S|ddd}|S#t $r tdwxYw)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`.choicestextr8) r!r6r completionscreate CompletionAttributeError ValueErrorrFrG) r1r)r*rCr+r!r3responseoutputs r'_callz Konko._call|s ($((@@ ~35,,33>g>25++22=W= >%%a(--F i(+F3F  6  s ABB$cKddl}|j||fi|} tr5|j}|jj di|d{}n$|j jdi|d{}tr|jdj} | S|ddd} | S7[78#t$r tdwxYww)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. rNrErFrGr8) r!r6r AsyncKonkorHrIrJacreaterKrLrFrG) r1r)r*rCr+r!r3clientrMrNs r'_acallz Konko._acalls ($((@@ ~))+!:!3!3!:!:!EW!EE!9!1!1!9!9!DG!DD >%%a(--F i(+F3F !FD 6  s@C 9B5B1$B5:B3;B5?2C 1B53B55C  C )N)NN)#__name__ __module__ __qualname____doc__rstr__annotations__r rrfloatrrintrrrr model_configr classmethodrrr(rr6propertyr: staticmethodr@r/rrOrrTr8r;r'rrs:Hc9! J$(K%'!E8E?!  E8C= !%J $3*..#Hhsm"L(#$sCx.S$*%) C CtCy! C C c3h C3*C**  S#X  %):> %%tCy!%67 %  % %T%)?C &&tCy!&;< &  & &r;r)rXloggingr$typingrrrrlangchain_core.callbacksrr#langchain_core.language_models.llmsr pydanticr r r langchain_community.utils.openair getLoggerrUloggerrr8r;r'risG/,,4;;9   8 $vCvr;