update device id

personas_vector/match_api.py

@@ -20,23 +20,25 @@ TAG_EMBEDDING_DICT = redis_client_db.hgetall("personas_tags_embedding")
 @bind("strategy_embedding/personas_vector/match")
 def match_tractate_by_device(device_id, n=10):
     try:
-        time_begin = time.time()
-        business_tags = get_user_portrait_tag3_from_redis(device_id, tags_num=4).get("business_tags", [])
-        # business_tags = ["假体下巴", "你好", "假体隆胸"]
-        res = []
-        vectors = []
-        for tag in business_tags:
-            lst = json.loads(TAG_EMBEDDING_DICT.get(bytes(tag, "utf-8"), b"[]"))
-            if lst:
-                vectors.append(np.array(lst).astype("float32"))
-        if vectors:
-            D, I = FAISS_TAGS_INDEX.search(np.array([np.average(vectors, axis=0)]).astype("float32"), n)
-            res = I.tolist()[0]
-        time_end = time.time() - time_begin
-        if time_end > 0.04:
-            send_performance_msg_to_dingtalk("match_tractate_by_device {} n={} cost {:.3f}ms".format(
-                device_id, n, time_end * 1000))
-        return res
+        if device_id not in ["0", "unknown"]:
+            time_begin = time.time()
+            business_tags = get_user_portrait_tag3_from_redis(device_id, tags_num=4).get("business_tags", [])
+            # business_tags = ["假体下巴", "你好", "假体隆胸"]
+            res = []
+            vectors = []
+            for tag in business_tags:
+                lst = json.loads(TAG_EMBEDDING_DICT.get(bytes(tag, "utf-8"), b"[]"))
+                if lst:
+                    vectors.append(np.array(lst).astype("float32"))
+            if vectors:
+                D, I = FAISS_TAGS_INDEX.search(np.array([np.average(vectors, axis=0)]).astype("float32"), n)
+                res = I.tolist()[0]
+            time_end = time.time() - time_begin
+            if time_end > 0.04:
+                send_performance_msg_to_dingtalk("match_tractate_by_device {} n={} cost {:.3f}ms".format(
+                    device_id, n, time_end * 1000))
+            return res
+        return []
     except Exception as e:
         send_msg_to_dingtalk(str(traceback.format_exc()))
         return []