o
|
Öph ; ã
|
@ s´ d dl mZmZmZmZ d dlZd dlmZ d dlm Z d dl
|
Z
|
d dl Z d dl Z d dl Z d dlZd dlmZ d dlZd dlmZ eeZe Ze Ze ¡ Zej e¡ dd Ze je j d e ¡ e d
|
¡gd e e¡Z d d Z!dd Z"dd Z#ej$dddgddd Z%dd Z&dd Z'dd ej(ej)dd dfd d Z*ed krØz e! e" d e_+W n e,yÉ Z- ze .d!e- ¡ e/d" W Y dZ-[-ndZ-[-ww e 0d#¡ ej1d$d%d& dS dS )'é )ÚFlaskÚrequestÚabortÚjsonifyN)ÚDatabaseManager)Ú
|
WeComUtils)Ú RedisManager)Ú global_configc C sF t j d¡r td } t | ¡W d S 1 sw Y td)Nz config.jsonu" é
ç½®æ件 config.json ä¸åå¨)ÚosÚpathÚexistsÚopenÚjsonÚloadÚFileNotFoundError)Úf© r ú*E:\project\python\wecom-dify-bridge\app.pyÚ load_config s
|
|
ÿr z4%(asctime)s - %(name)s - %(levelname)s - %(message)szapp.log)ÚlevelÚformatÚhandlersc
|
C sN zt d¡ t ¡ t d¡ W dS ty& } z t d| ¡ d} ~ ww )u åå§ååºç¨zInitializing application...z$Database initialized and data cachedzInitialization error: N)ÚloggerÚinfoÚ
|
db_managerÚinit_dbÚ ExceptionÚerror©Úer r r Úinitialize_app1 s
|
þr c C s dd } t j| dd ¡ dS )u å¯å¨åå°ä»»å¡c
|
S sb t d¡ z
|
t t d¡ W n ty/ } zt d| ¡ t d¡ W Y d } ~ nd } ~ ww q)NzMessage persister startedTé< zMessage persister error: )r r Úpersist_expired_messagesÚtimeÚsleepr r r r r r Úmessage_persister@ s
|
þüz1start_background_tasks.<locals>.message_persisterT)ÚtargetÚdaemonN)Ú threadingÚThreadÚstart)r% r r r Ústart_background_tasks> s
|
r+ c C s| t j d} t }d}|dkrAtjj|| d\}}|D ]!}t|tr(| d¡}n|}| d¡}t
|
|dkr<| |d ¡ q|dkst |}t ¡ }td td
|
| |D ]e}t |¡}t |¡} td | td | td | |szqV|d d }
|
||
|
t jkr»zt ||| ¡ t |¡ t d| ¡ W qV tyº } zt d| d| ¡ W Y d} ~ qVd} ~ ww qVdS )u æä¹
åè¿ææ¶æ¯z:user_messages:*Ú0r )ÚcursorÚmatchúutf-8ú:é é u æ§è¡æä¹
åzkeys:zuser_id:z messages:zconversation_id:éÿÿÿÿÚ timestampz Persisted messages for user: z$Failed to persist messages for user z: N)ÚconfigÚREDIS_KEY_PREFIXÚsetÚ redis_managerÚredisÚscanÚ
|
isinstanceÚbytesÚdecodeÚsplitÚlenÚaddÚlistr# ÚprintÚget_user_messagesÚget_user_conversation_idÚMYSQL_PERSIST_EXPIREr Úsave_message_historyÚclear_user_messagesr r r r ) ÚpatternÚ
|
unique_idsr- ÚkeysÚkeyÚkey_strÚpartsÚ current_timeÚmessagesÚconversation_idÚlast_message_timer r r r r" M sL
|
|
õ
|
|
|
"ÿûör" z/wecom/callbackÚGETÚPOST)Úmethodsc
|
à sz t dtj d¡ i } ttdsBzt t dt_t d¡ W n t yA } zt
|
d| ¡ t dd W Y d }~nd }~ww tjd
|
krtj d d ¡}tj d d ¡}tj dd ¡}tj dd ¡}t ||||¡\}}|dkrxt d¡ |S t
|
d¡ t dd d S tjdkr;tj d d ¡}tj d d ¡}tj dd ¡}t d| d| d| ¡ tj d¡}t d| ¡ i } zt ||||¡\}} | d¡} t | ¡} W n t yò } zt
|
d| ¡ t dd W Y d }~nd }~ww | d d kr#t t| ||¡ t | d d | d! ¡}
|
tj|
|
dd"} t | ||¡\}} t| S | d d#kr4t | ||} t| S t d$¡ d%S d S )&u! ä¼ä¸å¾®ä¿¡æºå¨äººåè°æ¥å£z Received z request at /wecom/callbackÚ initializedTzApplication initializedzInitialization failed: iô z Server initialization failedNrR Ú msg_signatureÚ r4 ÚnonceÚechostrr z!Signature verification successfulz Signature verification failedi rS z#POST request params: msg_signature=z , timestamp=z, nonce=r/ z post data: zError processing message: é zFailed to process messageÚmsgtypeÚtextÚfromÚuseridÚmsgid©ÚfirstÚstreamz;Ignoring non-text message of type or non-group chat of typeÚsuccess) r r r ÚmethodÚhasattrÚappr r+ rU r r r ÚargsÚgetr Ú
|
verify_urlÚdatar= Ú decrypt_msgr ÚloadsÚasyncioÚ create_taskÚhandler_text_requestÚgenerate_stream_idÚcreate_response_messageÚ encrypt_msgr Úhandler_stream_request) Úresponser rV r4 rX rY ÚretÚ reply_echostrÚ post_dataÚmsg_dataÚ stream_idÚ response_textr r r Úwecom_callback{ sl
|
þ
|
|
|
|
þ
|
Ør{ c
|
|
à s& | d d }| d }t ||¡}| d d tjd¡}| ¡ }t| d| t |¡}z|s>t |d|¡ t
|
|d ¡ t d
|
¡ W n t y^ } zt d | ¡ W Y d }~nd }~ww t |¡} z|sut|||i i | |d W d S W d S t y } zt d | ¡ W Y d }~d S d }~ww )Nr] r^ r_ r\ ÚcontentrW z-user: Úuserr, z Stored user message in Redisz Failed to store user message: )Úqueryr_ r} ÚinputsÚfilesrP ry zFailed to call Dify API: )r rp Úreplacer5 ÚWECOM_BOT_NAMEÚstriprB r8 Ú exists_streamÚadd_user_messageÚadd_stream_lockr r r r rD Úsend_chat_message)
|
rx rX r4 Úuser_idÚmsg_idry Ú msg_contentr r rP r r r ro Å s4
|
ÿ
|
ÿ ÿro c C sj | d d }t |¡rt |¡}t |||¡\}}|S tj|dd}t |||¡\}}t d| ¡ |S )Nrb ÚidTr` u æµå¼ json: )r8 r r rq rr r Údebug)rx rX r4 ry rz ru rt r r r rs ê s
|
|
rs Ú default_userTc
|
|
C s | st d|s t d|r|dkrt d| d¡ d}
|
d| dd } | d
|
||d } |r3|| d < i | d < i | d< zedddd} tj|
|
| | dd}|jdkrc| ¡ }| dd¡| d d¡dd} n<z| ¡ }d|j d| dd¡ }|||jd} W n tjy d|j d|j }|||jd} Y nw tj
|
|¡W nZ tj j yµ d|dd} tj
|
d¡ tj j yÉ d |d d} tj
|
d ¡ tj j
|
yÞ } zd |d d} |d }~w t yú } zd|d!d} tj
|
d"t| ¡d }~ww | d# dks | d# d$krt |d%| d ¡ t || d ¡ t | d ¡}t ||¡ t || d# ¡ t || d ¡ t| d&| d d S )'ue
|
åéè天æ¶æ¯å° Dify API
|
|
åæ°:
|
query (str): ç¨æ·è¾å
¥/æé®å
容
|
conversation_id (str, optional): ä¼è¯ IDï¼ç¨äºç»§ç»ä¹åç对è¯
|
user (str): ç¨æ·æ è¯ï¼ç¨äºå®ä¹ç»ç«¯ç¨æ·ç身份
|
api_key (str): Dify API å¯é¥
|
base_url (str): Dify API åºç¡ URL
|
inputs (dict, optional): App å®ä¹çååéå¼
|
auto_generate_name (bool): æ¯å¦èªå¨çææ é¢
|
|
è¿å:
|
dict: API ååºç»æ
|
|
å¼å¸¸:
|
requests.exceptions.RequestException: ç½ç»è¯·æ±å¼å¸¸
|
ValueError: åæ°éªè¯å¤±è´¥
|
u query åæ°ä¸è½ä¸ºç©ºu user åæ°ä¸è½ä¸ºç©ºÚ YOUR_API_KEYu 请设置ææç API å¯é¥ú/z/chat-messageszBearer zapplication/json)Ú Authorizationz Content-TypeÚblocking)r~ Ú response_moder} Úauto_generate_namerP r r rW r )ÚanswerrP Ústatuséx )ÚurlÚheadersr ÚtimeoutéÈ r u API é误: z - Úmessageu æªç¥é误u3 请æ±è¶
æ¶ï¼è¯·æ£æ¥ç½ç»è¿æ¥æç¨åéè¯i u2 è¿æ¥å¤±è´¥ï¼è¯·æ£æ¥ç½ç»è¿æ¥å API å°årZ u 请æ±é误iæ Niç u æªç¥é误: r Ú200Ú assistantz -assistant: )Ú
|
ValueErrorÚrstripÚrequestsÚpostÚ status_coder rh ÚJSONDecodeErrorr\ Ú
|
exceptionsÚRequestExceptionÚTimeoutÚConnectionErrorr Ústrr8 r
Úadd_conversation_idr Úsplit_string_safelyÚadd_stream_chunksÚadd_stream_statusrB )r~ r_ ry rP r} Úapi_keyÚbase_urlr r r r r ÚpayloadÚresrt Ú response_jsonÚ
|
error_dataÚ error_msgr Úchunksr r r r s® ! þüýü
|
|
|
ý
|
ý
|
ýþ éý ý ýýú r Ú__main__z"Failed to initialize application: é zStarting Flask applicationz0.0.0.0iG )ÚhostÚport)2Úflaskr r r r rm Údatabaser Ú wecom_utilsr r# r( Úloggingr r r8 r r
|
Úconfig_managerr Ú__name__rf r Ú
|
get_configr5 Ú from_objectr Ú basicConfigÚDEBUGÚ StreamHandlerÚ FileHandlerÚ getLoggerr r r+ r" Úrouter{ ro rs Ú DIFY_API_KEYÚDIFY_API_BASE_URLr rU r r ÚcriticalÚexitr Úrunr r r r Ú<module> sn þý
|
.
|
I%%
|
ö
|
þ
|
ô
|
