From 5cd2ae8cc8eee39b64c881e8d5a22207131495b2 Mon Sep 17 00:00:00 2001
From: 6vision <vision_wangpc@sina.com>
Date: Wed, 6 Nov 2024 21:45:03 +0800
Subject: [PATCH] Summary  supports app_code

---
 plugins/linkai/summary.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/plugins/linkai/summary.py b/plugins/linkai/summary.py
index 9a3c416..7060f2a 100644
--- a/plugins/linkai/summary.py
+++ b/plugins/linkai/summary.py
@@ -12,11 +12,14 @@ class LinkSummary:
     def summary_file(self, file_path: str, app_code: str):
         file_body = {
             "file": open(file_path, "rb"),
-            "name": file_path.split("/")[-1],
+            "name": file_path.split("/")[-1]
+        }
+        body = {
             "app_code": app_code
         }
         url = self.base_url() + "/v1/summary/file"
-        res = requests.post(url, headers=self.headers(), files=file_body, timeout=(5, 300))
+        logger.info(f"[LinkSum] file summary, app_code={app_code}")
+        res = requests.post(url, headers=self.headers(), files=file_body, data=body, timeout=(5, 300))
         return self._parse_summary_res(res)
 
     def summary_url(self, url: str, app_code: str):
@@ -25,6 +28,7 @@ class LinkSummary:
             "url": url,
             "app_code": app_code
         }
+        logger.info(f"[LinkSum] url summary, app_code={app_code}")
         res = requests.post(url=self.base_url() + "/v1/summary/url", headers=self.headers(), json=body, timeout=(5, 180))
         return self._parse_summary_res(res)
 
@@ -50,7 +54,7 @@ class LinkSummary:
     def _parse_summary_res(self, res):
         if res.status_code == 200:
             res = res.json()
-            logger.debug(f"[LinkSum] url summary, res={res}")
+            logger.debug(f"[LinkSum] summary result, res={res}")
             if res.get("code") == 200:
                 data = res.get("data")
                 return {