CodeCow/chat2api
1
0
Fork 0
mirror of https://github.com/lanqian528/chat2api.git synced 2026-06-13 21:02:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

every 1 hour, we get wss_url again

Browse Source
This commit is contained in:
Yuanzhang Hu 2024-05-11 09:21:51 -04:00
parent b0841b3134
commit ccd8aa7874
No known key found for this signature in database
GPG Key ID: 83B3E45AFE1D32EF
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -1,2 +1,3 @@
.env
/.idea/
*.pyc

6
chatgpt/ChatService.py
View File

@ -2,6 +2,7 @@ import json
import random
import types
import uuid
import time
import websockets
from fastapi import HTTPException
@ -22,6 +23,7 @@ class ChatService:
self.ws = None
self.wss_url = None
self.wss_mode = False
self.lasttime_get_wss_url = 0
self.user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36 Edg/124.0.0.0"
self.access_token = access_token
self.chat_token = "gAAAAAB"
@ -226,8 +228,9 @@ class ChatService:
wss_r = None
try:
if self.wss_mode:
if not self.wss_url:
if not self.wss_url or time.time() > 3600 + self.lasttime_get_wss_url:
self.wss_url = await self.get_wss_url() # first wss_url
self.lasttime_get_wss_url = time.time()
# websocket connection has to be established before posting conversation to avoid message loss.
if self.wss_url:
self.ws = await websockets.connect(self.wss_url, ping_interval=None, subprotocols=subprotocols)
@ -257,6 +260,7 @@ class ChatService:
rtext = await r.atext()
resp = json.loads(rtext)
self.wss_url = resp.get('wss_url') # used for next wss_url
self.lasttime_get_wss_url = time.time()
logger.info(f"next wss_url: {self.wss_url}")
if wss_r:
try: