From 48e33c7fbd6969d05b2affda0321595d2d6b370a Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 10 二月 2025 14:51:06 +0800
Subject: [PATCH] 增加ping機制

---
 app/service/v2/app_driver/chat_data.py |   15 +++++++++++++--
 1 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/app/service/v2/app_driver/chat_data.py b/app/service/v2/app_driver/chat_data.py
index 308f0fc..97939d2 100644
--- a/app/service/v2/app_driver/chat_data.py
+++ b/app/service/v2/app_driver/chat_data.py
@@ -47,6 +47,12 @@
         password = RagflowCrypto(settings.PUBLIC_KEY, settings.PRIVATE_KEY).encrypt(password)
         return password
 
+    @staticmethod
+    async def get_chat_headers(token):
+        return {
+            'Content-Type': 'application/json',
+            'Authorization': token
+        }
 
 
 
@@ -69,9 +75,14 @@
             'Content-Type': 'application/json',
             'Authorization': f"Bearer {token}"
         }
-        ans = await chat.chat_parameters(url, params, headers)
-        print(ans)
+        # ans = await chat.chat_parameters(url, params, headers)
+        # print(ans)
 
+        ping_url = "http://192.168.20.119:13002/console/api/workspaces"
+        user_token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoiNzBhNWVhZWYtYzU4Yi00YTE0LTliOGQtYWE2MzY0ZmE1ZDE3IiwiZXhwIjoxNzM5MjU2MDMyLCJpc3MiOiJTRUxGX0hPU1RFRCIsInN1YiI6IkNvbnNvbGUgQVBJIFBhc3Nwb3J0In0.tlxdRIVL2Ewgb0V4M5pj6v_U2yyvQxtCDrbvSxLjUBs"
+        token = "Bearer {}"
+        res = await chat.chat_ping(ping_url, {}, await chat.get_chat_headers(token.format(user_token)))
+        print(res)
 
     import asyncio
 

--
Gitblit v1.8.0