github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-04-28 05:03:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add dalle3 gpt-4-turbo model change

Browse Source
This commit is contained in:
zhayujie
2023-11-10 10:11:02 +08:00
parent db8e506de0
commit f001ac6903
16 changed files with 166 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/linkai/summary.py
View File

@@ -13,7 +13,8 @@ class LinkSummary:
"file": open(file_path, "rb"),
"name": file_path.split("/")[-1],
}
res = requests.post(url=self.base_url() + "/v1/summary/file", headers=self.headers(), files=file_body, timeout=(5, 300))
url = self.base_url() + "/v1/summary/file"
res = requests.post(url, headers=self.headers(), files=file_body, timeout=(5, 300))
return self._parse_summary_res(res)
def summary_url(self, url: str):
@@ -71,7 +72,7 @@ class LinkSummary:
return False
suffix = file_path.split(".")[-1]
support_list = ["txt", "csv", "docx", "pdf", "md"]
support_list = ["txt", "csv", "docx", "pdf", "md", "jpg", "jpeg", "png"]
if suffix not in support_list:
logger.warn(f"[LinkSum] unsupported file, suffix={suffix}, support_list={support_list}")
return False