o ?Gh  @s@ddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddlZddlmZddlmZe dZd Zd Zed Zd d ddgfdede jdedefddZefdedefddZ d(dede jdedededef ddZefdedefd d!Zd"d#Zed$dfdededed%edef d&d'Z dS))N)display) load_dotenv)spatial)get_openai_client)datetimeztext-embedding-3-small cl100k_basei@ GPT_MODELcCsdtj||SNr)rdistancecosine)xyr;/var/www/html/tg_maintain_bot/openai_config/openai_utils.pysrdFquerydftop_nshow_relatednessesc st}|jj|gtd}|jdjfdd|D}dd|D} | jdddd | s@td d t d t DSt | } t | } td d t | d| dD| d<t d| dtfdd | DS)Ninputmodelrcs<g|]\}ddftfddDqS)combined ada_embeddingc3s|]}|VqdSNr).0colrowrr (sz;strings_ranked_by_relatedness...)tuple)ri)additional_columnsquery_embeddingrelatedness_fnrr 's*z1strings_ranked_by_relatedness..cSsg|] }|ddkr|qS)rg333333?r)ritemrrrr',scSs|dSr rr rrrr0sz/strings_ranked_by_relatedness..T)keyreversecss|]}gVqdSrr)r_rrrr!3sz0strings_ranked_by_relatedness..css(|]\}}|dt|dVqdS)u 相關係數: N)round)rqrelrrrr!8s&r相關係數: c3s|] }|dVqdSrr)rresult)rrrr!<s)r embeddingscreateEMBEDDING_MODELdata embeddingiterrowssortr"rangelenziplistprint) rrr&rrr$clientquery_embedding_responsestrings_and_relatednesses"filtered_strings_and_relatednessesresultsr)r$r%r&rrstrings_ranked_by_relatednesss( $rFtextrcCst|}t||Sr)tiktokenencoding_for_modelr=encode)rGrencodingrrr num_tokens?s rL token_budget introductionreturnc Cst||ddd\}}|}|}|}td|t|dkr dS|D]} d| d} t|| ||d |kr;||S|| 7}q"||S) NT)rrr3r 無法辨識z! Wikipedia article section: """ z """)r)rFr@r=rL) rrrrNrOstrings relatednessesquestionmessagestring next_articlerrr query_messageCs    rYcCs,t}|dd}|jj|g|djdjS)N  rr)rreplacer5r6r8r9)rGrrArrr get_embedding^s r]cstjdrtd}|jtj|d<|Std}|gd}d|ddd|d dd |d |d <d }t t |j fdd|d<||j tk|}|j dd|d<|jdddtd|S)Nudata/維護訊息_embedding.csvrudata/維護訊息22.csv)類型 訊息內容 遊戲平台u問題類型為: r^z u內容訊息為: r_u遊戲平台為: r`rrcst|Sr)r=rJr)rKrrrvsz3load_or_create_maintain_embedding..n_tokenscSs t|tSr)r]r7r)rrrrxs F)indexz!!)ospathexistspdread_csvrapplyast literal_evalrHrIr7rrb MAX_TOKENStailto_csvr@)rrrrar!load_or_create_maintain_embeddingfs4     roi  print_messagec Cst}t}t}|d}|d}t|t||||d|d|d|d|d|d|d d } |r:t| | d kr@d } d ddd| dg} |jjj || dd} | 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 如果類型為你沒有看過的資訊請遵循以下規定： 回傳 例如: 訊息種類：未知 ---------------------------- 原始訊息：[“用戶回傳的訊息”] )rrNrOrRu 無法回答systemu 請根據使用者送出的訊息回答最相關的答案， 如果訊息相關度不高或是具有冒犯性的話， 請回覆“無法回答”。 )rolecontentuserr)rmessages temperature) rorrnowstrftimer@rYchat completionsr6choicesrVrs) rrrNrprrArw fmt_time_date fmt_time_timerVruresponseresponse_messagerrrask~sD   ( 2r)rM)!rdsysrjcsvnumpynpIPython.displayrdotenvrscipyrpandasrgrHopenai_connectrrr7EMBEDDING_ENCODINGrlgetenvr str DataFrameintboolrFrLrYr]rorrrrrsz       '