o Æpæi=ã@s0ddlmZddlZGdd„dƒZddd„ZdS) é)Ú annotationsNcsJeZdZUdZdZded<e ¡Zd ‡fdd„ Z dd d „Z d d „Z ‡Z S)ÚEmbedderuaå¥å‘é‡æ¨¡åž‹å°è£…；进程内å•ä¾‹ï¼Œå¤šæ¬¡ ``Embedder()`` åªåˆå§‹åŒ–一次底层模型。NzEmbedder | NoneÚ _instanceÚreturncsd|jdur/|j|jdurtƒ |¡|_Wdƒ|jSWdƒ|jS1s*wY|jS)N)rÚ _init_lockÚsuperÚ__new__)Úcls©Ú __class__©ú./home/dore/onebot/Bot_Ero/core/llm/embedder.pyr s   þþ ÿýzEmbedder.__new__ÚNonecCs0t|ddƒrdSddlm}|dƒ|_d|_dS)NÚ _initializedFr)ÚSentenceTransformerzBAAI/bge-small-zhT)ÚgetattrÚsentence_transformersrÚmodelr)Úselfrr r r Ú__init__s    zEmbedder.__init__cCs t|tƒr|g}|jj|ddS)NT)Únormalize_embeddings)Ú isinstanceÚstrrÚencode)rÚtextsr r r Úembeds zEmbedder.embed©rr)rr) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ__annotations__Ú threadingÚLockrrrrÚ __classcell__r r r r rs   rrcCstƒS)ul在进程å¯åŠ¨æ—¶è°ƒç”¨ï¼Œæå‰åŠ è½½å¥å‘é‡æ¨¡åž‹ï¼Œé¿å…首æ¡ä¾èµ–å‘é‡çš„任务长时间阻塞。)rr r r r Úpreload_embedder_model!sr%r)Ú __future__rr"rr%r r r r Ús