d0dc94e36d
新功能: - 创建AI聊天助手组件（AIChatWidget） - 支持文字+图片多模态对话 - 集成Gemini AI API - 支持图片编辑和内容生成 - 实时对话历史管理 - 优雅的UI设计，支持黑夜模式 技术实现: - 后端API代理Gemini API（/api/ai-chat） - 支持最多5张图片上传 - 流式响应处理 - 对话历史管理 - Base64图片处理 配置: - 在.env中添加Gemini API配置 - GEMINI_API_KEY - GEMINI_API_BASE_URL - GEMINI_MODEL 替换原弹幕功能，提供更强大的AI助手体验！
48 lines
2.0 KiB
Python
Executable File
48 lines
2.0 KiB
Python
Executable File
import os
|
|
import unittest
|
|
|
|
from volcengine.tls.TLSService import TLSService
|
|
|
|
class TestTLSService(unittest.TestCase):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.endpoint = os.environ["VOLCENGINE_ENDPOINT"]
|
|
self.region = os.environ["VOLCENGINE_REGION"]
|
|
self.access_key_id = os.environ["VOLCENGINE_ACCESS_KEY_ID"]
|
|
self.access_key_secret = os.environ["VOLCENGINE_ACCESS_KEY_SECRET"]
|
|
|
|
def test_tls_service(self):
|
|
tls_client1 = TLSService(self.endpoint, self.access_key_id, self.access_key_secret, self.region)
|
|
tls_client2 = TLSService(
|
|
self.endpoint + "test",
|
|
self.access_key_id + "test",
|
|
self.access_key_secret + "test",
|
|
self.region + "test"
|
|
)
|
|
|
|
self.assertNotEqual(tls_client1, tls_client2)
|
|
self.assertEqual(self.region, tls_client1.get_region())
|
|
self.assertEqual(self.region + "test", tls_client2.get_region())
|
|
|
|
def test_check_scheme_and_endpoint(self):
|
|
endpoint = "http://tls-cn-beijing.ivolces.com"
|
|
tls_client = TLSService(endpoint, self.access_key_id, self.access_key_secret, self.region)
|
|
server_info = tls_client.get_service_info()
|
|
self.assertEqual("http", server_info.scheme)
|
|
self.assertEqual("tls-cn-beijing.ivolces.com", server_info.host)
|
|
|
|
endpoint = "https://tls-cn-beijing.ivolces.com"
|
|
tls_client = TLSService(endpoint, self.access_key_id, self.access_key_secret, self.region)
|
|
server_info = tls_client.get_service_info()
|
|
self.assertEqual("https", server_info.scheme)
|
|
self.assertEqual("tls-cn-beijing.ivolces.com", server_info.host)
|
|
|
|
endpoint = "tls-cn-beijing.ivolces.com"
|
|
tls_client = TLSService(endpoint, self.access_key_id, self.access_key_secret, self.region)
|
|
server_info = tls_client.get_service_info()
|
|
self.assertEqual("https", server_info.scheme)
|
|
self.assertEqual(endpoint, server_info.host)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |