mirror/temporal-ai-agent
1
0
Fork 0
mirror of https://github.com/temporal-community/temporal-ai-agent.git synced 2026-03-15 14:08:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

start prompt comes from back-end

Browse Source
This commit is contained in:
Steve Androulakis
2025-02-03 13:32:33 -08:00
parent 2a2383bb71
commit 36d4bd5889
7 changed files with 64 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
workflows/tool_workflow.py
View File

@@ -150,24 +150,24 @@ class ToolWorkflow:
prompt = self.prompt_queue.popleft()
if not prompt.startswith("###"):
self.add_message("user", prompt)
# Validate the prompt before proceeding
validation_input = ValidationInput(
prompt=prompt,
conversation_history=self.conversation_history,
agent_goal=agent_goal,
)
validation_result = await workflow.execute_activity(
ToolActivities.validate_llm_prompt,
args=[validation_input],
schedule_to_close_timeout=LLM_ACTIVITY_TIMEOUT,
retry_policy=RetryPolicy(initial_interval=timedelta(seconds=5)),
)
# Validate the prompt before proceeding
validation_input = ValidationInput(
prompt=prompt,
conversation_history=self.conversation_history,
agent_goal=agent_goal,
)
validation_result = await workflow.execute_activity(
ToolActivities.validate_llm_prompt,
args=[validation_input],
schedule_to_close_timeout=LLM_ACTIVITY_TIMEOUT,
retry_policy=RetryPolicy(initial_interval=timedelta(seconds=5)),
)
if not validation_result.validationResult:
# Handle validation failure
self.add_message("agent", validation_result.validationFailedReason)
continue # Skip to the next iteration
if not validation_result.validationResult:
# Handle validation failure
self.add_message("agent", validation_result.validationFailedReason)
continue # Skip to the next iteration
# Proceed with generating the context and prompt