^hc! SSKrSSKrSSKrSSKJrJrJrJr SSKrS\ S\ S\ 4Sjr S\ S\ S \ S\ 4S jr SS \ S \ S\ 4S jjrS\ S\ S\ S\\\ \ 44SjrSS\\ S\ S\S\ 4SjjrS\ S\4SjrS \ S\\ 4SjrS\ S\\\ \ 44Sjrg)N)AnyDictListTuplemodelstringreturncS[S[4Sjn[R"U"U55n[UR U55nU$![ a gf=f)z.Returns the number of tokens in a text string.sr cRURSS5n[U5S:aUS$U$)N/)splitlen)r partss I/home/james-whalen/.local/lib/python3.13/site-packages/llm_utils/utils.pyextract_after_slash)count_tokens..extract_after_slash s*Qu:>uQx0q0r)strtiktokenencoding_for_modelrencodeKeyError)rrrencoding num_tokenss r count_tokensr s[1s1s1 ../B5/IJ01  s7A AAnum_input_tokensnum_output_tokens model_typec t0SSSS._SSSS._SSSS._SSSS._SSSS._S SS S._S SS S._S S S S._SS S S._SSSS._SSSS._SSSS._SSSS._SSSS._SSSS._SSSS._nURU5=n(d([SUSSRSU55S 35eUS!- US"-US!- US#--$)$a+ Calculate the cost of processing a request based on model type. Args: num_input_tokens (int): Number of input tokens. num_output_tokens (int): Number of output tokens. model_type (str): The type of GPT model used (model name). Returns: The cost of processing the request, in USD. zgpt-4-0125-preview )inputoutputzgpt-4-turbo-previewzgpt-4-1106-previewzgpt-4-vision-previewzgpt-4-1106-vision-previewzgpt-4<z gpt-4-0613z gpt-4-32kxzgpt-4-32k-0613zgpt-3.5-turbo-0125g?g?z gpt-3.5-turbozgpt-3.5-turbo-1106rzgpt-3.5-turbo-instructzgpt-3.5-turbo-16kzgpt-3.5-turbo-0613zgpt-3.5-turbo-16k-0613zUnknown model "z". Choose from: z, c3$# UHov M g7f)N).0ms r !calculate_cost..CsCcOb!AObs.i@Br$r%)get ValueErrorjoin)rrr PRICING_PER_MILLIONprice_per_millions rcalculate_costr7s 8b98 r:8 b9 8 " ; 8 $rR%@ 8 2,8 b18 rS18 B#68 s;8 3#68 Q78 !C1"=!8" qA6#8$ q9%8& !A";'8,"5!8!8!D D  Dj\)9$))CcObCc:c9dde f  9$'8'AA i '*;H*E E FrtextwidthcF/n/n/nSnURS5HnURS5(aZU(a%URU5 URU5 /nOURU5 U/nURU5 U(+nMsURU5 M URU5 URU5 /n/n[[ U55Hn X9(a&URX)5 URS5 M2/nX)HJnU(d/U(a&URU5 URS5 /nM7M9URU5 ML U(dMURU5 URS5 /nM UnUn[[ U55H9n X9(aMX)V s/sHn [ R "X5PM sn X)'M; SRUV s/sHn SRU 5PM sn 5$s sn fs sn f)aw Wraps text except for code blocks for nice terminal formatting. Splits the text into paragraphs and wraps each paragraph, except for paragraphs that are inside of code blocks denoted by ` ``` `. Returns the updated text. Args: text (str): The text to wrap. width (int): The width of the lines to wrap at, passed to `textwrap.fill`. Returns: The wrapped text. F z```T )r startswithappendrangertextwrapfillr4) r8r9blocks is_code_blockblock in_code_blockline copied_blockscopied_is_code_blockilbs rword_wrap_except_code_blocksrLLs FM EM 4  ??5 ! ! T" e$ e$   / --M LL ! MM%'M 3v;      + ' ' -E %,,U3,33E: " LL&"u$$U+$++E2# $F(M3v; :@)D)Qq0)DFI ;;f5f ! f5 66E6s ? H7H file_path start_lineend_linecJSnS[S[S[4Sjn[US5nUR5nUVs/sHot"UR 5U5PM nnSSS5 [ SU5n[ [W5U5nXaS- UU4$s snf!,(df  N==f) a Read lines from a file. Args: file_path (str): The path of the file to read. start_line (int): The line number of the first line to include (1-indexed). Will be bounded below by 1. end_line (int): The line number of the last line to include (1-indexed). Will be bounded above by file's line count. Returns: The lines read as an array and the number of the first line included. Raises: FileNotFoundError: If the file does not exist. r rJr c4[U5U:aU$USUS-$)zP Truncate the string to at most the given length, adding ellipses if truncated. Nz...)r)r rJs rtruncateread_lines..truncates$ q6A:HRa55= rrNr)rintopen readlinesrstripmaxminr)rMrNrOmax_chars_per_linerSflinesrFs r read_linesr_s !C!C!C! i  INO$++-);<O  Q #J3u:x(H q.8 ,j 99 P  sB BBB B"groupfirststripc  U(aU(aCUSR5(d+USSnUS- nU(aUSR5(dM+U(a>USR5(d&USSnU(aUSR5(dM&U[U5-S- n[[U55nSR[ U5VVs/sHupVSR X-XF5PM snn5nU$s snnf)a Add line numbers for each line in the input list. Args: group (List[str]): The lines to number. first (int): The number for the first line. strip (bool): Whether to strip leading and trailing blank lines. Returns: A string concatenation of the numbered lines. rrNr;z {0:>{1}} {2})rbrrr4 enumerateformat)r`rarblastmax_line_number_lengthrIrFresults rnumber_group_of_linesrjs E!HNN,,!"IE QJEE!HNN,,E"IOO--#2JEE"IOO-- 3u:  !D T^ YY%U+ +  ! !%)-C J+ F M  s C? my_stringc,URS5nURS5S-nUS:XdUS:XagXUnSn[R"USU[RS 9n[ R "US S 9$![ Ra nSnAgSnAff=f) z Parses JSON if valid, replacing any triple quotes with single quotes. Returns the parsed JSON blob if successful, None if not. {}rrdrNz """(.*?)"""cL[R"URS55$)Nr)jsondumpsr`)r.s r%contains_valid_json..s4::aggaj1r)flagsF)strict)findrfindresubDOTALLrploadsJSONDecodeError)rk start_posend_pos json_stringpatternprocessed_stringes rcontains_valid_jsonrss#Iooc"Q&GB'Q,g.KGvv1;biizz*599    s A55BBcSn[R"X[R5nUVs/sHo3SR5PM sn$s snf)Nz```(python)?(.*?)```r)rxfindallrzrb)r8rrBrDs rextract_code_blocksrs<%G ZZryy 1F*0 1&!HNN & 11 1sA logcURS5n/nUHNnUR5(dMURSS5tpEU(aUSOSnURXFS.5 MP U$)Nr<z: rr)rolecontent)rrbr>)rentriesrientryrr content_strs r parse_chatlogrsbiiG F{{}} T1-$+gaj  t<=  Mr)P)T)rxrpr@typingrrrrrrrVrfloatr7rLr_boolrjrrrr,rrrs ))  S S "...1.?B. .bB7sB73B7B7J#:##:3#:##:%S SVBW#:Lc3ts<33:2c2d3i2 s tDcN3 r