Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加提示词

Browse Source
This commit is contained in:
xunbu
2025-08-05 17:18:59 +08:00
parent a5b2b99ff8
commit 0c1b73bfb8
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docutranslate/agents/agent.py
View File

@@ -167,7 +167,8 @@ class Agent:
) -> list[str]:
max_concurrent = self.max_concurrent if max_concurrent is None else max_concurrent
total = len(prompts)
self.logger.info(f"收到{total}个片段，并发请求数:{max_concurrent}")
self.logger.info(f"base-url:{self.baseurl},model-id:{self.model_id}")
self.logger.info(f"预计发送{total}个请求，并发请求数:{max_concurrent}")
count = 0
semaphore = asyncio.Semaphore(max_concurrent)
tasks = []
@@ -237,8 +238,8 @@ class Agent:
prompts: list[str],
system_prompt: str | None = None,
) -> list[str]:
self.logger.info(f"收到{len(prompts)}个片段，并发请求数:{self.max_concurrent}")
self.logger.info(f"base-url:{self.baseurl},model-id:{self.model_id}")
self.logger.info(f"预计发送{len(prompts)}个请求，并发请求数:{self.max_concurrent}")
system_prompts = [system_prompt] * len(prompts)
counts = [PromptsCounter(len(prompts), self.logger)] * len(prompts)
output_list = []