测试GPU性能

applications/pipeline.py

@@ -6,13 +6,33 @@ import time
 import requests
 
 from applications.functions import title_sim_v2_by_list
-from applications.functions import get_article_titles
 
 
 class LongArticlesPipeline(object):
     """
     Long articles Pipeline
     """
+    @classmethod
+    def get_titles(cls, account_name, index_list):
+        """
+        :param account_name:
+        :param index_list:
+        :return:
+        """
+        url = "http://192.168.100.31:6062/title_list"
+        response = requests.request(
+            "POST",
+            url=url,
+            headers={},
+            json={
+                "account_name": account_name,
+                "index_list": index_list,
+                "min_time": None,
+                "max_time": None,
+                "msg_type": "9"
+            }
+        )
+        return response.json()['title_list']
 
     @classmethod
     def history_title(cls, account_nickname):
@@ -24,11 +44,11 @@ class LongArticlesPipeline(object):
         # if "【1】" in plan_name or "【2】" in plan_name:
         index_list_1 = [1, 2]
         index_list_2 = [1, 2, 3, 4, 5, 6, 7, 8]
-        account_title_list_1 = get_article_titles(
+        account_title_list_1 = cls.get_titles(
             account_nickname,
             index_list=index_list_1
         )
-        account_title_list_2 = get_article_titles(
+        account_title_list_2 = cls.get_titles(
             account_nickname,
             index_list=index_list_2
         )
@@ -57,7 +77,7 @@ class LongArticlesPipeline(object):
         """
         判断文章是否安全
         """
-        url = "http://192.168.100.31:8177/sensitive/is_sensitive"
+        url = "http://61.48.133.26:8177/sensitive/is_sensitive"
         body = {
             "text": title
         }

test/article_list.py

@@ -0,0 +1,20 @@
+"""
+@author: luojunhui
+"""
+import requests
+
+url = "http://192.168.100.31:6062/title_list"
+
+response = requests.request(
+    "POST",
+    url=url,
+    headers={},
+    json={
+        "account_name": "指尖奇文",
+        "index_list": [1, 2],
+        "min_time": None,
+        "max_time": None,
+        "msg_type": "9"
+    }
+)
+print(response.json())