minor code improvements to workflow

activities/tool_activities.py

@@ -196,7 +196,8 @@ class ToolActivities:
         response = client.messages.create(
             model="claude-3-5-sonnet-20241022",
             max_tokens=1024,
-            system=input.context_instructions,
+            system=input.context_instructions + ". The current date is "
+                + get_current_date_human_readable(),
             messages=[
                 {
                     "role": "user",

tools/tool_registry.py

@@ -51,7 +51,7 @@ search_flights_tool = ToolDefinition(
 # 3) Define the CreateInvoice tool
 create_invoice_tool = ToolDefinition(
     name="CreateInvoice",
-    description="Generate an invoice for the items described for the amount provided",
+    description="Generate an invoice for the items described for the amount provided. Returns URL to invoice.",
     arguments=[
         ToolArgument(
             name="amount",

workflows/tool_workflow.py

@@ -4,6 +4,7 @@ from typing import Dict, Any, Union, List, Optional, Deque, TypedDict
 
 from temporalio.common import RetryPolicy
 from temporalio import workflow
+from temporalio.exceptions import ActivityError
 
 from models.data_types import ConversationHistory, NextStep, ValidationInput
 
@@ -44,15 +45,20 @@ class ToolWorkflow:
         """Execute a tool after confirmation and handle its result."""
         workflow.logger.info(f"Confirmed. Proceeding with tool: {current_tool}")
         
+        try:
             dynamic_result = await workflow.execute_activity(
                 current_tool,
                 tool_data["args"],
                 schedule_to_close_timeout=TOOL_ACTIVITY_TIMEOUT,
+                retry_policy=RetryPolicy(maximum_attempts=3)
             )
             dynamic_result["tool"] = current_tool
-        self.add_message(
-            "tool_result", {"tool": current_tool, "result": dynamic_result}
-        )
+            self.tool_results.append(dynamic_result)
+        except ActivityError as e:
+            workflow.logger.error(f"Tool execution failed: {str(e)}")
+            dynamic_result = {"error": str(e), "tool": current_tool}
+        
+        self.add_message("tool_result", dynamic_result)
 
         self.prompt_queue.append(
             f"### The '{current_tool}' tool completed successfully with {dynamic_result}. "
@@ -165,9 +171,11 @@ class ToolWorkflow:
                     )
 
                     if not validation_result.validationResult:
-                        # Handle validation failure
+                        workflow.logger.warning(
+                            f"Prompt validation failed: {validation_result.validationFailedReason}"
+                        )
                         self.add_message("agent", validation_result.validationFailedReason)
-                        continue  # Skip to the next iteration
+                        continue
 
                 # Proceed with generating the context and prompt
 
@@ -226,6 +234,7 @@ class ToolWorkflow:
     @workflow.signal
     async def confirm(self) -> None:
         """Signal handler for user confirmation of tool execution."""
+        workflow.logger.info("Received user confirmation")
         self.confirm = True
 
     @workflow.query
@@ -285,6 +294,12 @@ class ToolWorkflow:
             actor: The entity that generated the message (e.g., "user", "agent")
             response: The message content, either as a string or structured data
         """
+        if isinstance(response, dict):
+            response_str = str(response)
+            workflow.logger.debug(f"Adding {actor} message: {response_str[:100]}...")
+        else:
+            workflow.logger.debug(f"Adding {actor} message: {response[:100]}...")
+            
         self.conversation_history["messages"].append(
             {"actor": actor, "response": response}
         )