fix some prompts missing current date (#2122)

This commit is contained in:
Shuchang Zheng
2025-04-10 22:29:47 -07:00
committed by GitHub
parent bdee1add43
commit a0bd4242f7
8 changed files with 39 additions and 2 deletions

View File

@@ -1204,6 +1204,7 @@ class ForgeAgent:
navigation_goal=task.navigation_goal, navigation_goal=task.navigation_goal,
navigation_payload=task.navigation_payload, navigation_payload=task.navigation_payload,
complete_criterion=task.complete_criterion, complete_criterion=task.complete_criterion,
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
# this prompt is critical to our agent so let's use the primary LLM API handler # this prompt is critical to our agent so let's use the primary LLM API handler
@@ -2155,6 +2156,7 @@ class ForgeAgent:
navigation_goal=task.navigation_goal, navigation_goal=task.navigation_goal,
navigation_payload=task.navigation_payload, navigation_payload=task.navigation_payload,
steps=steps_results, steps=steps_results,
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
json_response = await app.LLM_API_HANDLER( json_response = await app.LLM_API_HANDLER(
prompt=prompt, screenshots=screenshots, step=step, prompt_name="summarize-max-steps-reason" prompt=prompt, screenshots=screenshots, step=step, prompt_name="summarize-max-steps-reason"

View File

@@ -50,3 +50,8 @@ HTML elements:
``` ```
{{ elements }} {{ elements }}
``` ```
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -37,3 +37,8 @@ User details:
``` ```
{{ navigation_payload_str }} {{ navigation_payload_str }}
``` ```
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -46,3 +46,8 @@ Popped up elements:
``` ```
{{ popped_up_elements }} {{ popped_up_elements }}
``` ```
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -28,3 +28,8 @@ Elements on the page:
``` ```
{{ elements }} {{ elements }}
``` ```
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -37,3 +37,8 @@ Option list:
``` ```
{{ options }} {{ options }}
``` ```
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -16,3 +16,8 @@ User Details:
Actions Taken In Each Step: Actions Taken In Each Step:
{% for step in steps %}Step {{ step.order }} -- {{ step.actions_result }} {% for step in steps %}Step {{ step.order }} -- {{ step.actions_result }}
{% endfor %} {% endfor %}
Current datetime, ISO format:
```
{{ local_datetime }}
```

View File

@@ -60,6 +60,7 @@ from skyvern.forge.sdk.api.files import (
wait_for_download_finished, wait_for_download_finished,
) )
from skyvern.forge.sdk.api.llm.exceptions import LLMProviderError from skyvern.forge.sdk.api.llm.exceptions import LLMProviderError
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.core.aiohttp_helper import aiohttp_post from skyvern.forge.sdk.core.aiohttp_helper import aiohttp_post
from skyvern.forge.sdk.core.security import generate_skyvern_signature from skyvern.forge.sdk.core.security import generate_skyvern_signature
from skyvern.forge.sdk.core.skyvern_context import ensure_context from skyvern.forge.sdk.core.skyvern_context import ensure_context
@@ -1668,6 +1669,7 @@ async def choose_auto_completion_dropdown(
navigation_goal=task.navigation_goal, navigation_goal=task.navigation_goal,
navigation_payload_str=json.dumps(task.navigation_payload), navigation_payload_str=json.dumps(task.navigation_payload),
elements=html, elements=html,
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
LOG.info( LOG.info(
"Confirm if it's an auto completion dropdown", "Confirm if it's an auto completion dropdown",
@@ -1825,6 +1827,7 @@ async def input_or_auto_complete_input(
current_value=current_value, current_value=current_value,
navigation_goal=task.navigation_goal, navigation_goal=task.navigation_goal,
navigation_payload_str=json.dumps(task.navigation_payload), navigation_payload_str=json.dumps(task.navigation_payload),
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
LOG.info( LOG.info(
@@ -1888,6 +1891,7 @@ async def input_or_auto_complete_input(
navigation_payload_str=json.dumps(task.navigation_payload), navigation_payload_str=json.dumps(task.navigation_payload),
tried_values=json.dumps(tried_values), tried_values=json.dumps(tried_values),
popped_up_elements="".join([json_to_html(element) for element in cleaned_new_elements]), popped_up_elements="".join([json_to_html(element) for element in cleaned_new_elements]),
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
json_respone = await app.SECONDARY_LLM_API_HANDLER( json_respone = await app.SECONDARY_LLM_API_HANDLER(
prompt=prompt, step=step, prompt_name="auto-completion-tweak-value" prompt=prompt, step=step, prompt_name="auto-completion-tweak-value"
@@ -2644,6 +2648,7 @@ async def normal_select(
navigation_goal=task.navigation_goal, navigation_goal=task.navigation_goal,
navigation_payload_str=json.dumps(task.navigation_payload), navigation_payload_str=json.dumps(task.navigation_payload),
options=options_html, options=options_html,
local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
) )
json_response = await app.SELECT_AGENT_LLM_API_HANDLER(prompt=prompt, step=step, prompt_name="custom-select") json_response = await app.SELECT_AGENT_LLM_API_HANDLER(prompt=prompt, step=step, prompt_name="custom-select")