parent
d357047662
commit
29dc7b5562
@ -0,0 +1,45 @@
|
|||||||
|
from Crypto.Cipher import AES
|
||||||
|
from Crypto.Random import get_random_bytes
|
||||||
|
from Crypto.Util.Padding import pad, unpad
|
||||||
|
|
||||||
|
def encrypt_message(message, key):
|
||||||
|
"""
|
||||||
|
加密函数
|
||||||
|
:param message: 要加密的明文消息
|
||||||
|
:param key: 加密密钥
|
||||||
|
:return: 密文
|
||||||
|
"""
|
||||||
|
# 创建 AES 对象
|
||||||
|
cipher = AES.new(key, AES.MODE_CBC)
|
||||||
|
|
||||||
|
# 对明文进行填充,然后加密
|
||||||
|
ct_bytes = cipher.encrypt(pad(message.encode(), AES.block_size))
|
||||||
|
|
||||||
|
# 将 IV 和密文返回
|
||||||
|
return cipher.iv, ct_bytes
|
||||||
|
|
||||||
|
def decrypt_message(iv, ct, key):
|
||||||
|
"""
|
||||||
|
解密函数
|
||||||
|
:param iv: 初始化向量
|
||||||
|
:param ct: 密文
|
||||||
|
:param key: 加密密钥
|
||||||
|
:return: 解密后的明文
|
||||||
|
"""
|
||||||
|
# 创建一个新的 AES 对象
|
||||||
|
cipher = AES.new(key, AES.MODE_CBC, iv)
|
||||||
|
|
||||||
|
# 解密并去除填充
|
||||||
|
pt = unpad(cipher.decrypt(ct), AES.block_size).decode()
|
||||||
|
|
||||||
|
return pt
|
||||||
|
|
||||||
|
# 测试
|
||||||
|
key = get_random_bytes(16) # 生成随机密钥
|
||||||
|
message = "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"
|
||||||
|
|
||||||
|
iv, ct = encrypt_message(message, key)
|
||||||
|
print("Encrypted:", ct)
|
||||||
|
|
||||||
|
pt = decrypt_message(iv, ct, key)
|
||||||
|
print("Decrypted:", pt)
|
@ -1,19 +1,5 @@
|
|||||||
from openai import OpenAI
|
import json
|
||||||
import openai
|
with open('demo.json','r',encoding='utf-8') as f:
|
||||||
import os
|
data = json.load(f)
|
||||||
openai.api_base = "https://api.openai-proxy.com"
|
for i,j in enumerate(data.items()):
|
||||||
openai.api_key = "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"
|
print(i,j[-1])
|
||||||
# os.environ['http_proxy'] = 'http://127.0.0.1:10809'
|
|
||||||
# os.environ['https_proxy'] = 'http://127.0.0.1:10809'
|
|
||||||
client = OpenAI(
|
|
||||||
api_key='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',
|
|
||||||
base_url='https://api.openai-proxy.com'
|
|
||||||
)
|
|
||||||
|
|
||||||
resp = client.chat.completions.create(
|
|
||||||
model='gpt-3.5-turbo',
|
|
||||||
messages=[
|
|
||||||
{'role': 'user', 'content': '你好!'}
|
|
||||||
]
|
|
||||||
)
|
|
||||||
print(resp.choices[0].message.content)
|
|
Loading…
Reference in new issue