增加rpm,tpm限制
This commit is contained in:
@@ -3,14 +3,14 @@
|
||||
import asyncio
|
||||
from dataclasses import dataclass
|
||||
from io import BytesIO
|
||||
from typing import Self, Literal, List, Dict, Any, Tuple, Optional
|
||||
from typing import Self, Literal, List, Dict, Any, Tuple
|
||||
|
||||
from pptx import Presentation
|
||||
from pptx.enum.dml import MSO_COLOR_TYPE
|
||||
from pptx.enum.shapes import MSO_SHAPE_TYPE
|
||||
from pptx.enum.text import MSO_AUTO_SIZE
|
||||
from pptx.enum.dml import MSO_COLOR_TYPE
|
||||
from pptx.text.text import _Paragraph, TextFrame
|
||||
from pptx.oxml.ns import qn
|
||||
from pptx.text.text import _Paragraph, TextFrame
|
||||
|
||||
from docutranslate.agents.segments_agent import SegmentsTranslateAgentConfig, SegmentsTranslateAgent
|
||||
from docutranslate.ir.document import Document
|
||||
@@ -46,7 +46,9 @@ class PPTXTranslator(AiTranslator):
|
||||
api_key=config.api_key, model_id=config.model_id, temperature=config.temperature,
|
||||
thinking=config.thinking, concurrent=config.concurrent, timeout=config.timeout,
|
||||
logger=self.logger, glossary_dict=config.glossary_dict, retry=config.retry,
|
||||
system_proxy_enable=config.system_proxy_enable, force_json=config.force_json
|
||||
system_proxy_enable=config.system_proxy_enable, force_json=config.force_json,
|
||||
rpm=config.rpm,
|
||||
tpm=config.tpm
|
||||
)
|
||||
self.translate_agent = SegmentsTranslateAgent(agent_config)
|
||||
self.insert_mode = config.insert_mode
|
||||
@@ -335,4 +337,4 @@ class PPTXTranslator(AiTranslator):
|
||||
translated = await self.translate_agent.send_segments_async(originals,
|
||||
self.chunk_size) if self.translate_agent else originals
|
||||
document.content = await asyncio.to_thread(self._after_translate, prs, elements, translated, originals)
|
||||
return self
|
||||
return self
|
||||
|
||||
Reference in New Issue
Block a user