Fh  hddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddlZddlmZddlmZe dZd Zd Zejd Zd d dgfdede jdedefdZefdedefdZ d!dede jdedededef dZefdedefdZdZeddfdededededef d Z dS)"N)display) load_dotenv)spatial)get_openai_client)datetimeztext-embedding-3-small cl100k_basei@ GPT_MODELcHdtj||z SNr)rdistancecosine)xys K/Users/gualin/Documents/wsky_tg_auto_maintain/openai_config/openai_utils.pyrsQ)9)@)@1)E)E%EdFquerydftop_nshow_relatednessesc  t}|j|gt}|jdj  fd|D}d|D} | dd| s6tdtd tzDSt| } t| } td t| d | d D| d <td | d tfd| DS)Ninputmodelrcg|];\}ddftfdDzz;strings_ranked_by_relatedness...(s+V~V~dgWZ[^W_V~V~V~V~V~V~r)tuple)r#ir%additional_columnsquery_embeddingrelatedness_fns @r z1strings_ranked_by_relatedness..'sl!!! !S :O8LMMNQVV~V~V~V~k}V~V~V~Q~Q~~!!!rc*g|]}|ddk|S)rg333333?r")r#items rr,z1strings_ranked_by_relatedness..,s(***d1gnnnnnrc|dSr r"rs rrz/strings_ranked_by_relatedness..0s 1rT)keyreversec3K|]}gVdSr!r")r#_s rr&z0strings_ranked_by_relatedness..3s"DDARDDDDDDrc3FK|]\}}|dt|dVdS)u 相關係數: N)round)r#qrels rr&z0strings_ranked_by_relatedness..8s<hhC1<.<s+66F666666r)r embeddingscreateEMBEDDING_MODELdata embeddingiterrowssortr'rangelenziplistprint) rrr+rrr)clientquery_embedding_responsestrings_and_relatednesses"filtered_strings_and_relatednessesresultsr*s `` ` @rstrings_ranked_by_relatednessrPs F%077 8  /3A6@O!!!!!!KKMM!!! **2***&'++ d+KKK -EDDq3/A+B+B'B!C!CDDDDDD56G7mmGhh3wWY{\cde\fKgKghhhhhGBK '71: ' '((( 6666g666 6 66rtextrcntj|}t||Sr!)tiktokenencoding_for_modelrGencode)rQrencodings r num_tokensrW?s,*511H xt$$ % %%r token_budget introductionreturnct||dd\}}|}|}|}td|t|dkrdS|D]*} d| d} t|| z|z| |krn|| z }+||zS) NT)rrr<r 無法辨識z! Wikipedia article section: """ z """)r)rPrJrGrW) rrrrYrZstrings relatednessesquestionmessagestring next_articles r query_messagereCs;5"AaefffG]LzHG *= * *+++ 7||q$$LvLLL w-8 F F F   E | #GG X rct}|dd}|j|g|jdjS)N  rr)rreplacer?r@rBrC)rQrrKs r get_embeddingrj^sY  F <<S ! !D   # # $   1 rctjdr=tjd}|jtj|d<|Stjd}|gd}d|dzdzdz|d zdzd z|d z|d <d }tj t|j fd|d<||j tk|}|j d|d<|ddt#d|S)Nudata/維護訊息_embedding.csvrudata/維護訊息22.csv)類型 訊息內容 遊戲平台u問題類型為: rlz u內容訊息為: rmu遊戲平台為: rnrrcHt|Sr!)rGrU)rrVs rrz3load_or_create_maintain_embedding..vsS9K9K5L5Lrn_tokensc,t|tSr!)rjrAr0s rrz3load_or_create_maintain_embedding..xs-/:Z:ZrF)indexz!!)ospathexistspdread_csvrapplyast literal_evalrSrTrArrp MAX_TOKENStailto_csvrJ)rrrVs @r!load_or_create_maintain_embeddingr~fsY w~~788 [: ; ; .44S5EFF? [2 3 3 888 9 "X, . ;  "$^"4 57A B  "$^"4 5 : .??**+L+L+L+LMM:  z) * / / 6 6 k//0Z0Z[[? 3% @@@ d  ri  print_messagect}t}tj}|d}|d}t |t ||||d|d|d|d|d|d|d } |rt | | d krd } d ddd| dg} |jj || d} | j dj j } t | | S)Nz%Y-%m-%dz%H:%M:%Su 使用這份csv的“訊息內容”， 先判斷出“訊息內容”的類型是什麼， 訊息內沒有提及”維護“，以及符合”遊戲平台“的字眼的話，那他通常不是維護訊息， 請你自行判斷u裡面的維護時間，不要直接回傳csv檔案裡面的訊息，那只是給你參考用的 而不是csv裡面的時間， 如果類型為“維護訊息”請遵循以下規定： 找出使用者回傳最相關的那列， 並且將“訊息內容”整理出一個資訊：遊戲平台 請注意， 我要你自己從用戶傳的訊息判斷出“開始維護”和“結束維護”是什麼時候， 如果內容有多個時間的話，關鍵字要找“遊戲”之類的， 我提供你一些建議： 1. 03:00 a.m. - 12:00 p.m. 的維護時間代表 03:00:00 - 12:00:00 2. 05/02 (一) 1:00 的維護時間代表 2025-05-02 01:00:00 ---- 如果內容沒有開始日期的資訊， 請自動將時間設為u，格式為YYYY-MM-DD 如果內容沒有結束日期的資訊，且訊息內容沒有描述結束時間的資訊， 請自動將時間設為u+1天，格式為YYYY-MM-DD， 請注意，如果有明確的結束時間，通常結束日期會跟開始日期相同， 例如： ※ 5/2 星期一 02:00-20:30，那麼結束日期就是2025-05-02 ※ 5/2(一) 05:00AM-14:30PM，那麼結束日期就是2025-05-02 ---- 如果內容沒有開始時間的資訊， 請自動將時間設為ux，格式為HH:mm:ss 如果內容沒有結束時間的資訊， 請自動將時間設為uv+1天，格式為HH:mm:ss 回傳“類型”,“遊戲平台”,“開始日期”,“開始時間”,“結束日期“,”結束時間“,“用戶傳過來的訊息” 例如: 訊息種類：[類型] 遊戲平台：ATG 開始日期：2025-10-01 開始時間：00:00:00 結束日期：2025-10-02 結束時間：00:00:00 ---------------------------- 原始訊息：ur 如果類型為你沒有看過的資訊請遵循以下規定： 回傳 例如: 訊息種類：未知 ---------------------------- 原始訊息：[“用戶回傳的訊息”] )rrYrZr^u 無法回答systemu 請根據使用者送出的訊息回答最相關的答案， 如果訊息相關度不高或是具有冒犯性的話， 請回覆“無法回答”。 )rolecontentuserr)rmessages temperature) r~rrnowstrftimerJrechat completionsr@choicesrbr) rrrYrrrKr fmt_time_date fmt_time_timerbrresponseresponse_messages raskr~s + , ,B  F ,..CLL,,MLL,,M %LLLE"5l1Y % 1Y1Y"*7#1Y1Y&*7'1Y1Y4*751Y1Y8*791Y1YP%*Q1Y1Y1Y111Gd Gnnn.   %   '** H{&--.H  '*2:  r)rX)!rssysrycsvnumpynpIPython.displayrdotenvrscipyrpandasrvrSopenai_connectrrrAEMBEDDING_ENCODINGr{getenvr str DataFrameintboolrPrWrerjr~rr"rrrsI ######------ *"  BIk " " FE"' $7$7$7 <$7 $7  $7$7$7$7N%.&&C&c&&&&   |      6(7s4" RR R RR R  RRRRRRr