Merge pull request #259 from vvincent1234/fix/dr-bugs

warmshao · web-flow · commit 3a459a91ad4e · 2025-02-09T11:28:56.000+08:00
fix prompt
diff --git a/src/agent/custom_prompts.py b/src/agent/custom_prompts.py
@@ -158,7 +158,7 @@ def get_user_message(self) -> HumanMessage:
             step_info_description = ''
             
         time_str = datetime.now().strftime("%Y-%m-%d %H:%M")
-        step_info_description += "Current date and time: {time_str}"
+        step_info_description += f"Current date and time: {time_str}"
 
         elements_text = self.state.element_tree.clickable_elements_to_string(include_attributes=self.include_attributes)
 
diff --git a/src/utils/llm.py b/src/utils/llm.py
@@ -68,7 +68,7 @@ async def ainvoke(
         
         response = self.client.chat.completions.create(
             model=self.model_name,
-            messages=messages
+            messages=message_history
         )
 
         reasoning_content = response.choices[0].message.reasoning_content

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,7 @@ async def ainvoke(`
`68`	`68`
`69`	`69`	`response = self.client.chat.completions.create(`
`70`	`70`	`model=self.model_name,`
`71`		`- messages=messages`
	`71`	`+ messages=message_history`
`72`	`72`	`)`
`73`	`73`
`74`	`74`	`reasoning_content = response.choices[0].message.reasoning_content`