{h{%ddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZddlmZddlmZmZddlmZddlmZed Ze d e ZGd d e ZGddej>Z dedefde!fdZ"ejFdedefde!fdZ$dedefde%fdZ&GddZ'y)N)AnyCallableOptionalTypeVar TypedDictLiteralUnion) ParamSpec NotRequired)ChatCompletion)ChatCompletionMessageParam) BaseModel validate_call)OpenAI) openai_schemaPT_Retval)boundc>eZdZUeeed<eeeee fed<y)OpenAIChatKwargsmessages functionsN) __name__ __module__ __qualname__listr __annotations__r dictstrrR/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/instructor/distil.pyrrs&-..4S#X/00r!rceZdZdZdZy)FinetuneFormatrrawN)rrrMESSAGESRAWr r!r"r$r$#s H Cr!r$fn.returnctj|}d|j|}tj|}|rd|d}nd}|d|S)aJ Get the function signature as a string. :Example: >>> def my_function(a: int, b: int) -> int: >>> return a + b >>> >>> get_signature_from_fn(my_function) "def my_function(a: int, b: int) -> int" :param fn: Function to get the signature for. :return: Function signature as a string. def """  """ )inspect signaturergetdoc)r(siglines docstringformatted_docstrings r"get_signature_from_fnr7(sa   B C2;;-u %Er"I %i[6 WB*+ ,,r!funcc,tj|}dj|dj}tj|}|rd|d}nd}tj |}|j d|jd}|d|d|S)z@ Format a function as a string with docstring and body.  rr,r-r.r+r/)r0getsourcelinesjoinstripr2 getsourcereplacer)r8 source_lines definitionr5r6bodys r"format_functionrCAs ))$/L,q/*002Jt$I %i[6    T "D <<$t}}o. 3D\/04& 99r!ctj|j}|tjjk7sJdtj |xrt |tS)z Check if the return type of a function is a pydantic BaseModel or an instance of it. :param func: Function to check. :return: True if the return type is a pydantic BaseModel or an instance of it. z9Must have a return type hint that is a pydantic BaseModel)r0r1return_annotation Signatureemptyisclass issubclassr)r8 return_types r"%is_return_type_base_model_or_instancerKUs[##D);;K '++11 1C 1 ??; ' NJ{I,NNr!ceZdZdddejdddfdeedeedeeejdede d e d ee d dfd Z dd ddddedeededdedeed eeeeeeffeeeefgeeeeefffff dZedejfdedefdeedfdeeefdedeeded dfdZdededefdeedfdeeefdeed ef dZy) InstructionsNFnameid log_handlersfinetune_formatindentinclude_code_body openai_clientr)c||_|xsttj|_ttj|_||_||_||_|xs t|_ tj|j|_ |xsgD]}|jj|y)a Instructions for distillation and dispatch. :param name: Name of the instructions. :param id: ID of the instructions. :param log_handlers: List of log handlers to use. :param finetune_format: Format to use for finetuning. :param indent: Indentation to use for finetuning. :param include_code_body: Whether to include the code body in the finetuning. N)rOruuiduuid4rP unique_idrRrSrTrclientlogging getLoggerlogger addHandler) selfrOrPrQrRrSrTrUhandlers r"__init__zInstructions.__init__ds( )DJJL)TZZ\*. !2#/vx '' 2 #)r ,G KK " "7 + ,r!distilz gpt-3.5-turbo)rOmodemodelfine_tune_formatargsrc)rbdispatchrdrecddh}|vs Jd| jdtttfdttttt ffffd }t |dk(rt|dr ||dS|S) ag Decorator to track the function call and response, supports distillation and dispatch modes. If used without arguments, it must be used as a decorator. :Example: >>> @distil >>> def my_function() -> MyModel: >>> return MyModel() >>> >>> @distil(name="my_function") >>> def my_function() -> MyModel: >>> return MyModel() :param fn: Function to track. :param name: Name of the function to track. Defaults to the function name. :param mode: Mode to use for distillation. Defaults to "distil". rbrgz Must be in r(r)cdd}tsJ|tjjt j dt jdt jdtf fd }t j dt jdt jdtf fd }dk(r|S|S) NzReturn type hint for z# must subclass `pydantic.BaseModel'rfkwargsr)cjrn j||}jjjj di|dS)N)rOr(rfrj base_model)rdresponse_modelr ) openai_kwargsrrZchat completionscreate)rfrjrnr(rdrOreturn_base_modelr_s r" _dispatchz._wrap_distil.._dispatchsi $ 2 2!%2;;!0 !3! ;t{{''33::##4r!cF|i|}j||||S)N)rOrR)track)rfrjresprer(rOr_s r"_distilz:Instructions.distil.._wrap_distil.._distils<4*6* $4  r!rg) rKr0r1rE functoolswrapsrrfrjr r) r(msgrsrwrrrercrdrOr_s ` @r" _wrap_distilz)Instructions.distil.._wrap_distils*"-PQC8< Ac A< ' 1 1" 5 G G  __R   188   ! __R  qvv  h ! !% 29 ? ?r!r)rRrrrr r lencallable)r_rOrcrdrerf allowed_modesr{s````` r"rbzInstructions.distils<":. }$C M?&CC$  ##33 # @H%# @ ax788 9# @# @J t9>htAw/Q( (r!r(.rjrvc |r|n |j}t|}|tjk(rt |j}|j |||||} | dj d|j|j|jdd|g| d<|jjtj| |tjk(ret|t||||j!|j#} |jjtj| yy) a Track the function call and response in a log file, later used for finetuning. :param fn: Function to track. :param args: Arguments passed to the function. :param kwargs: Keyword arguments passed to the function. :param resp: Response returned by the function. :param name: Name of the function to track. Defaults to the function name. :param finetune_format: Format to use for finetuning. Defaults to "raw". r assistant)rS)rO arguments)role function_callr)fn_namefn_reprrfrjrvschemaN)rtyper$r&rrnappendmodel_dump_jsonrSr]infojsondumpsr'rrC model_dumpmodel_json_schema) r_r(rfrjrvrOrRrlopenai_function_callrn function_bodys r"ruzInstructions.tracks(t$Z n55 5#0#<#J#J  ..tRvzRM * % , ,' * 3 3%)%9%9%9%M& +?)?M+ & KK  TZZ 6 7 n00 0 '+__&!335 M KK  TZZ 6 7 1r!rlcD|jr t|}n t|}djt t |}djd|j Dxsd}djtd||g} ddd|ddd|d | d dgi} | S) Nz, c3VK|]!\}}|dtj|#yw)=N)rr).0kvs r" z-Instructions.openai_kwargs..s'HA1TZZ]O,Hs')rsystemz'Predict the results of this function: )rcontentuserzReturn `(z)`)rTrCr7r<mapritemsfilter) r_rOr(rfrjrlfunc_defstr_args str_kwargs call_argsrs r"rnzInstructions.openai_kwargss  ! !&r*H,R0H99Sd^, IIHH H PD IIfTHj+ABC  $!J8*U #!)$q 2>  + r!)rrrr$r&rrrr[Handlerintboolrrarrr rrrr rbrtuplerrrurrrnr r!r"rMrMcs!# 8<*8*A*A"'*.,sm, SM,tGOO45 , ( ,  , , ', ,F#.6$59 LLsmL*+ L  L #>2 L E(N2334(1h;'((1eHnrs 5<V-3cN :Y /1y1 TYY -hsCx0-S-2 :(38,:::& Oc0B Ot Or!