mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-12 02:20:18 +08:00
171dea5e0d
### v1.1.0 - #### Added - 新增 AI 笔记风格选择 - 新增 AI 笔记返回格式选择 - 添加 AI 自定义笔记备注 Prompt - 添加任务失败重试 - 添加全局设置页，可在设置页进行模型设置 - #### Optimize - 优化前端样式，优化用户体验 - 增加生成中间产物，可用于失败后加快生成速度 - #### Fix - 修复视频截图视频过早删除错误
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from app.gpt.base import GPT
|
|
from app.gpt.prompt_builder import generate_base_prompt
|
|
from app.models.gpt_model import GPTSource
|
|
from app.gpt.prompt import BASE_PROMPT, AI_SUM, SCREENSHOT, LINK
|
|
from app.gpt.utils import fix_markdown
|
|
from app.models.transcriber_model import TranscriptSegment
|
|
from datetime import timedelta
|
|
from typing import List
|
|
|
|
class UniversalGPT(GPT):
|
|
def __init__(self, client, model: str, temperature: float = 0.7):
|
|
self.client = client
|
|
self.model = model
|
|
self.temperature = temperature
|
|
self.screenshot = False
|
|
self.screenshot = False
|
|
self.link = False
|
|
|
|
def _format_time(self, seconds: float) -> str:
|
|
return str(timedelta(seconds=int(seconds)))[2:]
|
|
|
|
def _build_segment_text(self, segments: List[TranscriptSegment]) -> str:
|
|
return "\n".join(
|
|
f"{self._format_time(seg.start)} - {seg.text.strip()}"
|
|
for seg in segments
|
|
)
|
|
|
|
def ensure_segments_type(self, segments) -> List[TranscriptSegment]:
|
|
return [TranscriptSegment(**seg) if isinstance(seg, dict) else seg for seg in segments]
|
|
|
|
def create_messages(self, segments: List[TranscriptSegment],**kwargs):
|
|
print("UniversalGPT",kwargs)
|
|
content =generate_base_prompt(
|
|
title=kwargs.get('title'),
|
|
segment_text=self._build_segment_text(segments),
|
|
tags=kwargs.get('tags'),
|
|
_format=kwargs.get('_format'),
|
|
style=kwargs.get('style'),
|
|
extras=kwargs.get('extras')
|
|
)
|
|
|
|
return [{"role": "user", "content": content }]
|
|
|
|
def list_models(self):
|
|
return self.client.models.list()
|
|
def summarize(self, source: GPTSource) -> str:
|
|
self.screenshot = source.screenshot
|
|
self.link = source.link
|
|
source.segment = self.ensure_segments_type(source.segment)
|
|
|
|
messages = self.create_messages(
|
|
source.segment,
|
|
title=source.title,
|
|
tags=source.tags
|
|
,
|
|
_format=source._format,
|
|
style=source.style,
|
|
extras=source.extras
|
|
)
|
|
response = self.client.chat.completions.create(
|
|
model=self.model,
|
|
messages=messages,
|
|
temperature=0.7
|
|
)
|
|
return response.choices[0].message.content.strip()
|
|
|
|
if __name__ == '__main__':
|
|
print('s')
|
|
|