From 6bac1630e5af5890a6922bdc624e591eb19a12eb Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 13 三月 2025 18:36:07 +0800
Subject: [PATCH] 知识库对接rg
---
app/service/v2/app_driver/chat_data.py | 43 +++++++++++++++++++++++++++++++++++--------
1 files changed, 35 insertions(+), 8 deletions(-)
diff --git a/app/service/v2/app_driver/chat_data.py b/app/service/v2/app_driver/chat_data.py
index dad38e2..d642748 100644
--- a/app/service/v2/app_driver/chat_data.py
+++ b/app/service/v2/app_driver/chat_data.py
@@ -16,18 +16,22 @@
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)
- if res.status_code == 200:
+ if res.status_code == 200 or res.status_code == 201:
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