kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 13:47:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Summary supports app_code

Browse Source
This commit is contained in:
6vision
2024-11-06 21:45:03 +08:00
parent 22d67b3a59
commit 5cd2ae8cc8
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
plugins/linkai/summary.py
View File

@@ -12,11 +12,14 @@ class LinkSummary:
def summary_file(self, file_path: str, app_code: str): def summary_file(self, file_path: str, app_code: str):
file_body = { file_body = {
"file": open(file_path, "rb"), "file": open(file_path, "rb"),
"name": file_path.split("/")[-1], "name": file_path.split("/")[-1]
}
body = {
"app_code": app_code "app_code": app_code
} }
url = self.base_url() + "/v1/summary/file" 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) return self._parse_summary_res(res)
def summary_url(self, url: str, app_code: str): def summary_url(self, url: str, app_code: str):
@@ -25,6 +28,7 @@ class LinkSummary:
"url": url, "url": url,
"app_code": app_code "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)) res = requests.post(url=self.base_url() + "/v1/summary/url", headers=self.headers(), json=body, timeout=(5, 180))
return self._parse_summary_res(res) return self._parse_summary_res(res)
@@ -50,7 +54,7 @@ class LinkSummary:
def _parse_summary_res(self, res): def _parse_summary_res(self, res):
if res.status_code == 200: if res.status_code == 200:
res = res.json() res = res.json()
logger.debug(f"[LinkSum] url summary, res={res}") logger.debug(f"[LinkSum] summary result, res={res}")
if res.get("code") == 200: if res.get("code") == 200:
data = res.get("data") data = res.get("data")
return { return {