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助手体验！
24 lines
784 B
Python
Executable File
24 lines
784 B
Python
Executable File
from io import BytesIO
|
|
|
|
import avro.schema
|
|
from avro.io import DatumReader, BinaryDecoder
|
|
|
|
RecordSchema: str = "com.bytedance.dts.subscribe.avro.Record"
|
|
|
|
|
|
class RecordDeserializer:
|
|
# schema_file: path of schema file
|
|
def __init__(self, schema_file: str):
|
|
s = avro.schema.parse(open(schema_file, "rb").read())
|
|
self.all_schema = s
|
|
for schema in self.all_schema.schemas:
|
|
if isinstance(schema, avro.schema.RecordSchema) and schema.fullname == RecordSchema:
|
|
self.record_schema = schema
|
|
break
|
|
|
|
def deserialize(self, data: bytes) -> dict:
|
|
bio = BytesIO(initial_bytes=data)
|
|
decoder = BinaryDecoder(bio)
|
|
res = DatumReader(writers_schema=self.record_schema).read(decoder)
|
|
return res
|