提取关键词

dbgpt/app/dbgpt_server.py

@@ -43,6 +43,8 @@ from dbgpt.util.utils import (
 )
 
 from dbgpt.app.apps.utils.tools import import_modules
+from dbgpt.app.apps.utils.spach_keywords import my_spacy_nlp
+
 REQUEST_LOG_RECORD = False
 MIDDLEWARES = [
     "dbgpt.app.apps.core.middleware.register_request_log_middleware" if REQUEST_LOG_RECORD else None,

dbgpt/app/openapi/api_v1/api_v1.py

@@ -44,6 +44,14 @@ from dbgpt.util.executor_utils import (
 )
 from dbgpt.util.tracer import SpanType, root_tracer
 
+from dbgpt.app.apps.utils.spach_keywords import my_spacy_nlp
+
+from dbgpt.app.apps.vadmin.media.params.media_list import MediaListParams
+from dbgpt.app.apps.vadmin.media import schemas, crud
+from dbgpt.app.apps.vadmin.auth.utils.current import AllUserAuth, FullAdminAuth, OpenAuth
+from dbgpt.app.apps.vadmin.auth.utils.validation.auth import Auth
+
+
 router = APIRouter()
 CFG = Config()
 CHAT_FACTORY = ChatFactory()
@@ -338,11 +346,45 @@ async def chat_prepare(dialogue: ConversationVo = Body()):
     return Result.succ(resp)
 
 
+
 @router.post("/v1/chat/completions")
 async def chat_completions(
     dialogue: ConversationVo = Body(),
     flow_service: FlowService = Depends(get_chat_flow),
+    auth: Auth = Depends(OpenAuth())
 ):
+    print('----------------completions---------------->')
+    logger.info(f"-----用户输入为:---->:{dialogue.user_input}")
+    print('----------------begin---------------->')
+
+    #nlp = spacy.load("en_core_web_trf")
+    #nlp = spacy.load("zh_core_web_trf")
+    #nlp.add_pipe("textrank")
+    doc = my_spacy_nlp.nlp(dialogue.user_input)
+
+    words = []
+    # examine the top-ranked phrases in the document
+    for phrase in doc._.phrases:
+        #logger.info(f"----1--->:{phrase.rank}--->:{phrase.count}")
+        #logger.info(f"----2--->:{phrase.chunks[0]}")
+        words.append(phrase.chunks[0])
+
+    print(words)
+
+    if len(words) > 0 :
+        print(words[0])
+
+        # 取出词频率排名第一的词，进行数据库查询--->把关键词循环3次
+        mydic = {'page': 1, 'limit': 10, 'v_order': None, 'v_order_field': None, 'type': 1, 'group_id': None,
+                 'file_name': ('like', words[0])}
+
+        datas, count = await crud.MediaDal(auth.db).get_datas(**mydic, v_return_count=True)
+        logger.info(f"-----查询到的图片为:---->:{datas}")
+    else:
+        logger.info(f"-----没有找到需要查询的内容:---->")
+
+    print('----------------end---------------->')
+
     logger.info(
         f"chat_completions:{dialogue.chat_mode},{dialogue.select_param},{dialogue.model_name}"
     )