diff --git a/.env.example b/.env.example index ef6bf09..844fd27 100644 --- a/.env.example +++ b/.env.example @@ -1,5 +1,6 @@ RAPIDAPI_KEY=9df2cb5... -RAPIDAPI_HOST=sky-scrapper.p.rapidapi.com +RAPIDAPI_HOST_FLIGHTS=sky-scrapper.p.rapidapi.com #For travel flight information tool +RAPIDAPI_HOST_PACKAGE=trackingpackage.p.rapidapi.com #For eCommerce order status package tracking tool FOOTBALL_DATA_API_KEY=.... STRIPE_API_KEY=sk_test_51J... @@ -42,7 +43,7 @@ AGENT_GOAL=goal_choose_agent_type # for multi-goal start #Choose which category(ies) of goals you want to be listed by the Agent Goal picker if enabled above # - options are system (always included), hr, travel, or all. -GOAL_CATEGORIES=hr,travel-flights,travel-trains,fin # default is all +GOAL_CATEGORIES=hr,travel-flights,travel-trains,fin,ecommerce # default is all #GOAL_CATEGORIES=travel-flights # Set if the workflow should wait for the user to click a confirm button (and if the UI should show the confirm button and tool args) diff --git a/.gitignore b/.gitignore index c2d2dde..24f6b6f 100644 --- a/.gitignore +++ b/.gitignore @@ -32,5 +32,4 @@ coverage.xml .idea/ .env -*.env .env* diff --git a/README.md b/README.md index 91c8c1a..8a97fc7 100644 --- a/README.md +++ b/README.md @@ -48,13 +48,4 @@ See [the guide to adding goals and tools](./adding-goals-and-tools.md) for more ## For Temporal SAs Check out the [slides](https://docs.google.com/presentation/d/1wUFY4v17vrtv8llreKEBDPLRtZte3FixxBUn0uWy5NU/edit#slide=id.g3333e5deaa9_0_0) here and the enablement guide here (TODO). -## Tests -Running the tests requires `poe` and `pytest_asyncio` to be installed. - - python -m pip install poethepoet - python -m pip install pytest_asyncio - -Once you have `poe` and `pytest_asyncio` installed you can run: - - poe test diff --git a/pyproject.toml b/pyproject.toml index 4ab7bec..6f3c4f9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [tool.poetry] -name = "temporal-AI-agent" +name = "temporal_AI_agent" version = "0.1.0" description = "Temporal AI Agent" license = "MIT" @@ -13,7 +13,7 @@ packages = [ ] [tool.poetry.urls] -"Bug Tracker" = "https://github.com/temporalio/samples-python/issues" +"Bug Tracker" = "https://github.com/temporal-community/temporal-ai-agent/issues" [tool.poe.tasks] format = [{cmd = "black ."}, {cmd = "isort ."}] @@ -43,9 +43,16 @@ gtfs-kit = "^10.1.1" [tool.poetry.group.dev.dependencies] pytest = "^7.3" +pytest-asyncio = "^0.18.3" black = "^23.7" isort = "^5.12" [build-system] requires = ["poetry-core>=1.4.0"] -build-backend = "poetry.core.masonry.api" \ No newline at end of file +build-backend = "poetry.core.masonry.api" + +[tool.pytest.ini_options] +asyncio_mode = "auto" +log_cli = true +log_cli_level = "INFO" +log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)" \ No newline at end of file diff --git a/setup.md b/setup.md index a94f989..6a41558 100644 --- a/setup.md +++ b/setup.md @@ -204,6 +204,7 @@ FIN_START_REAL_WORKFLOW=FALSE #set this to true to start a real workflow By default it will _not_ make a real workflow, it'll just fake it. If you get the worker running and want to start a workflow, in your [.env](./.env): ```bash FIN_START_REAL_WORKFLOW=FALSE #set this to true to start a real workflow +``` #### Goals: HR/PTO diff --git a/todo.md b/todo.md index f67239e..00ad1d0 100644 --- a/todo.md +++ b/todo.md @@ -10,11 +10,19 @@ [ ] financial advise - args being freeform customer input about their financial situation, goals [ ] tool is maybe a new tool asking the LLM to advise +[ ] for demo simulate failure - add utilities/simulated failures from pipeline demo
+ +[ ] ecommerce goals
+- [ ] add to docs
+- [ ] decide about api key names with Laine
+ [ ] LLM failure->autoswitch:
- detect failure in the activity using failurecount
- activity switches to secondary LLM defined in .env - activity reports switch to workflow +[ ] for demo simulate failure - add utilities/simulated failures from pipeline demo
+ [ ] ask the ai agent how it did at the end of the conversation, was it efficient? successful? insert a search attribute to document that before return
- Insight into the agent’s performance
[ ] non-retry the api key error - "Invalid API Key provided: sk_test_**J..." and "AuthenticationError"
diff --git a/tools/__init__.py b/tools/__init__.py index e7df76c..6b7ab44 100644 --- a/tools/__init__.py +++ b/tools/__init__.py @@ -18,6 +18,10 @@ from .fin.get_account_balances import get_account_balance from .fin.move_money import move_money from .fin.submit_loan_application import submit_loan_application +from .ecommerce.get_order import get_order +from .ecommerce.track_package import track_package +from .ecommerce.list_orders import list_orders + from .give_hint import give_hint from .guess_location import guess_location @@ -57,6 +61,12 @@ def get_handler(tool_name: str): return move_money if tool_name == "FinCheckAccountSubmitLoanApproval": return submit_loan_application + if tool_name == "GetOrder": + return get_order + if tool_name == "TrackPackage": + return track_package + if tool_name == "ListOrders": + return list_orders if tool_name == "GiveHint": return give_hint if tool_name == "GuessLocation": diff --git a/tools/data/customer_order_data.json b/tools/data/customer_order_data.json new file mode 100644 index 0000000..8281351 --- /dev/null +++ b/tools/data/customer_order_data.json @@ -0,0 +1,81 @@ +{ + "orders": [ + { + "id": "100", + "summary": "Lawyer Books", + "email": "matt.murdock@nelsonmurdock.com", + "status": "cancelled", + "order_date": "2025-03-30", + "last_update": "2025-04-01" + }, + { + "id": "101", + "summary": "Bonking Sticks", + "email": "matt.murdock@nelsonmurdock.com", + "status": "paid", + "order_date": "2025-04-01", + "last_order_update": "2025-04-01" + }, + { + "id": "102", + "summary": "Red Sunglasses", + "email": "matt.murdock@nelsonmurdock.com", + "status": "shipped", + "order_date": "2025-04-01", + "last_order_update": "2025-04-01", + "tracking_id": "1Z111111" + }, + { + "id": "200", + "summary": "Paper", + "email": "foggy.nelson@nelsonmurdock.com", + "status": "shipped", + "order_date": "2025-04-03", + "last_update": "2025-04-06", + "tracking_id": "991111" + }, + { + "id": "300", + "summary": "Chemistry Books", + "email": "heisenberg@blue-meth.com", + "status": "shipped", + "order_date": "2025-03-30", + "last_update": "2025-04-06", + "tracking_id": "991111" + }, + { + "id": "301", + "summary": "Book: Being a Cool Bro", + "email": "heisenberg@blue-meth.com", + "status": "cancelled", + "order_date": "2025-04-01", + "last_update": "2025-04-02" + }, + { + "id": "302", + "summary": "Black Hat", + "email": "heisenberg@blue-meth.com", + "status": "delivered", + "order_date": "2025-04-01", + "last_update": "2025-04-06", + "tracking_id": "1Z11111" + }, + { + "id": "400", + "summary": "Giant Graphic Hoodie", + "email": "jessenotpinkman@blue-meth.com", + "status": "shipped", + "order_date": "2025-04-03", + "last_update": "2025-04-09", + "tracking_id": "1Z111111" + }, + { + "id": "401", + "summary": "Giant Pants", + "email": "jessenotpinkman@blue-meth.com", + "status": "processing", + "order_date": "2025-04-03", + "last_update": "2025-04-09" + } + ] +} \ No newline at end of file diff --git a/tools/ecommerce/get_order.py b/tools/ecommerce/get_order.py new file mode 100644 index 0000000..5634b7a --- /dev/null +++ b/tools/ecommerce/get_order.py @@ -0,0 +1,27 @@ +from pathlib import Path +import json + +# this is made to demonstrate functionality but it could just as durably be an API call +# called as part of a temporal activity with automatic retries +def get_order(args: dict) -> dict: + + order_id = args.get("order_id") + + file_path = Path(__file__).resolve().parent.parent / "data" / "customer_order_data.json" + if not file_path.exists(): + return {"error": "Data file not found."} + + with open(file_path, "r") as file: + data = json.load(file) + order_list = data["orders"] + + for order in order_list: + if order["id"] == order_id: + return order +# if order["status"] == "shipped": + # return{"status": order["status"], "tracking_id": order["tracking_id"]} + # else: + # return{"status": order["status"]} + + return_msg = "Order " + order_id + " not found." + return {"error": return_msg} \ No newline at end of file diff --git a/tools/ecommerce/list_orders.py b/tools/ecommerce/list_orders.py new file mode 100644 index 0000000..740aa20 --- /dev/null +++ b/tools/ecommerce/list_orders.py @@ -0,0 +1,30 @@ +from pathlib import Path +import json + +def sorting(e): + return e['order_date'] + +def list_orders(args: dict) -> dict: + + email_address = args.get("email_address") + + file_path = Path(__file__).resolve().parent.parent / "data" / "customer_order_data.json" + if not file_path.exists(): + return {"error": "Data file not found."} + + with open(file_path, "r") as file: + data = json.load(file) + order_list = data["orders"] + + rtn_order_list = [] + for order in order_list: + if order["email"] == email_address: + rtn_order_list.append(order) + + if len(rtn_order_list) > 0: + rtn_order_list.sort(key=sorting) + return {"orders": rtn_order_list} + else: + return_msg = "No orders for customer " + email_address + " found." + return {"error": return_msg} + diff --git a/tools/ecommerce/track_package.py b/tools/ecommerce/track_package.py new file mode 100644 index 0000000..55b14fd --- /dev/null +++ b/tools/ecommerce/track_package.py @@ -0,0 +1,110 @@ +import http +import os +import json + +def track_package_faked(args: dict) -> dict: + + tracking_id = args.get("tracking_id") + + #return_msg = "Account not found with email address " + email + " or account ID: " + account_id + return {"tracking_info": "delivered, probably"} + +'''Format of response: +{ + "TrackingNumber": "", + "Delivered": false, + "Carrier": "USPS", + "ServiceType": "USPS Ground Advantage™", + "PickupDate": "", + "ScheduledDeliveryDate": "April 14, 2025", + "ScheduledDeliveryDateInDateTimeFromat": "2025-04-14T00:00:00", + "StatusCode": "In Transit from Origin Processing", + "Status": "Departed Post Office", + "StatusSummary": "Your item has left our acceptance facility and is in transit to a sorting facility on April 10, 2025 at 7:06 am in IRON RIDGE, WI 53035.", + "Message": "", + "DeliveredDateTime": "", + "DeliveredDateTimeInDateTimeFormat": null, + "SignatureName": "", + "DestinationCity": "CITY", + "DestinationState": "ST", + "DestinationZip": "12345", + "DestinationCountry": null, + "EventDate": "2025-04-10T07:06:00", + "TrackingDetails": [ + { + "EventDateTime": "April 10, 2025 7:06 am", + "Event": "Departed Post Office", + "EventAddress": "IRON RIDGE WI 53035", + "State": "WI", + "City": "IRON RIDGE", + "Zip": "53035", + "EventDateTimeInDateTimeFormat": "2025-04-10T07:06:00" + }, + { + "EventDateTime": "April 9, 2025 11:29 am", + "Event": "USPS picked up item", + "EventAddress": "IRON RIDGE WI 53035", + "State": "WI", + "City": "IRON RIDGE", + "Zip": "53035", + "EventDateTimeInDateTimeFormat": "2025-04-09T11:29:00" + }, + { + "EventDateTime": "April 7, 2025 6:29 am", + "Event": "Shipping Label Created, USPS Awaiting Item", + "EventAddress": "IRON RIDGE WI 53035", + "State": "WI", + "City": "IRON RIDGE", + "Zip": "53035", + "EventDateTimeInDateTimeFormat": "2025-04-07T06:29:00" + } + ] +} +''' +def track_package(args: dict) -> dict: + + tracking_id = args.get("tracking_id") + + api_key = os.getenv("RAPIDAPI_KEY") + api_host = os.getenv("RAPIDAPI_HOST_PACKAGE", "trackingpackage.p.rapidapi.com") + + conn = http.client.HTTPSConnection(api_host) + headers = { + "x-rapidapi-key": api_key, + "x-rapidapi-host": api_host, + "Authorization": "Basic Ym9sZGNoYXQ6TGZYfm0zY2d1QzkuKz9SLw==", + } + + path = f"/TrackingPackage?trackingNumber={tracking_id}" + + conn.request("GET", path, headers=headers) + res = conn.getresponse() + data = res.read() + data_decoded = data.decode("utf-8") + conn.close() + + try: + json_data = json.loads(data_decoded) + except json.JSONDecodeError: + return {"error": "Invalid JSON response"} + + scheduled_delivery_date = json_data["ScheduledDeliveryDate"] + carrier = json_data["Carrier"] + status_summary = json_data["StatusSummary"] + tracking_details = json_data.get("TrackingDetails", []) + last_tracking_update = "" + if tracking_details and tracking_details is not None and tracking_details[0] is not None: + last_tracking_update = tracking_details[0]["EventDateTimeInDateTimeFormat"] + tracking_link = "" + if carrier == "USPS": + tracking_link = f"https://tools.usps.com/go/TrackConfirmAction?qtc_tLabels1={tracking_id}" + elif carrier == "UPS": + tracking_link = f"https://www.ups.com/track?track=yes&trackNums={tracking_id}" + + return { + "scheduled_delivery_date": scheduled_delivery_date, + "carrier": carrier, + "status_summary": status_summary, + "tracking_link": tracking_link, + "last_tracking_update": last_tracking_update + } \ No newline at end of file diff --git a/tools/goal_registry.py b/tools/goal_registry.py index 95c329f..c22eb21 100644 --- a/tools/goal_registry.py +++ b/tools/goal_registry.py @@ -95,6 +95,7 @@ goal_pirate_treasure = AgentGoal( ), ) +# ----- Travel Goals --- goal_match_train_invoice = AgentGoal( id = "goal_match_train_invoice", category_tag="travel-trains", @@ -180,6 +181,7 @@ goal_event_flight_invoice = AgentGoal( ), ) +# ----- HR Goals --- # This goal uses the data/employee_pto_data.json file as dummy data. goal_hr_schedule_pto = AgentGoal( id = "goal_hr_schedule_pto", @@ -268,6 +270,7 @@ goal_hr_check_paycheck_bank_integration_status = AgentGoal( ), ) +# ----- FinServ Goals --- # this tool checks account balances, and uses ./data/customer_account_data.json as dummy data goal_fin_check_account_balances = AgentGoal( id = "goal_fin_check_account_balances", @@ -368,6 +371,79 @@ goal_fin_loan_application = AgentGoal( "user_confirmed_tool_run: ", "tool_result: { 'status': submitted, 'detailed_status': loan application is submitted and initial validation is complete, 'confirmation id': 333421, 'next_step': You'll receive a confirmation for final approval in three business days }", "agent: I have submitted your loan application process and the initial validation is successful. Your application ID is 333421. You'll receive a notification for final approval from us in three business days. " + +# ----- E-Commerce Goals --- +#todo: add goal to list all orders for last X amount of time? +# this tool checks account balances, and uses ./data/customer_account_data.json as dummy data +goal_ecomm_order_status = AgentGoal( + id = "goal_ecomm_order_status", + category_tag="ecommerce", + agent_name="Check Order Status", + agent_friendly_description="Check the status of your order.", + tools=[ + tool_registry.ecomm_get_order, + tool_registry.ecomm_track_package, + ], + description="The user wants to learn the status of a specific order. If the status is 'shipped' or 'delivered', they might want to get the package tracking information. To assist with that goal, help the user gather args for these tools in order: " + "1. GetOrder: get information about an order" + "2. TrackPackage: provide tracking information for the package. This tool is optional and should only be offered if the status is 'shipped' OR 'delivered' - otherwise, skip this tool and do not mention it to the user.", + starter_prompt=starter_prompt_generic, + example_conversation_history="\n ".join( + [ + "user: I'd like to know the status of my order", + "agent: Sure! I can help you out with that. May I have your email address or order number?", + "user: email is bob.johnson@emailzzz.com ", + "user_confirmed_tool_run: ", + "tool_result: { 'id': '102', 'summary': 'Red Sunglasses', 'email': 'matt.murdock@nelsonmurdock.com', 'status': 'shipped', 'order_date': '2025-04-01', 'last_order_update': '2025-04-06', 'tracking_id': '039813852990618' }", + "agent: Your order 'Red Sunglasses,' placed April 1, 2025, was shipped on April 6, 2025. Would you like to see the tracking inforation?", + "user: Yes", + "user_confirmed_tool_run: ", + "tool_result: { 'scheduled_delivery_date': 'April 30, 2025', 'carrier': 'USPS', 'status_summary': 'Your item has left our acceptance facility and is in transit to a sorting facility on April 10, 2025 at 7:06 am in IRON RIDGE, WI 53035.', 'tracking_link': 'https://tools.usps.com/go/TrackConfirmAction?qtc_tLabels1=12345','last_tracking_update': '2025-03-22T16:14:48'}", + "agent: Your package is scheduled to be delivered on April 30, 2025 via USPS. Here is the most recent status from them regarding your package, updated as of March 22: \n" + "Your item has left our acceptance facility and is in transit to a sorting facility on April 10, 2025 at 7:06 am in IRON RIDGE, WI 53035. \n" + "You can find the full tracking details here: tracking_link !", + ] + ), +) + +goal_ecomm_list_orders = AgentGoal( + id = "goal_ecomm_list_orders", + category_tag="ecommerce", + agent_name="List All Orders", + agent_friendly_description="List all orders for a user.", + tools=[ + tool_registry.ecomm_list_orders, + tool_registry.ecomm_get_order, + tool_registry.ecomm_track_package, + ], + description="The user wants to see all of their orders. They may want more details about specific orders, and if the status of an order is 'shipped' or 'delivered', they might want to get the package tracking information. To assist with that goal, help the user gather args for this tool: " + "1. ListOrders: list orders for a user" + " and then offer the following tools, in a loop, until the user indicates they are done:" + "2. GetOrder: get information about an order. This tool is optional." + "3. TrackPackage: provide tracking information for the package. This tool is optional and should only be offered if the status is 'shipped' OR 'delivered' - otherwise, skip this tool and do not mention it to the user.", + starter_prompt=starter_prompt_generic, + example_conversation_history="\n ".join( + [ + "user: I'd like to see all of my orders.", + "agent: Sure! I can help you out with that. May I have your email address?", + "user: email is bob.johnson@emailzzz.com ", + "user_confirmed_tool_run: ", + "tool_result: a list of orders including [{'id': '102', 'summary': 'Red Sunglasses', 'email': 'matt.murdock@nelsonmurdock.com', 'status': 'shipped', 'order_date': '2025-04-01', 'last_order_update': '2025-04-06', 'tracking_id': '039813852990618' }, { 'id': '103', 'summary': 'Blue Sunglasses', 'email': 'matt.murdock@nelsonmurdock.com', 'status': 'paid', 'order_date': '2025-04-03', 'last_order_update': '2025-04-07' }]", + "agent: Your orders are as follows: \n", + "1. Red Sunglasses, ordered 4/1/2025 \n", + "2. Blue Sunglasses, ordered 4/3/2025 \n", + "Would you like more information about any of your orders?" + "user: Yes, the Red Sunglasses", + "agent: Your order 'Red Sunglasses,' placed April 1, 2025, was shipped on April 6, 2025. Would you like to see the tracking inforation?", + "user: Yes", + "user_confirmed_tool_run: ", + "tool_result: { 'scheduled_delivery_date': 'April 30, 2025', 'carrier': 'USPS', 'status_summary': 'Your item has left our acceptance facility and is in transit to a sorting facility on April 10, 2025 at 7:06 am in IRON RIDGE, WI 53035.', 'tracking_link': 'https://tools.usps.com/go/TrackConfirmAction?qtc_tLabels1=12345','last_tracking_update': '2025-03-22T16:14:48'}", + "agent: Your package is scheduled to be delivered on April 30, 2025 via USPS. Here is the most recent status from them regarding your package \n, updated as of March 22: \n" + "Your item has left our acceptance facility and is in transit to a sorting facility on April 10, 2025 at 7:06 am in IRON RIDGE, WI 53035. \n" + "You can find the full tracking details here: tracking_link ! \n" + "Would you like more information about any of your other orders?", + "user: No" + "agent: Thanks, and have a great day!" ] ), ) @@ -384,6 +460,6 @@ goal_list.append(goal_hr_check_paycheck_bank_integration_status) goal_list.append(goal_fin_check_account_balances) goal_list.append(goal_fin_move_money) goal_list.append(goal_fin_loan_application) - - +goal_list.append(goal_ecomm_list_orders) +goal_list.append(goal_ecomm_order_status) diff --git a/tools/search_flights.py b/tools/search_flights.py index a2335f0..24a907d 100644 --- a/tools/search_flights.py +++ b/tools/search_flights.py @@ -11,7 +11,7 @@ def search_airport(query: str) -> list: """ load_dotenv(override=True) api_key = os.getenv("RAPIDAPI_KEY", "YOUR_DEFAULT_KEY") - api_host = os.getenv("RAPIDAPI_HOST", "sky-scrapper.p.rapidapi.com") + api_host = os.getenv("RAPIDAPI_HOST_FLIGHTS", "sky-scrapper.p.rapidapi.com") conn = http.client.HTTPSConnection(api_host) headers = { @@ -73,7 +73,7 @@ def search_flights_real_api( # Step 2: Call flight search with resolved codes load_dotenv(override=True) api_key = os.getenv("RAPIDAPI_KEY", "YOUR_DEFAULT_KEY") - api_host = os.getenv("RAPIDAPI_HOST", "sky-scrapper.p.rapidapi.com") + api_host = os.getenv("RAPIDAPI_HOST_FLIGHTS", "sky-scrapper.p.rapidapi.com") conn = http.client.HTTPSConnection(api_host) headers = { diff --git a/tools/tool_registry.py b/tools/tool_registry.py index c9b601b..efc3cf3 100644 --- a/tools/tool_registry.py +++ b/tools/tool_registry.py @@ -333,6 +333,45 @@ financial_submit_loan_approval = ToolDefinition( name="amount", type="string", description="amount requested for the loan", + +# ----- ECommerce Use Case Tools ----- +ecomm_list_orders = ToolDefinition( + name="ListOrders", + description="Get all orders for a certain email address.", + arguments=[ + ToolArgument( + name="email_address", + type="string", + description="Email address of user by which to find orders", + ), + ], +) + +ecomm_get_order = ToolDefinition( + name="GetOrder", + description="Get infromation about an order by order ID.", + arguments=[ + ToolArgument( + name="order_id", + type="string", + description="ID of order to determine status of", + ), + ], +) + +ecomm_track_package = ToolDefinition( + name="TrackPackage", + description="Get tracking information for a package by shipping provider and tracking ID", + arguments=[ + ToolArgument( + name="tracking_id", + type="string", + description="ID of package to track", + ), + ToolArgument( + name="userConfirmation", + type="string", + description="Indication of user's desire to get package tracking information", ), ], ) \ No newline at end of file diff --git a/workflows/agent_goal_workflow.py b/workflows/agent_goal_workflow.py index 083332b..28a2b52 100644 --- a/workflows/agent_goal_workflow.py +++ b/workflows/agent_goal_workflow.py @@ -108,7 +108,7 @@ class AgentGoalWorkflow: conversation_history=self.conversation_history, agent_goal=self.goal, ) - validation_result = await workflow.execute_activity( + validation_result = await workflow.execute_activity_method( ToolActivities.agent_validatePrompt, args=[validation_input], schedule_to_close_timeout=LLM_ACTIVITY_SCHEDULE_TO_CLOSE_TIMEOUT, @@ -134,7 +134,7 @@ class AgentGoalWorkflow: prompt_input = ToolPromptInput(prompt=prompt, context_instructions=context_instructions) # connect to LLM and execute to get next steps - tool_data = await workflow.execute_activity( + tool_data = await workflow.execute_activity_method( ToolActivities.agent_toolPlanner, prompt_input, schedule_to_close_timeout=LLM_ACTIVITY_SCHEDULE_TO_CLOSE_TIMEOUT, @@ -210,7 +210,7 @@ class AgentGoalWorkflow: #Signal that comes from api/main.py via a post to /confirm @workflow.signal - async def confirmed(self) -> None: + async def confirm(self) -> None: """Signal handler for user confirmation of tool execution.""" workflow.logger.info("Received user signal: confirmation") self.confirmed = True @@ -316,8 +316,9 @@ class AgentGoalWorkflow: async def lookup_wf_env_settings(self, combined_input: CombinedInput)->None: env_lookup_input = EnvLookupInput( show_confirm_env_var_name = "SHOW_CONFIRM", - show_confirm_default = True) - env_output:EnvLookupOutput = await workflow.execute_activity( + show_confirm_default = True, + ) + env_output:EnvLookupOutput = await workflow.execute_activity_method( ToolActivities.get_wf_env_vars, env_lookup_input, start_to_close_timeout=LLM_ACTIVITY_START_TO_CLOSE_TIMEOUT, @@ -362,9 +363,11 @@ class AgentGoalWorkflow: # also don't forget you can look at the workflow itself and do queries if you want def print_useful_workflow_vars(self, status_or_step:str) -> None: print(f"***{status_or_step}:***") - print(f"force confirm? {self.tool_data['force_confirm']}") - print(f"next step: {self.tool_data.get('next')}") - print(f"current_tool: {self.tool_data.get('tool')}") - print(f"self.confirm: {self.confirmed}") - print(f"waiting_for_confirm (about to be set to true): {self.waiting_for_confirm}") + if self.tool_data: + print(f"force confirm? {self.tool_data['force_confirm']}") + print(f"next step: {self.tool_data.get('next')}") + print(f"current_tool: {self.tool_data.get('tool')}") + else: + print("no tool data initialized yet") + print(f"self.confirmed: {self.confirmed}")