diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 43bf8dc..3dc4da5 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -6,9 +6,7 @@
-
-
@@ -109,29 +107,6 @@
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
@@ -155,6 +130,29 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
@@ -279,9 +277,9 @@
+
-
@@ -309,7 +307,8 @@
-
+
+
@@ -318,15 +317,16 @@
-
-
+
+
+
+
-
\ No newline at end of file
diff --git a/docutranslate/utils/agent_utils.py b/docutranslate/utils/agent_utils.py
index d86d19d..8edeed0 100644
--- a/docutranslate/utils/agent_utils.py
+++ b/docutranslate/utils/agent_utils.py
@@ -24,7 +24,7 @@ class Agent:
data = {
"model": self.model_id,
"messages": [
- {"role": "system", "content": "所有回复必须严格以<>开头,包括简短回答。示例:\n<>这是示例回答\n"+system_prompt},
+ {"role": "system", "content": "回复必须严格以BEGIN>>开头,包括简短回答。\n示例:BEGIN>>这是示例回答\n"+system_prompt},
{"role": "user", "content": prompt}
],
"temperature": temperature,
@@ -54,10 +54,10 @@ class Agent:
)
response.raise_for_status()
result=response.json()["choices"][0]["message"]["content"]
- pattern=r"<>(.*)"
+ pattern=r"BEGIN>>(.*)"
match= re.search(pattern,result, re.DOTALL)
if match is None:
- print("检测开头<失败")
+ print("检测开头`BEGIN>>`失败")
else:
result=match.group(1)
return result