6i%ddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZddlmZddlmZmZddlmZdd lmZed Ze d e ZGd de 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 dictstrr\/root/.openclaw/workspace/visionaryfx/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_fnr8(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,r0)r1getsourcelinesjoinstripr3 getsourcereplacer)r9 source_lines definitionr6r7bodys r#format_functionrDAs ))$/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)r1r2return_annotation Signatureemptyisclass issubclassr)r9 return_types r#%is_return_type_base_model_or_instancerLUs[##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)rPr uuiduuid4rQ unique_idrSrTrUrclientlogging getLoggerlogger addHandler) selfrPrQrRrSrTrUrVhandlers r#__init__zInstructions.__init__ds( )DJJL)TZZ\*. !2#/vx '' 2 #)r ,G KK " "7 + ,r"distilz gpt-3.5-turbo)rPmodemodelfine_tune_formatargsrd)rcdispatchrerfcddh}|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". rcrhz 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'rgkwargsr*cjrn j||}jjjj di|dS)N)rPr)rgrk base_model)reresponse_modelr!) openai_kwargsrr[chat completionscreate)rgrkror)rerPreturn_base_modelr`s r# _dispatchz._wrap_distil.._dispatchsi $ 2 2!%2;;!0 !3! ;t{{''33::##4r"cF|i|}j||||S)N)rPrS)track)rgrkresprfr)rPr`s r#_distilz:Instructions.distil.._wrap_distil.._distils<4*6* $4  r"rh) rLr1r2rF functoolswrapsrrgrkr r) r)msgrtrxrsrfrdrerPr`s ` @r# _wrap_distilz)Instructions.distil.._wrap_distils*"-PQC8< Ac A< ' 1 1" 5 G G  __R   188   ! __R  qvv  h ! !% 29 ? ?r"rr)rSrrrr r lencallable)r`rPrdrerfrg allowed_modesr|s````` r#rczInstructions.distils<":. }$C M?&CC$  ##33 # @H%# @ ax788 9# @# @J t9>htAw/Q( (r"r).rkrwc |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)rT)rP arguments)role function_callr)fn_namefn_reprrgrkrwschemaN)rtyper%r'rroappendmodel_dump_jsonrTr^infojsondumpsr(rrD model_dumpmodel_json_schema) r`r)rgrkrwrPrSrmopenai_function_callro function_bodys r#rvzInstructions.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"rmcD|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)`)rUrDr8r=mapr itemsfilter) r`rPr)rgrkrmfunc_defstr_args str_kwargs call_argsrs r#rozInstructions.openai_kwargss  ! !&r*H,R0H99Sd^, IIHH H PD IIfTHj+ABC  $!J8*U #!)$q 2>  + r")rrrr%r'rr rr\Handlerintboolrrbrrr rrrr rcrtuplerrrvrrror!r"r#rNrNcs!# 8<*8*A*A"'*.,sm, SM,tGOO45 , ( ,  , , ', ,F#.6$59 LLsmL*+ L  L #>2 L E(N2334(1h;'((1eHnrs 5<V-4cN :Y /1y1 TYY -hsCx0-S-2 :(38,:::& Oc0B Ot Or"