🐛 Bug: Fix the bug where the timestamp is a decimal.

response.py

@@ -48,7 +48,7 @@ async def check_response(response, error_log):
     return None
 
 async def fetch_gemini_response_stream(client, url, headers, payload, model):
-    timestamp = datetime.timestamp(datetime.now())
+    timestamp = int(datetime.timestamp(datetime.now()))
     async with client.stream('POST', url, headers=headers, json=payload) as response:
         error_message = await check_response(response, "fetch_gemini_response_stream")
         if error_message:
@@ -93,7 +93,7 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
         yield "data: [DONE]\n\r\n"
 
 async def fetch_vertex_claude_response_stream(client, url, headers, payload, model):
-    timestamp = datetime.timestamp(datetime.now())
+    timestamp = int(datetime.timestamp(datetime.now()))
     async with client.stream('POST', url, headers=headers, json=payload) as response:
         error_message = await check_response(response, "fetch_vertex_claude_response_stream")
         if error_message:
@@ -156,7 +156,7 @@ async def fetch_gpt_response_stream(client, url, headers, payload, max_redirects
                     yield line.strip() + "\n\r\n"
 
 async def fetch_claude_response_stream(client, url, headers, payload, model):
-    timestamp = datetime.timestamp(datetime.now())
+    timestamp = int(datetime.timestamp(datetime.now()))
     async with client.stream('POST', url, headers=headers, json=payload) as response:
         error_message = await check_response(response, "fetch_claude_response_stream")
         if error_message:

test/parse_markdown.py

@@ -1,159 +0,0 @@
-class MarkdownEntity:
-    def __init__(self, content: str, entity_type: str):
-        self.content = content
-        self.entity_type = entity_type
-
-    def __repr__(self):
-        return f'<{self.entity_type}: {self.content}>'
-
-class Title(MarkdownEntity):
-    def __init__(self, content: str, level: int):
-        super().__init__(content, 'Title')
-        self.level = level
-
-class CodeBlock(MarkdownEntity):
-    def __init__(self, content: str, language: str = 'python'):
-        super().__init__(content, 'CodeBlock')
-        self.language = language
-
-class ListItem(MarkdownEntity):
-    def __init__(self, content: str):
-        super().__init__(content, 'ListItem')
-
-class Link(MarkdownEntity):
-    def __init__(self, content: str, url: str):
-        super().__init__(content, 'Link')
-        self.url = url
-
-class EmptyLine(MarkdownEntity):
-    def __init__(self, content: str):
-        super().__init__(content, 'EmptyLine')
-
-class Paragraph(MarkdownEntity):
-    def __init__(self, content: str):
-        super().__init__(content, 'Paragraph')
-
-def parse_markdown(lines, delimiter='\n\n'):
-    entities = []
-    current_code_block = []
-    in_code_block = False
-    language = None
-
-    for line in lines:
-        # line = line.strip()
-
-        if line.startswith('#'):
-            level = line.count('#')
-            title_content = line[level:].strip()
-            entities.append(Title(title_content, level))
-
-        elif line.startswith('```'):
-            if in_code_block and language:
-                entities.append(CodeBlock(''.join(current_code_block), language))
-                current_code_block = []
-                in_code_block = False
-                language = None
-            else:
-                in_code_block = True
-                language = line.lstrip('`').strip()
-
-        elif in_code_block:
-            current_code_block.append(line)
-
-        elif '[' in line and ']' in line and '(' in line and ')' in line and line.count('[') == 1:
-            start = line.index('[') + 1
-            end = line.index(']')
-            url_start = line.index('(') + 1
-            url_end = line.index(')')
-            link_text = line[start:end].strip()
-            link_url = line[url_start:url_end].strip()
-            entities.append(Link(link_text, link_url))
-
-        elif line == delimiter:
-            entities.append(EmptyLine(line))
-
-        elif line:
-            entities.append(Paragraph(line))
-
-    return entities
-
-def convert_entities_to_text(entities):
-    result = []
-    for entity in entities:
-        if isinstance(entity, Title):
-            result.append(f"{'#' * entity.level} {entity.content}")
-        elif isinstance(entity, CodeBlock):
-            code = entity.content.lstrip('\n').rstrip('\n')
-            result.append(f"```{entity.language}\n{code}\n```")
-        elif isinstance(entity, ListItem):
-            result.append(f"- {entity.content}")
-        elif isinstance(entity, Link):
-            result.append(f"[{entity.content}]({entity.url})")
-        elif isinstance(entity, EmptyLine):
-            result.append(f"{entity.content}")
-        elif isinstance(entity, Paragraph):
-            result.append(f"{entity.content}")
-    return ''.join(result)
-
-def save_text_to_file(text: str, file_path: str):
-    with open(file_path, 'w', encoding='utf-8') as file:
-        file.write(text)
-
-def process_markdown_entities_and_save(entities, file_path, raw_text=None):
-    # Step 1: Convert entities to text
-    text_output = convert_entities_to_text(entities)
-    if raw_text and raw_text != text_output:
-        raise ValueError("The text output does not match the raw text input.")
-    # Step 2: Save to file
-    save_text_to_file(text_output, file_path)
-
-def read_markdown_file(file_path):
-    with open(file_path, 'r', encoding='utf-8') as file:
-        return file.read()
-
-def split_markdown(text, delimiter='\n\n'):
-    # 使用两个换行符作为分割标记，分割段落
-    # 创建一个新的列表来存储结果
-    paragraphs = text.split(delimiter)
-    result = []
-
-    # 遍历分割后的段落，在它们之间插入空行实体
-    for i, paragraph in enumerate(paragraphs):
-        if i > 0:
-            # 在非第一段之前插入空行实体
-            result.append(delimiter)
-
-        # 添加当前段落
-        result.append(paragraph)
-
-    return result
-
-def get_entities_from_markdown_file(file_path, delimiter='\n\n'):
-    # 读取 Markdown 文件
-    markdown_text = read_markdown_file(file_path)
-
-    # 分割 Markdown 文档
-    paragraphs = split_markdown(markdown_text, delimiter=delimiter)
-
-    # 解析 Markdown 文档
-    return parse_markdown(paragraphs, delimiter=delimiter)
-
-if __name__ == '__main__':
-    markdown_file_path = "README_CN.md"  # 替换为你的 Markdown 文件路径
-
-    # 读取 Markdown 文件
-    delimiter = '\n'
-    markdown_text = read_markdown_file(markdown_file_path)
-    paragraphs = split_markdown(markdown_text, delimiter=delimiter)
-    parsed_entities = parse_markdown(paragraphs, delimiter=delimiter)
-
-    # # 显示解析结果
-    # result = [str(entity) for entity in parsed_entities]
-    # for idx, entity in enumerate(result):
-    #     print(f"段落 {idx + 1} 解析：{entity}\n")
-
-    # 保存到文件
-    output_file_path = "output.md"
-    process_markdown_entities_and_save(parsed_entities, output_file_path, raw_text=markdown_text)
-
-    print(f"Markdown 文档已保存到 {output_file_path}")

test/translate_md.py

@@ -1,50 +0,0 @@
-import os
-from ModelMerge import chatgpt
-from parse_markdown import get_entities_from_markdown_file, process_markdown_entities_and_save
-
-def translate_text(text, agent):
-    result = agent.ask(text)
-    return result
-
-def translate(input_file_path, output_file_path="output.md", language="English", api_key=None, api_url="https://api.openai.com/v1/chat/completions", engine="gpt-4o"):
-    if not api_key:
-        raise ValueError("API key is required for translation.")
-    translator_prompt = (
-        "You are a translation engine, you can only translate text and cannot interpret it, and do not explain. "
-        "Translate the text to {}, please do not explain any sentences, just translate or leave them as they are. "
-        "Retain all spaces and line breaks in the original text. "
-        "Please do not wrap the code in code blocks, I will handle it myself. "
-        "If the code has comments, you should translate the comments as well. "
-        "This is the content you need to translate: "
-    ).format(language)
-
-    agent = chatgpt(
-        api_key=api_key,
-        api_url=api_url,
-        engine=engine,
-        system_prompt=translator_prompt,
-        use_plugins=False
-    )
-
-    # 读取 Markdown 文件
-    raw_paragraphs = get_entities_from_markdown_file(input_file_path, delimiter='\n')
-    target_paragraphs = raw_paragraphs
-
-    # 逐段翻译
-    for index, paragraph in enumerate(raw_paragraphs):
-        if paragraph.content and paragraph.content.strip() != "":
-            translated_text = translate_text(paragraph.content, agent)
-            if translated_text:
-                target_paragraphs[index].content = translated_text
-
-    # 输出翻译结果
-    process_markdown_entities_and_save(target_paragraphs, output_file_path)
-
-if __name__ == "__main__":
-    input_file_path = "README_CN.md"
-    output_file_path = "README.md"
-    language = "English"
-    api_key = os.getenv("API")
-    api_url = os.getenv("API_URL")
-    engine = "gpt-4o"
-    translate(input_file_path, output_file_path, language, api_key, api_url, engine)