From e5e2be29cfc033c976558a4d0b5fa73426818b90 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 26 二月 2025 11:30:48 +0800 Subject: [PATCH] 增加系统信息修改 --- app/service/v2/app_driver/chat_data.py | 41 ++++++++++++++++++++++++++++++++++------- 1 files changed, 34 insertions(+), 7 deletions(-) diff --git a/app/service/v2/app_driver/chat_data.py b/app/service/v2/app_driver/chat_data.py index 308f0fc..d642748 100644 --- a/app/service/v2/app_driver/chat_data.py +++ b/app/service/v2/app_driver/chat_data.py @@ -16,11 +16,6 @@ else: return {} - async def chat_ping(self, url, params, headers): - res = await self.http_get(url, params, headers) - return res.status_code - - async def chat_post(self, url, data, headers): res = await self.http_post(url, data, headers) @@ -28,6 +23,15 @@ return res.json() else: return {} + + async def chat_ping(self, url, params, headers): + res = await self.http_get(url, params, headers) + # print(res.text) + if res.status_code != 200: + return 0 + if res.json().get("code") == "unauthorized" or res.json().get("code") == 401: + return 0 + return 200 async def chat_login(self, url, data, headers): @@ -41,12 +45,26 @@ else: return {} + async def chat_upload(self, url, files, data, headers): + + res = await self.http_upload_file(url, headers=headers, files=files, data=data) + if res.status_code == 200 or res.status_code == 201: + return res.json() + else: + return {} + @staticmethod async def password_encrypt(password): 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 +87,18 @@ '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.116:11080/v1/user/info" + # ping_url = "http://smartai.com:8294/v1/llm/list" + # ping_url = "http://192.168.20.119:13002/console/api/workspaces" + user_token = "eyJhbG|ciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoiNjEzNzdiYzctZTViYy00YjhiLTgxYTYtNWZkOTVhODVlMmE4IiwiZXhwIjoxNzM5MjU3Njk1LCJpc3MiOiJTRUxGX0hPU1RFRCIsInN1YiI6IkNvbnNvbGUgQVBJIFBhc3Nwb3J0In0.w7xQrepd1dYR4iPXcbuthIZjdm45bTJFbolOM_SE9aQ" + user_token = "IjU4OTM5M2UyZjMyNzExZWZhZmVjMDI0MmFjMTIwMDA2Ig.Z702yg.Pmyy58wg-YBU5t50mHuqvwTDyjc" + # token = "Bearer {}" + token = "{}" + res = await chat.chat_ping(ping_url, {}, await chat.get_chat_headers(token.format(user_token))) + print(res) import asyncio -- Gitblit v1.8.0