Update app.py

app.py

@@ -33,13 +33,48 @@ def check_api_key():
 
 @app.route('/v1/models', methods=['GET'])
 def get_models():
+    global COOKIES
     logging.info("Received /v1/models request")
     _cookie = get_cookie()
-    logging.info(_cookie[:50])
+    logging.info(_cookie)
     headers = {"Content-Type": "application/json", "Cookie": _cookie}
+
     response = requests.get('https://chat.akash.network/api/models', headers=headers)
     models_data = response.json()
     print(models_data)
+    if models_data.get('success',False) == False:
+        response = requests.post('https://chat.akash.network/api/auth/session/refresh', headers=headers)
+        _new_cookies = response.cookies
+        cookie_str = '; '.join([f'{cookie.name}={cookie.value}' for cookie in _new_cookies])
+        COOKIES = [cookie_str if x == _cookie else x for x in COOKIES]
+        headers = {"Content-Type": "application/json", "Cookie": cookie_str}
+        response = requests.get('https://chat.akash.network/api/models', headers=headers)
+        models_data = response.json()
+        current_timestamp = int(time.time())
+        converted_data = {
+            "object": "list",
+            "data": [
+                {
+                    "id": model["id"],
+                    "object": "model",
+                    "created": current_timestamp,
+                    "owned_by": "openai" if "Meta" in model["id"] else "third_party",
+                    "permissions": [],
+                    "root": model["id"],
+                    "parent": None,
+                    "capabilities": {
+                        "temperature": model.get("temperature"),
+                        "top_p": model.get("top_p")
+                    },
+                    "name": model.get("name"),
+                    "description": model.get("description"),
+                    "available": model.get("available")
+                }
+                for model in models_data
+            ]
+        }
+        logging.info("Response for /v1/models: %s", json.dumps(converted_data, ensure_ascii=False))
+        return jsonify(converted_data)
     current_timestamp = int(time.time())
     converted_data = {
         "object": "list",
@@ -93,7 +128,7 @@ def generate_stream(akash_response, chat_id, model):
                         "finish_reason": None
                     }]
                 }
-                logging.debug("Streaming chunk: %s", json.dumps(chunk, ensure_ascii=False))
+                logging.info("Streaming chunk: %s", json.dumps(chunk, ensure_ascii=False))
                 yield f"data: {json.dumps(chunk, ensure_ascii=False)}\n\n"
             elif msg_type in ['e', 'd']:
                 chunk = {
@@ -107,7 +142,7 @@ def generate_stream(akash_response, chat_id, model):
                         "finish_reason": "stop"
                     }]
                 }
-                logging.debug("Streaming finish chunk: %s", json.dumps(chunk, ensure_ascii=False))
+                logging.info("Streaming finish chunk: %s", json.dumps(chunk, ensure_ascii=False))
                 yield f"data: {json.dumps(chunk, ensure_ascii=False)}\n\n"
                 yield "data: [DONE]\n\n"
                 break
@@ -117,10 +152,13 @@ def generate_stream(akash_response, chat_id, model):
 
 @app.route('/v1/chat/completions', methods=['POST'])
 def chat_completions():
+    global COOKIES
     try:
         data = request.get_json()
         logging.info("Received /v1/chat/completions request: %s", json.dumps(data, ensure_ascii=False))
         chat_id = str(uuid.uuid4()).replace('-', '')[:16]
+        _cookie = get_cookie()
+        logging.info(_cookie)
         model = data.get('model', "DeepSeek-R1")
         akash_data = {
             "id": chat_id,
@@ -130,16 +168,13 @@ def chat_completions():
             "temperature": data.get('temperature', 0.6),
             "topP": data.get('top_p', 0.95)
         }
-        _cookie = get_cookie()
-        logging.info(_cookie[:50])
         headers = {"Content-Type": "application/json", "Cookie": _cookie}
+
         # 默认 stream 模式开启，但针对 AkashGen 模型关闭流式响应
-        stream_flag = True
+        stream_flag = data.get('stream', True)
         if model == "AkashGen":
             stream_flag = False
-        else:
-            stream_flag = data.get('stream',False)
-        logging.info("streamflag: %s", stream_flag)    
+
         akash_response = requests.post(
             'https://chat.akash.network/api/chat',
             json=akash_data,
@@ -147,7 +182,19 @@ def chat_completions():
             stream=stream_flag
         )
         logging.info("Akash API response status: %s", akash_response.status_code)
-
+        if akash_response.status_code == 403:
+            response = requests.post('https://chat.akash.network/api/auth/session/refresh', headers=headers)
+            _new_cookies = response.cookies
+            cookie_str = '; '.join([f'{cookie.name}={cookie.value}' for cookie in _new_cookies])
+            COOKIES = [cookie_str if x == _cookie else x for x in COOKIES]
+            headers = {"Content-Type": "application/json", "Cookie": cookie_str}
+            akash_response = requests.post(
+                'https://chat.akash.network/api/chat',
+                json=akash_data,
+                headers=headers,
+                stream=stream_flag
+            )
+        
         if stream_flag:
             return Response(
                 generate_stream(akash_response, chat_id, model),
@@ -198,7 +245,7 @@ def chat_completions():
                                         "index": 0,
                                         "message": {
                                             "role": "assistant",
-                                            "content": f"{img_data[0]['result']}"
+                                            "content": f"根据您的描述，这里是一张生成的图片：\n\n![生成的图片]({img_data[0]['result']})"
                                         },
                                         "finish_reason": "stop"
                                     }]