setup update

clean up setup.md
bump dependencies
2026-03-15 14:08:08 +01:00 · 2025-05-26 14:33:32 -07:00 · 2025-05-26 14:28:45 -07:00 · 2025-05-26 14:20:28 -07:00 · 2025-05-26 14:20:28 -07:00 · 2025-05-26 14:20:28 -07:00
59 changed files with 585 additions and 2844 deletions
--- a/.env.example
+++ b/.env.example
@@ -1,8 +1,7 @@
 RAPIDAPI_KEY=9df2cb5...
 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=
-# Leave blank to use the built-in mock fixtures generator
+FOOTBALL_DATA_API_KEY=....

 STRIPE_API_KEY=sk_test_51J...

--- a/.gitignore
+++ b/.gitignore
@@ -33,6 +33,3 @@ coverage.xml

 .env
 .env*
-
-# Cursor
-.cursor
--- a/AGENTS.md
+++ b/AGENTS.md
@@ -1,175 +0,0 @@
-# Temporal AI Agent Contribution Guide
-
-## Repository Layout
- `workflows/` - Temporal workflows including the main AgentGoalWorkflow for multi-turn AI conversations
- `activities/` - Temporal activities for tool execution and LLM interactions  
- `tools/` - AI agent tools organized by category (finance, HR, ecommerce, travel, etc.)
- `models/` - Data types and tool definitions used throughout the system
- `prompts/` - Agent prompt generators and templates
- `api/` - FastAPI server that exposes REST endpoints to interact with workflows
- `frontend/` - React-based web UI for chatting with the AI agent
- `tests/` - Comprehensive test suite for workflows and activities using Temporal's testing framework
- `enterprise/` - .NET worker implementation for enterprise activities (train booking)
- `scripts/` - Utility scripts for running workers and testing tools
-
-## Running the Application
-
-### Quick Start with Docker
-```bash
-# Start all services with development hot-reload
-docker compose up -d
-
-# Quick rebuild without infrastructure
-docker compose up -d --no-deps --build api worker frontend
-```
-
-Default URLs:
- Temporal UI: http://localhost:8080
- API: http://localhost:8000  
- Frontend: http://localhost:5173
-
-### Local Development Setup
-
-1. **Prerequisites:**
-   ```bash
-   # Install Poetry for Python dependency management
-   curl -sSL https://install.python-poetry.org | python3 -
-   
-   # Start Temporal server (Mac)
-   brew install temporal
-   temporal server start-dev
-   ```
-
-2. **Backend (Python):**
-   ```bash
-   # Quick setup using Makefile
-   make setup              # Creates venv and installs dependencies
-   make run-worker         # Starts the Temporal worker
-   make run-api            # Starts the API server
-   
-   # Or manually:
-   poetry install
-   poetry run python scripts/run_worker.py    # In one terminal
-   poetry run uvicorn api.main:app --reload   # In another terminal
-   ```
-
-3. **Frontend (React):**
-   ```bash
-   make run-frontend       # Using Makefile
-   
-   # Or manually:
-   cd frontend
-   npm install
-   npx vite
-   ```
-
-4. **Enterprise .NET Worker (optional):**
-   ```bash
-   make run-enterprise     # Using Makefile
-   
-   # Or manually:
-   cd enterprise
-   dotnet build
-   dotnet run
-   ```
-
-### Environment Configuration
-Copy `.env.example` to `.env` and configure:
-```bash
-# Required: LLM Configuration
-LLM_MODEL=openai/gpt-4o          # or anthropic/claude-3-sonnet, etc.
-LLM_KEY=your-api-key-here
-
-# Optional: Agent Goals and Categories  
-AGENT_GOAL=goal_choose_agent_type
-GOAL_CATEGORIES=hr,travel-flights,travel-trains,fin
-
-# Optional: Tool-specific APIs
-STRIPE_API_KEY=sk_test_...       # For invoice creation
-FOOTBALL_DATA_API_KEY=...        # For real football fixtures
-```
-
-## Testing
-
-The project includes comprehensive tests using Temporal's testing framework:
-
-```bash
-# Install test dependencies
-poetry install --with dev
-
-# Run all tests
-poetry run pytest
-
-# Run with time-skipping for faster execution  
-poetry run pytest --workflow-environment=time-skipping
-
-# Run specific test categories
-poetry run pytest tests/test_tool_activities.py -v     # Activity tests
-poetry run pytest tests/test_agent_goal_workflow.py -v # Workflow tests
-
-# Run with coverage
-poetry run pytest --cov=workflows --cov=activities
-```
-
-**Test Coverage:**
- ✅ **Workflow Tests**: AgentGoalWorkflow signals, queries, state management
- ✅ **Activity Tests**: ToolActivities, LLM integration (mocked), environment configuration  
- ✅ **Integration Tests**: End-to-end workflow and activity execution
-
-**Documentation:**
- **Quick Start**: [TESTING.md](TESTING.md) - Simple commands to run tests
- **Comprehensive Guide**: [tests/README.md](tests/README.md) - Detailed testing patterns and best practices
-
-## Linting and Code Quality
-
-```bash
-# Using Poetry tasks
-poetry run poe format    # Format code with black and isort
-poetry run poe lint      # Check code style and types
-poetry run poe test      # Run test suite
-
-# Manual commands  
-poetry run black .
-poetry run isort .
-poetry run mypy --check-untyped-defs --namespace-packages .
-```
-
-## Agent Customization
-
-### Adding New Tools
-1. Create tool implementation in `tools/` directory
-2. Add tool function mapping in `tools/__init__.py`  
-3. Register tool definition in `tools/tool_registry.py`
-4. Associate with goals in `tools/goal_registry.py`
-
-### Configuring Goals
-The agent supports multiple goal categories:
- **Financial**: Money transfers, loan applications (`fin/`)
- **HR**: PTO booking, payroll status (`hr/`)  
- **Travel**: Flight/train booking, event finding
- **Ecommerce**: Order tracking, package management (`ecommerce/`)
-
-See [adding-goals-and-tools.md](adding-goals-and-tools.md) for detailed customization guide.
-
-## Architecture
-
-This system implements "Agentic AI" with these key components:
-1. **Goals** - High-level objectives accomplished through tool sequences
-2. **Agent Loops** - LLM execution → tool calls → human input → repeat until goal completion
-3. **Tool Approval** - Human confirmation for sensitive operations
-4. **Conversation Management** - LLM-powered input validation and history summarization
-5. **Durability** - Temporal workflows ensure reliable execution across failures
-
-For detailed architecture information, see [architecture.md](architecture.md).
-
-## Commit Messages and Pull Requests
- Use clear commit messages describing the change purpose
- Reference specific files and line numbers when relevant (e.g., `workflows/agent_goal_workflow.py:125`)
- Open PRs describing **what changed** and **why**
- Ensure tests pass before submitting: `poetry run pytest --workflow-environment=time-skipping`
-
-## Additional Resources
- **Setup Guide**: [SETUP.md](SETUP.md) - Detailed configuration instructions
- **Architecture Decisions**: [architecture-decisions.md](architecture-decisions.md) - Why Temporal for AI agents
- **Demo Video**: [5-minute YouTube overview](https://www.youtube.com/watch?v=GEXllEH2XiQ)
- **Multi-Agent Demo**: [Advanced multi-agent execution](https://www.youtube.com/watch?v=8Dc_0dC14yY)
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,12 +8,12 @@ All notable changes to this project will be documented in this file.

 ### Added
 - **Multi‑goal agent architecture** with dynamic goal switching (`goal_choose_agent_type`, `ListAgents`, `ChangeGoal`).
-    - See [the architecture guide](./architecture.md) and [setup guide](./SETUP.md).
+    - See [the architecture guide](./architecture.md) and [setup guide](./setup.md).
 - **New goal categories & agents**: HR PTO scheduling/checking, paycheck integration, Financial (balances, money movement, loan application), E‑commerce order tracking.
    - See [the guide for adding goals and tools](./adding-goals-and-tools.md).
 - **Force Confirmation**: `SHOW_CONFIRM` will show a confirmation box before allowing the agent to run a tool.
 - **Grok (`x.ai`) LLM provider** support via `GROK_API_KEY`.
- Extensive **docs**: `SETUP.md`, `architecture.md`, `architecture-decisions.md`, `adding-goals-and-tools.md`, plus new diagrams & assets.
+- Extensive **docs**: `setup.md`, `architecture.md`, `architecture-decisions.md`, `adding-goals-and-tools.md`, plus new diagrams & assets.

 ### Changed
 - **UI Confirmation Box** is less 'debug' looking and prettier.
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -1,106 +0,0 @@
-# Contributing to the Temporal AI Agent Project
-
-This document provides guidelines for contributing to `temporal-ai-agent`. All setup and installation instructions can be found in [./SETUP.md](./SETUP.md).
-
-## Getting Started
-
-### Code Style & Formatting
-We use `black` for code formatting and `isort` for import sorting to maintain a consistent codebase.
-   **Format code:**
-    ```bash
-    poetry run poe format
-    ```
-    Or manually:
-    ```bash
-    poetry run black .
-    poetry run isort .
-    ```
-    Please format your code before committing.
-
-### Linting & Type Checking
-We use `mypy` for static type checking and other linters configured via `poe the poet`.
-   **Run linters and type checks:**
-    ```bash
-    poetry run poe lint
-    ```
-    Or manually for type checking:
-    ```bash
-    poetry run mypy --check-untyped-defs --namespace-packages .
-    ```
-    Ensure all linting and type checks pass before submitting a pull request.
-
-## Testing
-Comprehensive testing is crucial for this project. We use `pytest` and Temporal's testing framework.
-   **Install test dependencies** (if not already done with `poetry install --with dev`):
-    ```bash
-    poetry install --with dev
-    ```
-   **Run all tests:**
-    ```bash
-    poetry run pytest
-    ```
-   **Run tests with time-skipping (recommended for faster execution, especially in CI):**
-    ```bash
-    poetry run pytest --workflow-environment=time-skipping
-    ```
-
-For detailed information on test categories, running specific tests, test environments, coverage, and troubleshooting, please refer to:
-   [TESTING.md](./TESTING.md) (Quick Start and overview)
-   [tests/README.md](./tests/README.md) (Comprehensive guide, patterns, and best practices)
-
-**Ensure all tests pass before submitting a pull request.**
-
-## Making Changes
-
-### Adding New Tools or Goals
-If you're looking to extend the agent's capabilities:
-1.  Create your tool implementation in the `tools/` directory.
-2.  Register your tool and associate it with relevant goals.
-For detailed instructions, please see:
-   [Agent Customization in agents.md](./agents.md#agent-customization)
-   [Adding Goals and Tools Guide](./adding-goals-and-tools.md)
-
-### General Code Changes
-   Follow the existing code style and patterns.
-   Ensure any new code is well-documented with comments.
-   Write new tests for new functionality or bug fixes.
-   Update existing tests if necessary.
-
-## Submitting Contributions
-
-### Pull Requests
-When you're ready to submit your changes:
-1.  Push your branch to the remote repository.
-2.  Open a Pull Request (PR) against the `main` branch.
-3.  **Describe your changes:** Clearly explain what you changed and why. Reference any related issues.
-4.  **Ensure tests pass:** All CI checks, including tests and linters, must pass. The command `poetry run pytest --workflow-environment=time-skipping` is a good one to run locally.
-5.  **Request review:** Request a review from one or more maintainers.
-
-## Reporting Bugs
-If you encounter a bug, please:
-1.  **Search existing issues:** Check if the bug has already been reported.
-2.  **Open a new issue:** If not, create a new issue.
-    -   Provide a clear and descriptive title.
-    -   Include steps to reproduce the bug.
-    -   Describe the expected behavior and what actually happened.
-    -   Provide details about your environment (OS, Python version, Temporal server version, etc.).
-    -   Include any relevant logs or screenshots.
-
-## Suggesting Enhancements
-We welcome suggestions for new features or improvements!
-1.  **Search existing issues/discussions:** See if your idea has already been discussed.
-2.  **Open a new issue:**
-    -   Use a clear and descriptive title.
-    -   Provide a detailed explanation of the enhancement and its benefits.
-    -   Explain the use case or problem it solves.
-    -   Include any potential implementation ideas if you have them.
-
-## Key Resources
-   **Project Overview**: [README.md](./README.md)
-   **Detailed Contribution & Development Guide**: [agents.md](./agents.md)
-   **Setup Instructions**: [SETUP.md](./SETUP.md)
-   **Comprehensive Testing Guide**: [TESTING.md](./TESTING.md) and [tests/README.md](./tests/README.md)
-   **System Architecture**: [architecture.md](./architecture.md)
-   **Architecture Decisions**: [architecture-decisions.md](./architecture-decisions.md)
-   **Customizing Agent Tools and Goals**: [adding-goals-and-tools.md](./adding-goals-and-tools.md)
-   **To-Do List / Future Enhancements**: [todo.md](./todo.md)
--- a/3
+++ b/3
@@ -5,7 +5,6 @@ setup: check-python setup-venv install

 check-python:
 	@which python3 >/dev/null 2>&1 || (echo "Python 3 is required. Please install it first." && exit 1)
-	@which poetry >/dev/null 2>&1 || (echo "Poetry is required. Please install it first." && exit 1)

 setup-venv:
 	python3 -m venv venv
@@ -60,4 +59,4 @@ help:
 	@echo "  make run-legacy-worker  - Start the legacy worker"
 	@echo "  make run-enterprise     - Build and run the enterprise .NET worker"
 	@echo "  make setup-temporal-mac - Install and start Temporal server on Mac"
-	@echo "  make run-dev            - Start all development services (worker, API, frontend) in parallel"
+	@echo "  make run-dev            - Start all development services (worker, API, frontend) in parallel" 
--- a/README.md
+++ b/README.md
@@ -34,7 +34,7 @@ These are the key elements of an agentic framework:
 For a deeper dive into this, check out the [architecture guide](./architecture.md).

 ## Setup and Configuration
-See [the Setup guide](./SETUP.md) for detailed instructions. The basic configuration requires just two environment variables:
+See [the Setup guide](./setup.md) for detailed instructions. The basic configuration requires just two environment variables:
 ```bash
 LLM_MODEL=openai/gpt-4o  # or any other model supported by LiteLLM
 LLM_KEY=your-api-key-here
@@ -46,43 +46,14 @@ See [the guide to adding goals and tools](./adding-goals-and-tools.md).
 ## Architecture
 See [the architecture guide](./architecture.md).

-## Testing
-
-The project includes comprehensive tests for workflows and activities using Temporal's testing framework:
-
-```bash
-# Install dependencies including test dependencies
-poetry install --with dev
-
-# Run all tests
-poetry run pytest
-
-# Run with time-skipping for faster execution
-poetry run pytest --workflow-environment=time-skipping
-```
-
-**Test Coverage:**
- ✅ **Workflow Tests**: AgentGoalWorkflow signals, queries, state management
- ✅ **Activity Tests**: ToolActivities, LLM integration (mocked), environment configuration
- ✅ **Integration Tests**: End-to-end workflow and activity execution
-
-**Documentation:**
- **Quick Start**: [TESTING.md](TESTING.md) - Simple commands to run tests
- **Comprehensive Guide**: [tests/README.md](tests/README.md) - Detailed testing documentation, patterns, and best practices
-
-## Development
-
-To contribute to this project, see [CONTRIBUTING.md](CONTRIBUTING.md).
-
-Start the Temporal Server and API server, see [setup](SETUP.md)
-
 ## Productionalization & Adding Features
 - In a prod setting, I would need to ensure that payload data is stored separately (e.g. in S3 or a noSQL db - the claim-check pattern), or otherwise 'garbage collected'. Without these techniques, long conversations will fill up the workflow's conversation history, and start to breach Temporal event history payload limits.
 - A single worker can easily support many agent workflows (chats) running at the same time. Currently the workflow ID is the same each time, so it will only run one agent at a time. To run multiple agents, you can use a different workflow ID each time (e.g. by using a UUID or timestamp).
 - Perhaps the UI should show when the LLM response is being retried (i.e. activity retry attempt because the LLM provided bad output)
- The project now includes comprehensive tests for workflows and activities! [See testing guide](TESTING.md).
+- Tests would be nice! [See tests](./tests/).

-See [the todo](./todo.md) for more details on things we want to do (or that you could contribute!).
+
+See [the todo](./todo.md) for more details.

 See [the guide to adding goals and tools](./adding-goals-and-tools.md) for more ways you can add features.

--- a/TESTING.md
+++ b/TESTING.md
@@ -1,163 +0,0 @@
-# Testing the Temporal AI Agent
-
-This guide provides instructions for running the comprehensive test suite for the Temporal AI Agent project.
-
-## Quick Start
-
-1. **Install dependencies**:
-   ```bash
-   poetry install --with dev
-   ```
-
-2. **Run all tests**:
-   ```bash
-   poetry run pytest
-   ```
-
-3. **Run with time-skipping for faster execution**:
-   ```bash
-   poetry run pytest --workflow-environment=time-skipping
-   ```
-
-## Test Categories
-
-### Unit Tests
- **Activity Tests**: `tests/test_tool_activities.py`
-  - LLM integration (mocked)
-  - Environment configuration
-  - JSON processing
-  - Dynamic tool execution
-
-### Integration Tests  
- **Workflow Tests**: `tests/test_agent_goal_workflow.py`
-  - Full workflow execution
-  - Signal and query handling
-  - State management
-  - Error scenarios
-
-## Running Specific Tests
-
-```bash
-# Run only activity tests
-poetry run pytest tests/test_tool_activities.py -v
-
-# Run only workflow tests  
-poetry run pytest tests/test_agent_goal_workflow.py -v
-
-# Run a specific test
-poetry run pytest tests/test_tool_activities.py::TestToolActivities::test_sanitize_json_response -v
-
-# Run tests matching a pattern
-poetry run pytest -k "validation" -v
-```
-
-## Test Environment Options
-
-### Local Environment (Default)
-```bash
-poetry run pytest --workflow-environment=local
-```
-
-### Time-Skipping Environment (Recommended for CI)
-```bash
-poetry run pytest --workflow-environment=time-skipping
-```
-
-### External Temporal Server
-```bash
-poetry run pytest --workflow-environment=localhost:7233
-```
-
-## Environment Variables
-
-Tests can be configured with these environment variables:
-
- `LLM_MODEL`: Model for LLM testing (default: "openai/gpt-4")
- `LLM_KEY`: API key for LLM service (mocked in tests)
- `LLM_BASE_URL`: Custom LLM endpoint (optional)
-
-## Test Coverage
-
-The test suite covers:
-
-✅ **Workflows**
- AgentGoalWorkflow initialization and execution
- Signal handling (user_prompt, confirm, end_chat)
- Query methods (conversation history, agent goal, tool data)
- State management and conversation flow
- Validation and error handling
-
-✅ **Activities**  
- ToolActivities class methods
- LLM integration (mocked)
- Environment variable handling
- JSON response processing
- Dynamic tool activity execution
-
-✅ **Integration**
- End-to-end workflow execution
- Activity registration in workers
- Temporal client interactions
-
-## Test Output
-
-Successful test run example:
-```
-============================== test session starts ==============================
-platform darwin -- Python 3.11.3, pytest-8.3.5, pluggy-1.5.0
-rootdir: /Users/steveandroulakis/Documents/Code/agentic/temporal-demo/temporal-ai-agent
-configfile: pyproject.toml
-plugins: anyio-4.5.2, asyncio-0.26.0
-collected 21 items
-
-tests/test_tool_activities.py::TestToolActivities::test_sanitize_json_response PASSED
-tests/test_tool_activities.py::TestToolActivities::test_parse_json_response_success PASSED
-tests/test_tool_activities.py::TestToolActivities::test_get_wf_env_vars_default_values PASSED
-...
-
-============================== 21 passed in 12.5s ==============================
-```
-
-## Troubleshooting
-
-### Common Issues
-
-1. **Module not found errors**: Run `poetry install --with dev`
-2. **Async warnings**: These are expected with pytest-asyncio and can be ignored  
-3. **Test timeouts**: Use `--workflow-environment=time-skipping` for faster execution
-4. **Import errors**: Check that you're running tests from the project root directory
-
-### Debugging Tests
-
-Enable verbose logging:
-```bash
-poetry run pytest --log-cli-level=DEBUG -s
-```
-
-Run with coverage:
-```bash
-poetry run pytest --cov=workflows --cov=activities
-```
-
-## Continuous Integration
-
-For CI environments, use:
-```bash
-poetry run pytest --workflow-environment=time-skipping --tb=short
-```
-
-## Additional Resources
-
- See `tests/README.md` for detailed testing documentation
- Review `tests/conftest.py` for available test fixtures
- Check individual test files for specific test scenarios
-
-## Test Architecture
-
-The tests use:
- **Temporal Testing Framework**: For workflow and activity testing
- **pytest-asyncio**: For async test support  
- **unittest.mock**: For mocking external dependencies
- **Test Fixtures**: For consistent test data and setup
-
-All external dependencies (LLM calls, file I/O) are mocked to ensure fast, reliable tests.
--- a/activities/tool_activities.py
+++ b/activities/tool_activities.py
@@ -1,25 +1,16 @@
 import inspect
+from temporalio import activity
 import json
+from typing import Optional, Sequence
+from temporalio.common import RawValue
 import os
 from datetime import datetime
-from typing import Sequence
-
 from dotenv import load_dotenv
+from models.data_types import EnvLookupOutput, ValidationInput, ValidationResult, ToolPromptInput, EnvLookupInput
 from litellm import completion
-from temporalio import activity
-from temporalio.common import RawValue
-
-from models.data_types import (
-    EnvLookupInput,
-    EnvLookupOutput,
-    ToolPromptInput,
-    ValidationInput,
-    ValidationResult,
-)

 load_dotenv(override=True)

-
 class ToolActivities:
    def __init__(self):
        """Initialize LLM client using LiteLLM."""
@@ -31,9 +22,7 @@ class ToolActivities:
            print(f"Using custom base URL: {self.llm_base_url}")

    @activity.defn
-    async def agent_validatePrompt(
-        self, validation_input: ValidationInput
-    ) -> ValidationResult:
+    async def agent_validatePrompt(self, validation_input: ValidationInput) -> ValidationResult:
        """
        Validates the prompt in the context of the conversation history and agent goal.
        Returns a ValidationResult indicating if the prompt makes sense given the context.
@@ -110,15 +99,15 @@ class ToolActivities:
            completion_kwargs = {
                "model": self.llm_model,
                "messages": messages,
-                "api_key": self.llm_key,
+                "api_key": self.llm_key
            }
-
+            
            # Add base_url if configured
            if self.llm_base_url:
                completion_kwargs["base_url"] = self.llm_base_url

            response = completion(**completion_kwargs)
-
+            
            response_content = response.choices[0].message.content
            activity.logger.info(f"LLM response: {response_content}")

@@ -147,20 +136,19 @@ class ToolActivities:
        """
        # Remove any markdown code block markers
        response_content = response_content.replace("```json", "").replace("```", "")
-
+        
        # Remove any leading/trailing whitespace
        response_content = response_content.strip()
-
+        
        return response_content

    @activity.defn
    async def get_wf_env_vars(self, input: EnvLookupInput) -> EnvLookupOutput:
-        """gets env vars for workflow as an activity result so it's deterministic
-        handles default/None
+        """ gets env vars for workflow as an activity result so it's deterministic
+            handles default/None
        """
-        output: EnvLookupOutput = EnvLookupOutput(
-            show_confirm=input.show_confirm_default, multi_goal_mode=True
-        )
+        output: EnvLookupOutput = EnvLookupOutput(show_confirm=input.show_confirm_default, 
+                                                  multi_goal_mode=True)
        show_confirm_value = os.getenv(input.show_confirm_env_var_name)
        if show_confirm_value is None:
            output.show_confirm = input.show_confirm_default
@@ -168,21 +156,17 @@ class ToolActivities:
            output.show_confirm = False
        else:
            output.show_confirm = True
-
+        
        first_goal_value = os.getenv("AGENT_GOAL")
        if first_goal_value is None:
-            output.multi_goal_mode = True  # default if unset
-        elif (
-            first_goal_value is not None
-            and first_goal_value.lower() != "goal_choose_agent_type"
-        ):
+            output.multi_goal_mode = True # default if unset
+        elif first_goal_value is not None and first_goal_value.lower() != "goal_choose_agent_type":
            output.multi_goal_mode = False
        else:
            output.multi_goal_mode = True

        return output

-
@activity.defn(dynamic=True)
 async def dynamic_tool_activity(args: Sequence[RawValue]) -> dict:
    from tools import get_handler
@@ -201,3 +185,5 @@ async def dynamic_tool_activity(args: Sequence[RawValue]) -> dict:
    # Optionally log or augment the result
    activity.logger.info(f"Tool '{tool_name}' result: {result}")
    return result
+
+
--- a/adding-goals-and-tools.md
+++ b/adding-goals-and-tools.md
@@ -49,7 +49,7 @@ description="Help the user gather args for these tools in order: "
 ```

 Tools should generally return meaningful information and be generally ‘failsafe’ in returning a useful result based on input.
-(If you're doing a local data approach like those in [.tools/data/](./tools/data/)) it's good to document how they can be setup to get a good result in tool specific [setup](./SETUP.md).
+(If you're doing a local data approach like those in [.tools/data/](./tools/data/)) it's good to document how they can be setup to get a good result in tool specific [setup](./setup.md).

 ### Add to Tool Registry
 1.  Open [/tools/tool_registry.py](tools/tool_registry.py) - this file contains mapping of tool names to tool definitions (so the AI understands how to use them)
--- a/api/main.py
+++ b/api/main.py
@@ -1,18 +1,18 @@
-import asyncio
 import os
+from fastapi import FastAPI
 from typing import Optional
-
-from dotenv import load_dotenv
-from fastapi import FastAPI, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
-from temporalio.api.enums.v1 import WorkflowExecutionStatus
 from temporalio.client import Client
 from temporalio.exceptions import TemporalError
+from temporalio.api.enums.v1 import WorkflowExecutionStatus
+from fastapi import HTTPException
+from dotenv import load_dotenv
+import asyncio

-from models.data_types import AgentGoalWorkflowParams, CombinedInput
-from shared.config import TEMPORAL_TASK_QUEUE, get_temporal_client
-from tools.goal_registry import goal_list
 from workflows.agent_goal_workflow import AgentGoalWorkflow
+from models.data_types import CombinedInput, AgentGoalWorkflowParams
+from tools.goal_registry import goal_list
+from fastapi.middleware.cors import CORSMiddleware
+from shared.config import get_temporal_client, TEMPORAL_TASK_QUEUE

 app = FastAPI()
 temporal_client: Optional[Client] = None
@@ -23,9 +23,7 @@ load_dotenv()

 def get_initial_agent_goal():
    """Get the agent goal from environment variables."""
-    env_goal = os.getenv(
-        "AGENT_GOAL", "goal_choose_agent_type"
-    )  # if no goal is set in the env file, default to choosing an agent
+    env_goal = os.getenv("AGENT_GOAL", "goal_choose_agent_type") #if no goal is set in the env file, default to choosing an agent
    for listed_goal in goal_list:
        if listed_goal.id == env_goal:
            return listed_goal
@@ -121,8 +119,7 @@ async def get_conversation_history():
            raise HTTPException(
                status_code=500, detail="Internal server error while querying workflow."
            )
-
-
+    
@app.get("/agent-goal")
 async def get_agent_goal():
    """Calls the workflow's 'get_agent_goal' query."""
@@ -151,7 +148,7 @@ async def send_prompt(prompt: str):
    combined_input = CombinedInput(
        tool_params=AgentGoalWorkflowParams(None, None),
        agent_goal=get_initial_agent_goal(),
-        # change to get from workflow query
+        #change to get from workflow query
    )

    workflow_id = "agent-workflow"
--- a/architecture-decisions.md
+++ b/architecture-decisions.md
@@ -3,7 +3,7 @@ This documents some of the "why" behind the [architecture](./architecture.md).

 ## AI Models
 We wanted to have flexibility to use different models, because this space is changing rapidly and models get better regularly.
-Also, for you, we wanted to let you pick your model of choice. The system is designed to make changing models out simple. For how to do that, checkout the [setup guide](./SETUP.md).
+Also, for you, we wanted to let you pick your model of choice. The system is designed to make changing models out simple. For how to do that, checkout the [setup guide](./setup.md).

 ## Temporal
 We asked one of the AI models used in this demo to answer this question (edited minorly):
--- a/architecture.md
+++ b/architecture.md
@@ -39,7 +39,7 @@ This is where you can add probabalistic business logic to
 ## LLM
 Probabalistic execution: it will _probably_ do what you tell it to do.
 Turns the guidance from the prompts (see [agent prompts](./prompts/agent_prompt_generators.py) and [goal prompts](./tools/goal_registry.py)) into 
-You have a choice of providers - see [setup](./SETUP.md). 
+You have a choice of providers - see [setup](./setup.md). 
 The LLM:
 - Drives toward the initial Goal and any subsequent Goals selected by user
 - Decides what to do based on input, such as:
--- a/frontend/src/pages/App.jsx
+++ b/frontend/src/pages/App.jsx
@@ -3,7 +3,7 @@ import NavBar from "../components/NavBar";
 import ChatWindow from "../components/ChatWindow";
 import { apiService } from "../services/api";

-const POLL_INTERVAL = 600; // 0.6 seconds
+const POLL_INTERVAL = 500; // 0.5 seconds
 const INITIAL_ERROR_STATE = { visible: false, message: '' };
 const DEBOUNCE_DELAY = 300; // 300ms debounce for user input

--- a/models/data_types.py
+++ b/models/data_types.py
@@ -1,6 +1,5 @@
 from dataclasses import dataclass
-from typing import Any, Deque, Dict, List, Literal, Optional, Union
-
+from typing import Optional, Deque, Dict, Any, List, Union, Literal
 from models.tool_definitions import AgentGoal


@@ -44,14 +43,12 @@ class ValidationResult:
        if self.validationFailedReason is None:
            self.validationFailedReason = {}

-
@dataclass
 class EnvLookupInput:
    show_confirm_env_var_name: str
    show_confirm_default: bool

-
@dataclass
 class EnvLookupOutput:
    show_confirm: bool
-    multi_goal_mode: bool
+    multi_goal_mode: bool
--- a/models/tool_definitions.py
+++ b/models/tool_definitions.py
@@ -15,7 +15,6 @@ class ToolDefinition:
    description: str
    arguments: List[ToolArgument]

-
@dataclass
 class AgentGoal:
    id: str
@@ -25,4 +24,6 @@ class AgentGoal:
    tools: List[ToolDefinition]
    description: str = "Description of the tools purpose and overall goal"
    starter_prompt: str = "Initial prompt to start the conversation"
-    example_conversation_history: str = "Example conversation history to help the AI agent understand the context of the conversation"
+    example_conversation_history: str = (
+        "Example conversation history to help the AI agent understand the context of the conversation"
+    )
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.5.1 and should not be changed by hand.
+# This file is automatically @generated by Poetry 2.1.3 and should not be changed by hand.

 [[package]]
 name = "aiohappyeyeballs"
@@ -6,6 +6,7 @@ version = "2.6.1"
 description = "Happy Eyeballs for asyncio"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "aiohappyeyeballs-2.6.1-py3-none-any.whl", hash = "sha256:f349ba8f4b75cb25c99c5c2d84e997e485204d2902a9597802b0371f09331fb8"},
    {file = "aiohappyeyeballs-2.6.1.tar.gz", hash = "sha256:c3f9d0113123803ccadfdf3f0faa505bc78e6a72d1cc4806cbd719826e943558"},
@@ -17,6 +18,7 @@ version = "3.11.18"
 description = "Async http client/server framework (asyncio)"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "aiohttp-3.11.18-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:96264854fedbea933a9ca4b7e0c745728f01380691687b7365d18d9e977179c4"},
    {file = "aiohttp-3.11.18-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:9602044ff047043430452bc3a2089743fa85da829e6fc9ee0025351d66c332b6"},
@@ -112,7 +114,7 @@ propcache = ">=0.2.0"
 yarl = ">=1.17.0,<2.0"

 [package.extras]
-speedups = ["Brotli", "aiodns (>=3.2.0)", "brotlicffi"]
+speedups = ["Brotli ; platform_python_implementation == \"CPython\"", "aiodns (>=3.2.0) ; sys_platform == \"linux\" or sys_platform == \"darwin\"", "brotlicffi ; platform_python_implementation != \"CPython\""]

 [[package]]
 name = "aiosignal"
@@ -120,6 +122,7 @@ version = "1.3.2"
 description = "aiosignal: a list of registered asynchronous callbacks"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "aiosignal-1.3.2-py2.py3-none-any.whl", hash = "sha256:45cde58e409a301715980c2b01d0c28bdde3770d8290b5eb2173759d9acb31a5"},
    {file = "aiosignal-1.3.2.tar.gz", hash = "sha256:a8c255c66fafb1e499c9351d0bf32ff2d8a0321595ebac3b93713656d2436f54"},
@@ -134,6 +137,7 @@ version = "0.7.0"
 description = "Reusable constraint types to use with typing.Annotated"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "annotated_types-0.7.0-py3-none-any.whl", hash = "sha256:1f02e8b43a8fbbc3f3e0d4f0f4bfc8131bcb4eebe8849b8e5c773f3a1c582a53"},
    {file = "annotated_types-0.7.0.tar.gz", hash = "sha256:aff07c09a53a08bc8cfccb9c85b05f1aa9a2a6f23728d790723543408344ce89"},
@@ -145,6 +149,7 @@ version = "4.5.2"
 description = "High level compatibility layer for multiple asynchronous event loop implementations"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "anyio-4.5.2-py3-none-any.whl", hash = "sha256:c011ee36bc1e8ba40e5a81cb9df91925c218fe9b778554e0b56a21e1b5d4716f"},
    {file = "anyio-4.5.2.tar.gz", hash = "sha256:23009af4ed04ce05991845451e11ef02fc7c5ed29179ac9a420e5ad0ac7ddc5b"},
@@ -158,7 +163,7 @@ typing-extensions = {version = ">=4.1", markers = "python_version < \"3.11\""}

 [package.extras]
 doc = ["Sphinx (>=7.4,<8.0)", "packaging", "sphinx-autodoc-typehints (>=1.2.0)", "sphinx-rtd-theme"]
-test = ["anyio[trio]", "coverage[toml] (>=7)", "exceptiongroup (>=1.2.0)", "hypothesis (>=4.0)", "psutil (>=5.9)", "pytest (>=7.0)", "pytest-mock (>=3.6.1)", "trustme", "truststore (>=0.9.1)", "uvloop (>=0.21.0b1)"]
+test = ["anyio[trio]", "coverage[toml] (>=7)", "exceptiongroup (>=1.2.0)", "hypothesis (>=4.0)", "psutil (>=5.9)", "pytest (>=7.0)", "pytest-mock (>=3.6.1)", "trustme", "truststore (>=0.9.1) ; python_version >= \"3.10\"", "uvloop (>=0.21.0b1) ; platform_python_implementation == \"CPython\" and platform_system != \"Windows\""]
 trio = ["trio (>=0.26.1)"]

 [[package]]
@@ -167,6 +172,8 @@ version = "5.0.1"
 description = "Timeout context manager for asyncio programs"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
+markers = "python_version == \"3.10\""
 files = [
    {file = "async_timeout-5.0.1-py3-none-any.whl", hash = "sha256:39e3809566ff85354557ec2398b55e096c8364bacac9405a7a1fa429e77fe76c"},
    {file = "async_timeout-5.0.1.tar.gz", hash = "sha256:d9321a7a3d5a6a5e187e824d2fa0793ce379a202935782d555d6e9d2735677d3"},
@@ -178,18 +185,19 @@ version = "25.3.0"
 description = "Classes Without Boilerplate"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "attrs-25.3.0-py3-none-any.whl", hash = "sha256:427318ce031701fea540783410126f03899a97ffc6f61596ad581ac2e40e3bc3"},
    {file = "attrs-25.3.0.tar.gz", hash = "sha256:75d7cefc7fb576747b2c81b4442d4d4a1ce0900973527c011d1030fd3bf4af1b"},
 ]

 [package.extras]
-benchmark = ["cloudpickle", "hypothesis", "mypy (>=1.11.1)", "pympler", "pytest (>=4.3.0)", "pytest-codspeed", "pytest-mypy-plugins", "pytest-xdist[psutil]"]
-cov = ["cloudpickle", "coverage[toml] (>=5.3)", "hypothesis", "mypy (>=1.11.1)", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins", "pytest-xdist[psutil]"]
-dev = ["cloudpickle", "hypothesis", "mypy (>=1.11.1)", "pre-commit-uv", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins", "pytest-xdist[psutil]"]
+benchmark = ["cloudpickle ; platform_python_implementation == \"CPython\"", "hypothesis", "mypy (>=1.11.1) ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pympler", "pytest (>=4.3.0)", "pytest-codspeed", "pytest-mypy-plugins ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pytest-xdist[psutil]"]
+cov = ["cloudpickle ; platform_python_implementation == \"CPython\"", "coverage[toml] (>=5.3)", "hypothesis", "mypy (>=1.11.1) ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pytest-xdist[psutil]"]
+dev = ["cloudpickle ; platform_python_implementation == \"CPython\"", "hypothesis", "mypy (>=1.11.1) ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pre-commit-uv", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pytest-xdist[psutil]"]
 docs = ["cogapp", "furo", "myst-parser", "sphinx", "sphinx-notfound-page", "sphinxcontrib-towncrier", "towncrier"]
-tests = ["cloudpickle", "hypothesis", "mypy (>=1.11.1)", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins", "pytest-xdist[psutil]"]
-tests-mypy = ["mypy (>=1.11.1)", "pytest-mypy-plugins"]
+tests = ["cloudpickle ; platform_python_implementation == \"CPython\"", "hypothesis", "mypy (>=1.11.1) ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pytest-xdist[psutil]"]
+tests-mypy = ["mypy (>=1.11.1) ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\"", "pytest-mypy-plugins ; platform_python_implementation == \"CPython\" and python_version >= \"3.10\""]

 [[package]]
 name = "black"
@@ -197,6 +205,7 @@ version = "23.12.1"
 description = "The uncompromising code formatter."
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
 files = [
    {file = "black-23.12.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:e0aaf6041986767a5e0ce663c7a2f0e9eaf21e6ff87a5f95cbf3675bfd4c41d2"},
    {file = "black-23.12.1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:c88b3711d12905b74206227109272673edce0cb29f27e1385f33b0163c414bba"},
@@ -233,7 +242,7 @@ typing-extensions = {version = ">=4.0.1", markers = "python_version < \"3.11\""}

 [package.extras]
 colorama = ["colorama (>=0.4.3)"]
-d = ["aiohttp (>=3.7.4)", "aiohttp (>=3.7.4,!=3.9.0)"]
+d = ["aiohttp (>=3.7.4) ; sys_platform != \"win32\" or implementation_name != \"pypy\"", "aiohttp (>=3.7.4,!=3.9.0) ; sys_platform == \"win32\" and implementation_name == \"pypy\""]
 jupyter = ["ipython (>=7.8.0)", "tokenize-rt (>=3.2.0)"]
 uvloop = ["uvloop (>=0.15.2)"]

@@ -243,6 +252,7 @@ version = "0.8.1"
 description = "Generate complex HTML+JS pages with Python"
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "branca-0.8.1-py3-none-any.whl", hash = "sha256:d29c5fab31f7c21a92e34bf3f854234e29fecdcf5d2df306b616f20d816be425"},
    {file = "branca-0.8.1.tar.gz", hash = "sha256:ac397c2d79bd13af0d04193b26d5ed17031d27609a7f1fab50c438b8ae712390"},
@@ -257,6 +267,7 @@ version = "2024.12.14"
 description = "Python package for providing Mozilla's CA Bundle."
 optional = false
 python-versions = ">=3.6"
+groups = ["main"]
 files = [
    {file = "certifi-2024.12.14-py3-none-any.whl", hash = "sha256:1275f7a45be9464efc1173084eaa30f866fe2e47d389406136d332ed4967ec56"},
    {file = "certifi-2024.12.14.tar.gz", hash = "sha256:b650d30f370c2b724812bee08008be0c4163b163ddaec3f2546c1caf65f191db"},
@@ -268,6 +279,7 @@ version = "3.4.1"
 description = "The Real First Universal Charset Detector. Open, modern and actively maintained alternative to Chardet."
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "charset_normalizer-3.4.1-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:91b36a978b5ae0ee86c394f5a54d6ef44db1de0815eb43de826d41d21e4af3de"},
    {file = "charset_normalizer-3.4.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:7461baadb4dc00fd9e0acbe254e3d7d2112e7f92ced2adc96e54ef6501c5f176"},
@@ -369,6 +381,7 @@ version = "8.1.8"
 description = "Composable command line interface toolkit"
 optional = false
 python-versions = ">=3.7"
+groups = ["main", "dev"]
 files = [
    {file = "click-8.1.8-py3-none-any.whl", hash = "sha256:63c132bbbed01578a06712a2d1f497bb62d9c1c0d329b7903a866228027263b2"},
    {file = "click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a"},
@@ -383,10 +396,12 @@ version = "0.4.6"
 description = "Cross-platform colored terminal text."
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,>=2.7"
+groups = ["main", "dev"]
 files = [
    {file = "colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6"},
    {file = "colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44"},
 ]
+markers = {main = "platform_system == \"Windows\"", dev = "platform_system == \"Windows\" or sys_platform == \"win32\""}

 [[package]]
 name = "distro"
@@ -394,6 +409,7 @@ version = "1.9.0"
 description = "Distro - an OS platform information API"
 optional = false
 python-versions = ">=3.6"
+groups = ["main"]
 files = [
    {file = "distro-1.9.0-py3-none-any.whl", hash = "sha256:7bffd925d65168f85027d8da9af6bddab658135b840670a223589bc0c8ef02b2"},
    {file = "distro-1.9.0.tar.gz", hash = "sha256:2fa77c6fd8940f116ee1d6b94a2f90b13b5ea8d019b98bc8bafdcabcdd9bdbed"},
@@ -405,6 +421,8 @@ version = "1.2.2"
 description = "Backport of PEP 654 (exception groups)"
 optional = false
 python-versions = ">=3.7"
+groups = ["main", "dev"]
+markers = "python_version == \"3.10\""
 files = [
    {file = "exceptiongroup-1.2.2-py3-none-any.whl", hash = "sha256:3111b9d131c238bec2f8f516e123e14ba243563fb135d3fe885990585aa7795b"},
    {file = "exceptiongroup-1.2.2.tar.gz", hash = "sha256:47c2edf7c6738fafb49fd34290706d1a1a2f4d1c6df275526b62cbb4aa5393cc"},
@@ -419,6 +437,7 @@ version = "0.115.6"
 description = "FastAPI framework, high performance, easy to learn, fast to code, ready for production"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "fastapi-0.115.6-py3-none-any.whl", hash = "sha256:e9240b29e36fa8f4bb7290316988e90c381e5092e0cbe84e7818cc3713bcf305"},
    {file = "fastapi-0.115.6.tar.gz", hash = "sha256:9ec46f7addc14ea472958a96aae5b5de65f39721a46aaf5705c480d9a8b76654"},
@@ -439,6 +458,7 @@ version = "3.18.0"
 description = "A platform independent file lock."
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "filelock-3.18.0-py3-none-any.whl", hash = "sha256:c401f4f8377c4464e6db25fff06205fd89bdd83b65eb0488ed1b160f780e21de"},
    {file = "filelock-3.18.0.tar.gz", hash = "sha256:adbc88eabb99d2fec8c9c1b229b171f18afa655400173ddc653d5d01501fb9f2"},
@@ -447,7 +467,7 @@ files = [
 [package.extras]
 docs = ["furo (>=2024.8.6)", "sphinx (>=8.1.3)", "sphinx-autodoc-typehints (>=3)"]
 testing = ["covdefaults (>=2.3)", "coverage (>=7.6.10)", "diff-cover (>=9.2.1)", "pytest (>=8.3.4)", "pytest-asyncio (>=0.25.2)", "pytest-cov (>=6)", "pytest-mock (>=3.14)", "pytest-timeout (>=2.3.1)", "virtualenv (>=20.28.1)"]
-typing = ["typing-extensions (>=4.12.2)"]
+typing = ["typing-extensions (>=4.12.2) ; python_version < \"3.11\""]

 [[package]]
 name = "folium"
@@ -455,6 +475,7 @@ version = "0.19.4"
 description = "Make beautiful maps with Leaflet.js & Python"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "folium-0.19.4-py2.py3-none-any.whl", hash = "sha256:bea5246b6a6aa61b96d1c51399dd63254bacbd6ba8a826eeb491f45242032dfd"},
    {file = "folium-0.19.4.tar.gz", hash = "sha256:431a655b52a9bf3efda336f2be022103f0106504a0599e7c349efbfd30bafda6"},
@@ -476,6 +497,7 @@ version = "1.6.0"
 description = "A list-like structure which implements collections.abc.MutableSequence"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "frozenlist-1.6.0-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:e6e558ea1e47fd6fa8ac9ccdad403e5dd5ecc6ed8dda94343056fa4277d5c65e"},
    {file = "frozenlist-1.6.0-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:f4b3cd7334a4bbc0c472164f3744562cb72d05002cc6fcf58adb104630bbc352"},
@@ -589,6 +611,7 @@ version = "2025.5.0"
 description = "File-system specification"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "fsspec-2025.5.0-py3-none-any.whl", hash = "sha256:0ca253eca6b5333d8a2b8bd98c7326fe821f1f0fdbd34e1b445bddde8e804c95"},
    {file = "fsspec-2025.5.0.tar.gz", hash = "sha256:e4f4623bb6221f7407fd695cc535d1f857a077eb247580f4ada34f5dc25fd5c8"},
@@ -628,6 +651,7 @@ version = "1.0.1"
 description = "Geographic pandas extensions"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "geopandas-1.0.1-py3-none-any.whl", hash = "sha256:01e147d9420cc374d26f51fc23716ac307f32b49406e4bd8462c07e82ed1d3d6"},
    {file = "geopandas-1.0.1.tar.gz", hash = "sha256:b8bf70a5534588205b7a56646e2082fb1de9a03599651b3d80c99ea4c2ca08ab"},
@@ -651,6 +675,7 @@ version = "10.1.1"
 description = "A Python library for analyzing GTFS feeds."
 optional = false
 python-versions = ">=3.10"
+groups = ["main"]
 files = [
    {file = "gtfs_kit-10.1.1-py3-none-any.whl", hash = "sha256:2a54982d30993c365ee082eb3f5dc981ecd89c294728199a1f39776dee6c71b2"},
    {file = "gtfs_kit-10.1.1.tar.gz", hash = "sha256:b94135883fbb4a5135b33d66215e12507a0480218f53df8c6a3a88ee359e7ab4"},
@@ -671,6 +696,7 @@ version = "0.14.0"
 description = "A pure-Python, bring-your-own-I/O implementation of HTTP/1.1"
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "h11-0.14.0-py3-none-any.whl", hash = "sha256:e3fe4ac4b851c468cc8363d500db52c2ead036020723024a109d37346efaa761"},
    {file = "h11-0.14.0.tar.gz", hash = "sha256:8f19fbbe99e72420ff35c00b27a34cb9937e902a8b810e2c88300c6f0a3b699d"},
@@ -682,6 +708,7 @@ version = "1.0.7"
 description = "A minimal low-level HTTP client."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "httpcore-1.0.7-py3-none-any.whl", hash = "sha256:a3fff8f43dc260d5bd363d9f9cf1830fa3a458b332856f34282de498ed420edd"},
    {file = "httpcore-1.0.7.tar.gz", hash = "sha256:8551cb62a169ec7162ac7be8d4817d561f60e08eaa485234898414bb5a8a0b4c"},
@@ -703,6 +730,7 @@ version = "0.27.2"
 description = "The next generation HTTP client."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "httpx-0.27.2-py3-none-any.whl", hash = "sha256:7bb2708e112d8fdd7829cd4243970f0c223274051cb35ee80c03301ee29a3df0"},
    {file = "httpx-0.27.2.tar.gz", hash = "sha256:f7c2be1d2f3c3c3160d441802406b206c2b76f5947b11115e6df10c6c65e66c2"},
@@ -716,7 +744,7 @@ idna = "*"
 sniffio = "*"

 [package.extras]
-brotli = ["brotli", "brotlicffi"]
+brotli = ["brotli ; platform_python_implementation == \"CPython\"", "brotlicffi ; platform_python_implementation != \"CPython\""]
 cli = ["click (==8.*)", "pygments (==2.*)", "rich (>=10,<14)"]
 http2 = ["h2 (>=3,<5)"]
 socks = ["socksio (==1.*)"]
@@ -728,6 +756,7 @@ version = "0.31.4"
 description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
 optional = false
 python-versions = ">=3.8.0"
+groups = ["main"]
 files = [
    {file = "huggingface_hub-0.31.4-py3-none-any.whl", hash = "sha256:4f70704760296cc69b612916056e9845f5490a33782b924fc531767967acc15d"},
    {file = "huggingface_hub-0.31.4.tar.gz", hash = "sha256:5a7bc710b9f9c028aee5b1476867b4ec5c1b92f043cb364d5fdc54354757e4ce"},
@@ -763,6 +792,7 @@ version = "3.10"
 description = "Internationalized Domain Names in Applications (IDNA)"
 optional = false
 python-versions = ">=3.6"
+groups = ["main"]
 files = [
    {file = "idna-3.10-py3-none-any.whl", hash = "sha256:946d195a0d259cbba61165e88e65941f16e9b36ea6ddb97f00452bae8b1287d3"},
    {file = "idna-3.10.tar.gz", hash = "sha256:12f65c9b470abda6dc35cf8e63cc574b1c52b11df2c86030af0ac09b01b13ea9"},
@@ -777,6 +807,7 @@ version = "8.7.0"
 description = "Read metadata from Python packages"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "importlib_metadata-8.7.0-py3-none-any.whl", hash = "sha256:e5dd1551894c77868a30651cef00984d50e1002d06942a7101d34870c5f02afd"},
    {file = "importlib_metadata-8.7.0.tar.gz", hash = "sha256:d13b81ad223b890aa16c5471f2ac3056cf76c5f10f82d6f9292f0b415f389000"},
@@ -786,12 +817,12 @@ files = [
 zipp = ">=3.20"

 [package.extras]
-check = ["pytest-checkdocs (>=2.4)", "pytest-ruff (>=0.2.1)"]
+check = ["pytest-checkdocs (>=2.4)", "pytest-ruff (>=0.2.1) ; sys_platform != \"cygwin\""]
 cover = ["pytest-cov"]
 doc = ["furo", "jaraco.packaging (>=9.3)", "jaraco.tidelift (>=1.4)", "rst.linker (>=1.9)", "sphinx (>=3.5)", "sphinx-lint"]
 enabler = ["pytest-enabler (>=2.2)"]
 perf = ["ipython"]
-test = ["flufl.flake8", "importlib_resources (>=1.3)", "jaraco.test (>=5.4)", "packaging", "pyfakefs", "pytest (>=6,!=8.1.*)", "pytest-perf (>=0.9.2)"]
+test = ["flufl.flake8", "importlib_resources (>=1.3) ; python_version < \"3.9\"", "jaraco.test (>=5.4)", "packaging", "pyfakefs", "pytest (>=6,!=8.1.*)", "pytest-perf (>=0.9.2)"]
 type = ["pytest-mypy"]

 [[package]]
@@ -800,6 +831,7 @@ version = "2.0.0"
 description = "brain-dead simple config-ini parsing"
 optional = false
 python-versions = ">=3.7"
+groups = ["dev"]
 files = [
    {file = "iniconfig-2.0.0-py3-none-any.whl", hash = "sha256:b6a85871a79d2e3b22d2d1b94ac2824226a63c6b741c88f7ae975f18b6778374"},
    {file = "iniconfig-2.0.0.tar.gz", hash = "sha256:2d91e135bf72d31a410b17c16da610a82cb55f6b0477d1a902134b24a455b8b3"},
@@ -811,6 +843,7 @@ version = "5.13.2"
 description = "A Python utility / library to sort Python imports."
 optional = false
 python-versions = ">=3.8.0"
+groups = ["dev"]
 files = [
    {file = "isort-5.13.2-py3-none-any.whl", hash = "sha256:8ca5e72a8d85860d5a3fa69b8745237f2939afe12dbf656afbcb47fe72d947a6"},
    {file = "isort-5.13.2.tar.gz", hash = "sha256:48fdfcb9face5d58a4f6dde2e72a1fb8dcaf8ab26f95ab49fab84c2ddefb0109"},
@@ -825,6 +858,7 @@ version = "3.1.5"
 description = "A very fast and expressive template engine."
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "jinja2-3.1.5-py3-none-any.whl", hash = "sha256:aba0f4dc9ed8013c424088f68a5c226f7d6097ed89b246d7749c2ec4175c6adb"},
    {file = "jinja2-3.1.5.tar.gz", hash = "sha256:8fefff8dc3034e27bb80d67c671eb8a9bc424c0ef4c0826edbff304cceff43bb"},
@@ -842,6 +876,7 @@ version = "0.8.2"
 description = "Fast iterable JSON parser."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "jiter-0.8.2-cp310-cp310-macosx_10_12_x86_64.whl", hash = "sha256:ca8577f6a413abe29b079bc30f907894d7eb07a865c4df69475e868d73e71c7b"},
    {file = "jiter-0.8.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:b25bd626bde7fb51534190c7e3cb97cee89ee76b76d7585580e22f34f5e3f393"},
@@ -927,6 +962,7 @@ version = "1.3.0"
 description = "JSON to HTML Table Representation"
 optional = false
 python-versions = "*"
+groups = ["main"]
 files = [
    {file = "json2html-1.3.0.tar.gz", hash = "sha256:8951a53662ae9cfd812685facdba693fc950ffc1c1fd1a8a2d3cf4c34600689c"},
 ]
@@ -937,6 +973,7 @@ version = "4.23.0"
 description = "An implementation of JSON Schema validation for Python"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "jsonschema-4.23.0-py3-none-any.whl", hash = "sha256:fbadb6f8b144a8f8cf9f0b89ba94501d143e50411a1278633f56a7acf7fd5566"},
    {file = "jsonschema-4.23.0.tar.gz", hash = "sha256:d71497fef26351a33265337fa77ffeb82423f3ea21283cd9467bb03999266bc4"},
@@ -958,6 +995,7 @@ version = "2025.4.1"
 description = "The JSON Schema meta-schemas and vocabularies, exposed as a Registry"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "jsonschema_specifications-2025.4.1-py3-none-any.whl", hash = "sha256:4653bffbd6584f7de83a67e0d620ef16900b390ddc7939d56684d6c81e33f1af"},
    {file = "jsonschema_specifications-2025.4.1.tar.gz", hash = "sha256:630159c9f4dbea161a6a2205c3011cc4f18ff381b189fff48bb39b9bf26ae608"},
@@ -972,6 +1010,7 @@ version = "1.70.0"
 description = "Library to easily interface with LLM API providers"
 optional = false
 python-versions = "!=2.7.*,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,!=3.7.*,>=3.8"
+groups = ["main"]
 files = [
    {file = "litellm-1.70.0-py3-none-any.whl", hash = "sha256:7e094057b38ddb1d77f61452895835aa5d376db1850e9a1bc0342c5631d89638"},
    {file = "litellm-1.70.0.tar.gz", hash = "sha256:357f3891e38f23a12f0932c235ed860dc41bc5880afaee7229e6d25318652706"},
@@ -991,8 +1030,8 @@ tiktoken = ">=0.7.0"
 tokenizers = "*"

 [package.extras]
-extra-proxy = ["azure-identity (>=1.15.0,<2.0.0)", "azure-keyvault-secrets (>=4.8.0,<5.0.0)", "google-cloud-kms (>=2.21.3,<3.0.0)", "prisma (==0.11.0)", "redisvl (>=0.4.1,<0.5.0)", "resend (>=0.8.0,<0.9.0)"]
-proxy = ["PyJWT (>=2.8.0,<3.0.0)", "apscheduler (>=3.10.4,<4.0.0)", "backoff", "boto3 (==1.34.34)", "cryptography (>=43.0.1,<44.0.0)", "fastapi (>=0.115.5,<0.116.0)", "fastapi-sso (>=0.16.0,<0.17.0)", "gunicorn (>=23.0.0,<24.0.0)", "litellm-enterprise (==0.1.3)", "litellm-proxy-extras (==0.1.21)", "mcp (==1.5.0)", "orjson (>=3.9.7,<4.0.0)", "pynacl (>=1.5.0,<2.0.0)", "python-multipart (>=0.0.18,<0.0.19)", "pyyaml (>=6.0.1,<7.0.0)", "rich (==13.7.1)", "rq", "uvicorn (>=0.29.0,<0.30.0)", "uvloop (>=0.21.0,<0.22.0)", "websockets (>=13.1.0,<14.0.0)"]
+extra-proxy = ["azure-identity (>=1.15.0,<2.0.0)", "azure-keyvault-secrets (>=4.8.0,<5.0.0)", "google-cloud-kms (>=2.21.3,<3.0.0)", "prisma (==0.11.0)", "redisvl (>=0.4.1,<0.5.0) ; python_version >= \"3.9\" and python_version < \"3.14\"", "resend (>=0.8.0,<0.9.0)"]
+proxy = ["PyJWT (>=2.8.0,<3.0.0)", "apscheduler (>=3.10.4,<4.0.0)", "backoff", "boto3 (==1.34.34)", "cryptography (>=43.0.1,<44.0.0)", "fastapi (>=0.115.5,<0.116.0)", "fastapi-sso (>=0.16.0,<0.17.0)", "gunicorn (>=23.0.0,<24.0.0)", "litellm-enterprise (==0.1.3)", "litellm-proxy-extras (==0.1.21)", "mcp (==1.5.0) ; python_version >= \"3.10\"", "orjson (>=3.9.7,<4.0.0)", "pynacl (>=1.5.0,<2.0.0)", "python-multipart (>=0.0.18,<0.0.19)", "pyyaml (>=6.0.1,<7.0.0)", "rich (==13.7.1)", "rq", "uvicorn (>=0.29.0,<0.30.0)", "uvloop (>=0.21.0,<0.22.0) ; sys_platform != \"win32\"", "websockets (>=13.1.0,<14.0.0)"]
 utils = ["numpydoc"]

 [[package]]
@@ -1001,6 +1040,7 @@ version = "3.0.2"
 description = "Safely add untrusted strings to HTML/XML markup."
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "MarkupSafe-3.0.2-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:7e94c425039cde14257288fd61dcfb01963e658efbc0ff54f5306b06054700f8"},
    {file = "MarkupSafe-3.0.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:9e2d922824181480953426608b81967de705c3cef4d1af983af849d7bd619158"},
@@ -1071,6 +1111,7 @@ version = "6.4.4"
 description = "multidict implementation"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "multidict-6.4.4-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:8adee3ac041145ffe4488ea73fa0a622b464cc25340d98be76924d0cda8545ff"},
    {file = "multidict-6.4.4-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:b61e98c3e2a861035aaccd207da585bdcacef65fe01d7a0d07478efac005e028"},
@@ -1181,66 +1222,13 @@ files = [
 [package.dependencies]
 typing-extensions = {version = ">=4.1.0", markers = "python_version < \"3.11\""}

-[[package]]
-name = "mypy"
-version = "1.16.0"
-description = "Optional static typing for Python"
-optional = false
-python-versions = ">=3.9"
-files = [
-    {file = "mypy-1.16.0-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:7909541fef256527e5ee9c0a7e2aeed78b6cda72ba44298d1334fe7881b05c5c"},
-    {file = "mypy-1.16.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:e71d6f0090c2256c713ed3d52711d01859c82608b5d68d4fa01a3fe30df95571"},
-    {file = "mypy-1.16.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:936ccfdd749af4766be824268bfe22d1db9eb2f34a3ea1d00ffbe5b5265f5491"},
-    {file = "mypy-1.16.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:4086883a73166631307fdd330c4a9080ce24913d4f4c5ec596c601b3a4bdd777"},
-    {file = "mypy-1.16.0-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:feec38097f71797da0231997e0de3a58108c51845399669ebc532c815f93866b"},
-    {file = "mypy-1.16.0-cp310-cp310-win_amd64.whl", hash = "sha256:09a8da6a0ee9a9770b8ff61b39c0bb07971cda90e7297f4213741b48a0cc8d93"},
-    {file = "mypy-1.16.0-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:9f826aaa7ff8443bac6a494cf743f591488ea940dd360e7dd330e30dd772a5ab"},
-    {file = "mypy-1.16.0-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:82d056e6faa508501af333a6af192c700b33e15865bda49611e3d7d8358ebea2"},
-    {file = "mypy-1.16.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:089bedc02307c2548eb51f426e085546db1fa7dd87fbb7c9fa561575cf6eb1ff"},
-    {file = "mypy-1.16.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:6a2322896003ba66bbd1318c10d3afdfe24e78ef12ea10e2acd985e9d684a666"},
-    {file = "mypy-1.16.0-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:021a68568082c5b36e977d54e8f1de978baf401a33884ffcea09bd8e88a98f4c"},
-    {file = "mypy-1.16.0-cp311-cp311-win_amd64.whl", hash = "sha256:54066fed302d83bf5128632d05b4ec68412e1f03ef2c300434057d66866cea4b"},
-    {file = "mypy-1.16.0-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:c5436d11e89a3ad16ce8afe752f0f373ae9620841c50883dc96f8b8805620b13"},
-    {file = "mypy-1.16.0-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:f2622af30bf01d8fc36466231bdd203d120d7a599a6d88fb22bdcb9dbff84090"},
-    {file = "mypy-1.16.0-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:d045d33c284e10a038f5e29faca055b90eee87da3fc63b8889085744ebabb5a1"},
-    {file = "mypy-1.16.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:b4968f14f44c62e2ec4a038c8797a87315be8df7740dc3ee8d3bfe1c6bf5dba8"},
-    {file = "mypy-1.16.0-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:eb14a4a871bb8efb1e4a50360d4e3c8d6c601e7a31028a2c79f9bb659b63d730"},
-    {file = "mypy-1.16.0-cp312-cp312-win_amd64.whl", hash = "sha256:bd4e1ebe126152a7bbaa4daedd781c90c8f9643c79b9748caa270ad542f12bec"},
-    {file = "mypy-1.16.0-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:a9e056237c89f1587a3be1a3a70a06a698d25e2479b9a2f57325ddaaffc3567b"},
-    {file = "mypy-1.16.0-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:0b07e107affb9ee6ce1f342c07f51552d126c32cd62955f59a7db94a51ad12c0"},
-    {file = "mypy-1.16.0-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:c6fb60cbd85dc65d4d63d37cb5c86f4e3a301ec605f606ae3a9173e5cf34997b"},
-    {file = "mypy-1.16.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:a7e32297a437cc915599e0578fa6bc68ae6a8dc059c9e009c628e1c47f91495d"},
-    {file = "mypy-1.16.0-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:afe420c9380ccec31e744e8baff0d406c846683681025db3531b32db56962d52"},
-    {file = "mypy-1.16.0-cp313-cp313-win_amd64.whl", hash = "sha256:55f9076c6ce55dd3f8cd0c6fff26a008ca8e5131b89d5ba6d86bd3f47e736eeb"},
-    {file = "mypy-1.16.0-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:f56236114c425620875c7cf71700e3d60004858da856c6fc78998ffe767b73d3"},
-    {file = "mypy-1.16.0-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:15486beea80be24ff067d7d0ede673b001d0d684d0095803b3e6e17a886a2a92"},
-    {file = "mypy-1.16.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:f2ed0e0847a80655afa2c121835b848ed101cc7b8d8d6ecc5205aedc732b1436"},
-    {file = "mypy-1.16.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:eb5fbc8063cb4fde7787e4c0406aa63094a34a2daf4673f359a1fb64050e9cb2"},
-    {file = "mypy-1.16.0-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:a5fcfdb7318c6a8dd127b14b1052743b83e97a970f0edb6c913211507a255e20"},
-    {file = "mypy-1.16.0-cp39-cp39-win_amd64.whl", hash = "sha256:2e7e0ad35275e02797323a5aa1be0b14a4d03ffdb2e5f2b0489fa07b89c67b21"},
-    {file = "mypy-1.16.0-py3-none-any.whl", hash = "sha256:29e1499864a3888bca5c1542f2d7232c6e586295183320caa95758fc84034031"},
-    {file = "mypy-1.16.0.tar.gz", hash = "sha256:84b94283f817e2aa6350a14b4a8fb2a35a53c286f97c9d30f53b63620e7af8ab"},
-]
-
-[package.dependencies]
-mypy_extensions = ">=1.0.0"
-pathspec = ">=0.9.0"
-tomli = {version = ">=1.1.0", markers = "python_version < \"3.11\""}
-typing_extensions = ">=4.6.0"
-
-[package.extras]
-dmypy = ["psutil (>=4.0)"]
-faster-cache = ["orjson"]
-install-types = ["pip"]
-mypyc = ["setuptools (>=50)"]
-reports = ["lxml"]
-
 [[package]]
 name = "mypy-extensions"
 version = "1.0.0"
 description = "Type system extensions for programs checked with the mypy type checker."
 optional = false
 python-versions = ">=3.5"
+groups = ["dev"]
 files = [
    {file = "mypy_extensions-1.0.0-py3-none-any.whl", hash = "sha256:4392f6c0eb8a5668a69e23d168ffa70f0be9ccfd32b5cc2d26a34ae5b844552d"},
    {file = "mypy_extensions-1.0.0.tar.gz", hash = "sha256:75dbf8955dc00442a438fc4d0666508a9a97b6bd41aa2f0ffe9d2f2725af0782"},
@@ -1252,6 +1240,7 @@ version = "2.2.2"
 description = "Fundamental package for array computing in Python"
 optional = false
 python-versions = ">=3.10"
+groups = ["main"]
 files = [
    {file = "numpy-2.2.2-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:7079129b64cb78bdc8d611d1fd7e8002c0a2565da6a47c4df8062349fee90e3e"},
    {file = "numpy-2.2.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:2ec6c689c61df613b783aeb21f945c4cbe6c51c28cb70aae8430577ab39f163e"},
@@ -1316,6 +1305,7 @@ version = "1.75.0"
 description = "The official Python library for the openai API"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "openai-1.75.0-py3-none-any.whl", hash = "sha256:fe6f932d2ded3b429ff67cc9ad118c71327db32eb9d32dd723de3acfca337125"},
    {file = "openai-1.75.0.tar.gz", hash = "sha256:fb3ea907efbdb1bcfd0c44507ad9c961afd7dce3147292b54505ecfd17be8fd1"},
@@ -1342,6 +1332,7 @@ version = "24.2"
 description = "Core utilities for Python packages"
 optional = false
 python-versions = ">=3.8"
+groups = ["main", "dev"]
 files = [
    {file = "packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759"},
    {file = "packaging-24.2.tar.gz", hash = "sha256:c228a6dc5e932d346bc5739379109d49e8853dd8223571c7c5b55260edc0b97f"},
@@ -1353,6 +1344,7 @@ version = "2.2.3"
 description = "Powerful data structures for data analysis, time series, and statistics"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "pandas-2.2.3-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:1948ddde24197a0f7add2bdc4ca83bf2b1ef84a1bc8ccffd95eda17fd836ecb5"},
    {file = "pandas-2.2.3-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:381175499d3802cde0eabbaf6324cce0c4f5d52ca6f8c377c29ad442f50f6348"},
@@ -1439,6 +1431,7 @@ version = "0.12.1"
 description = "Utility library for gitignore style pattern matching of file paths."
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
 files = [
    {file = "pathspec-0.12.1-py3-none-any.whl", hash = "sha256:a0d503e138a4c123b27490a4f7beda6a01c6f288df0e4a8b79c7eb0dc7b4cc08"},
    {file = "pathspec-0.12.1.tar.gz", hash = "sha256:a482d51503a1ab33b1c67a6c3813a26953dbdc71c31dacaef9a838c4e29f5712"},
@@ -1450,6 +1443,7 @@ version = "4.3.6"
 description = "A small Python package for determining appropriate platform-specific dirs, e.g. a `user data dir`."
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
 files = [
    {file = "platformdirs-4.3.6-py3-none-any.whl", hash = "sha256:73e575e1408ab8103900836b97580d5307456908a03e92031bab39e4554cc3fb"},
    {file = "platformdirs-4.3.6.tar.gz", hash = "sha256:357fb2acbc885b0419afd3ce3ed34564c13c9b95c89360cd9563f73aa5e2b907"},
@@ -1466,6 +1460,7 @@ version = "1.5.0"
 description = "plugin and hook calling mechanisms for python"
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
 files = [
    {file = "pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669"},
    {file = "pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1"},
@@ -1481,6 +1476,7 @@ version = "0.3.1"
 description = "Accelerated property cache"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "propcache-0.3.1-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:f27785888d2fdd918bc36de8b8739f2d6c791399552333721b58193f68ea3e98"},
    {file = "propcache-0.3.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:d4e89cde74154c7b5957f87a355bb9c8ec929c167b59c83d90654ea36aeb6180"},
@@ -1588,6 +1584,7 @@ version = "5.29.2"
 description = ""
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "protobuf-5.29.2-cp310-abi3-win32.whl", hash = "sha256:c12ba8249f5624300cf51c3d0bfe5be71a60c63e4dcf51ffe9a68771d958c851"},
    {file = "protobuf-5.29.2-cp310-abi3-win_amd64.whl", hash = "sha256:842de6d9241134a973aab719ab42b008a18a90f9f07f06ba480df268f86432f9"},
@@ -1608,6 +1605,7 @@ version = "2.10.4"
 description = "Data validation using Python type hints"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "pydantic-2.10.4-py3-none-any.whl", hash = "sha256:597e135ea68be3a37552fb524bc7d0d66dcf93d395acd93a00682f1efcb8ee3d"},
    {file = "pydantic-2.10.4.tar.gz", hash = "sha256:82f12e9723da6de4fe2ba888b5971157b3be7ad914267dea8f05f82b28254f06"},
@@ -1620,7 +1618,7 @@ typing-extensions = ">=4.12.2"

 [package.extras]
 email = ["email-validator (>=2.0.0)"]
-timezone = ["tzdata"]
+timezone = ["tzdata ; python_version >= \"3.9\" and platform_system == \"Windows\""]

 [[package]]
 name = "pydantic-core"
@@ -1628,6 +1626,7 @@ version = "2.27.2"
 description = "Core functionality for Pydantic validation and serialization"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "pydantic_core-2.27.2-cp310-cp310-macosx_10_12_x86_64.whl", hash = "sha256:2d367ca20b2f14095a8f4fa1210f5a7b78b8a20009ecced6b12818f455b1e9fa"},
    {file = "pydantic_core-2.27.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:491a2b73db93fab69731eaee494f320faa4e093dbed776be1a829c2eb222c34c"},
@@ -1740,6 +1739,7 @@ version = "0.10.0"
 description = "Vectorized spatial vector file format I/O using GDAL/OGR"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "pyogrio-0.10.0-cp310-cp310-macosx_12_0_arm64.whl", hash = "sha256:046eeeae12a03a3ebc3dc5ff5a87664e4f5fc0a4fb1ea5d5c45d547fa941072b"},
    {file = "pyogrio-0.10.0-cp310-cp310-macosx_12_0_x86_64.whl", hash = "sha256:44380f4d9245c776f432526e29ce4d29238aea26adad991803c4f453474f51d3"},
@@ -1791,6 +1791,7 @@ version = "3.7.0"
 description = "Python interface to PROJ (cartographic projections and coordinate transformations library)"
 optional = false
 python-versions = ">=3.10"
+groups = ["main"]
 files = [
    {file = "pyproj-3.7.0-cp310-cp310-macosx_12_0_x86_64.whl", hash = "sha256:d5c7e7d24b967e328a5efd013f466804a1f226d1106ac7efc47dcc99360dbc8f"},
    {file = "pyproj-3.7.0-cp310-cp310-macosx_14_0_arm64.whl", hash = "sha256:448958c46bd3fe2da91c89ba551ac5835e63073ca861422c6eb1af89979dfab1"},
@@ -1828,6 +1829,7 @@ version = "8.3.5"
 description = "pytest: simple powerful testing with Python"
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
 files = [
    {file = "pytest-8.3.5-py3-none-any.whl", hash = "sha256:c69214aa47deac29fad6c2a4f590b9c4a9fdb16a403176fe154b79c0b4d4d820"},
    {file = "pytest-8.3.5.tar.gz", hash = "sha256:f4efe70cc14e511565ac476b57c279e12a855b11f48f212af1080ef2263d3845"},
@@ -1850,6 +1852,7 @@ version = "0.26.0"
 description = "Pytest support for asyncio"
 optional = false
 python-versions = ">=3.9"
+groups = ["dev"]
 files = [
    {file = "pytest_asyncio-0.26.0-py3-none-any.whl", hash = "sha256:7b51ed894f4fbea1340262bdae5135797ebbe21d8638978e35d31c6d19f72fb0"},
    {file = "pytest_asyncio-0.26.0.tar.gz", hash = "sha256:c4df2a697648241ff39e7f0e4a73050b03f123f760673956cf0d72a4990e312f"},
@@ -1868,6 +1871,7 @@ version = "2.9.0.post0"
 description = "Extensions to the standard Python datetime module"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,>=2.7"
+groups = ["main"]
 files = [
    {file = "python-dateutil-2.9.0.post0.tar.gz", hash = "sha256:37dd54208da7e1cd875388217d5e00ebd4179249f90fb72437e91a35459a0ad3"},
    {file = "python_dateutil-2.9.0.post0-py2.py3-none-any.whl", hash = "sha256:a8b2bc7bffae282281c8140a97d3aa9c14da0b136dfe83f850eea9a5f7470427"},
@@ -1882,6 +1886,7 @@ version = "1.0.1"
 description = "Read key-value pairs from a .env file and set them as environment variables"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "python-dotenv-1.0.1.tar.gz", hash = "sha256:e324ee90a023d808f1959c46bcbc04446a10ced277783dc6ee09987c37ec10ca"},
    {file = "python_dotenv-1.0.1-py3-none-any.whl", hash = "sha256:f7b63ef50f1b690dddf550d03497b66d609393b40b564ed0d674909a68ebf16a"},
@@ -1896,6 +1901,7 @@ version = "2025.1"
 description = "World timezone definitions, modern and historical"
 optional = false
 python-versions = "*"
+groups = ["main"]
 files = [
    {file = "pytz-2025.1-py2.py3-none-any.whl", hash = "sha256:89dd22dca55b46eac6eda23b2d72721bf1bdfef212645d81513ef5d03038de57"},
    {file = "pytz-2025.1.tar.gz", hash = "sha256:c2db42be2a2518b28e65f9207c4d05e6ff547d1efa4086469ef855e4ab70178e"},
@@ -1907,6 +1913,7 @@ version = "6.0.2"
 description = "YAML parser and emitter for Python"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "PyYAML-6.0.2-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:0a9a2848a5b7feac301353437eb7d5957887edbf81d56e903999a75a3d743086"},
    {file = "PyYAML-6.0.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:29717114e51c84ddfba879543fb232a6ed60086602313ca38cce623c1d62cfbf"},
@@ -1969,6 +1976,7 @@ version = "0.36.2"
 description = "JSON Referencing + Python"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "referencing-0.36.2-py3-none-any.whl", hash = "sha256:e8699adbbf8b5c7de96d8ffa0eb5c158b3beafce084968e2ea8bb08c6794dcd0"},
    {file = "referencing-0.36.2.tar.gz", hash = "sha256:df2e89862cd09deabbdba16944cc3f10feb6b3e6f18e902f7cc25609a34775aa"},
@@ -1985,6 +1993,7 @@ version = "2024.11.6"
 description = "Alternative regular expression module, to replace re."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "regex-2024.11.6-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:ff590880083d60acc0433f9c3f713c51f7ac6ebb9adf889c79a261ecf541aa91"},
    {file = "regex-2024.11.6-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:658f90550f38270639e83ce492f27d2c8d2cd63805c65a13a14d36ca126753f0"},
@@ -2088,6 +2097,7 @@ version = "2.32.3"
 description = "Python HTTP for Humans."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "requests-2.32.3-py3-none-any.whl", hash = "sha256:70761cfe03c773ceb22aa2f671b4757976145175cdfca038c02654d061d6dcc6"},
    {file = "requests-2.32.3.tar.gz", hash = "sha256:55365417734eb18255590a9ff9eb97e9e1da868d4ccd6402399eaf68af20a760"},
@@ -2109,6 +2119,7 @@ version = "0.25.0"
 description = "Python bindings to Rust's persistent data structures (rpds)"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "rpds_py-0.25.0-cp310-cp310-macosx_10_12_x86_64.whl", hash = "sha256:c146a24a8f0dc4a7846fb4640b88b3a68986585b8ce8397af15e66b7c5817439"},
    {file = "rpds_py-0.25.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:77814c7a4e1dc43fba73aeb4c1ef0fe37d901f3aa869a4823de5ea843a283fd0"},
@@ -2232,6 +2243,7 @@ version = "1.3.0"
 description = "R-Tree spatial index for Python GIS"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "Rtree-1.3.0-py3-none-macosx_10_9_x86_64.whl", hash = "sha256:80879d9db282a2273ca3a0d896c84583940e9777477727a277624ebfd424c517"},
    {file = "Rtree-1.3.0-py3-none-macosx_11_0_arm64.whl", hash = "sha256:4328e9e421797c347e6eb08efbbade962fe3664ebd60c1dffe82c40911b1e125"},
@@ -2251,6 +2263,7 @@ version = "2.0.7"
 description = "Manipulation and analysis of geometric objects"
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "shapely-2.0.7-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:33fb10e50b16113714ae40adccf7670379e9ccf5b7a41d0002046ba2b8f0f691"},
    {file = "shapely-2.0.7-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:f44eda8bd7a4bccb0f281264b34bf3518d8c4c9a8ffe69a1a05dabf6e8461147"},
@@ -2309,6 +2322,7 @@ version = "1.17.0"
 description = "Python 2 and 3 compatibility utilities"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,>=2.7"
+groups = ["main"]
 files = [
    {file = "six-1.17.0-py2.py3-none-any.whl", hash = "sha256:4721f391ed90541fddacab5acf947aa0d3dc7d27b2e1e8eda2be8970586c3274"},
    {file = "six-1.17.0.tar.gz", hash = "sha256:ff70335d468e7eb6ec65b95b99d3a2836546063f63acc5171de367e834932a81"},
@@ -2320,6 +2334,7 @@ version = "1.3.1"
 description = "Sniff out which async library your code is running under"
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "sniffio-1.3.1-py3-none-any.whl", hash = "sha256:2f6da418d1f1e0fddd844478f41680e794e6051915791a034ff65e5f100525a2"},
    {file = "sniffio-1.3.1.tar.gz", hash = "sha256:f4324edc670a0f49750a81b895f35c3adb843cca46f0530f79fc1babb23789dc"},
@@ -2331,6 +2346,7 @@ version = "0.41.3"
 description = "The little ASGI library that shines."
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "starlette-0.41.3-py3-none-any.whl", hash = "sha256:44cedb2b7c77a9de33a8b74b2b90e9f50d11fcf25d8270ea525ad71a25374ff7"},
    {file = "starlette-0.41.3.tar.gz", hash = "sha256:0e4ab3d16522a255be6b28260b938eae2482f98ce5cc934cb08dce8dc3ba5835"},
@@ -2342,27 +2358,13 @@ anyio = ">=3.4.0,<5"
 [package.extras]
 full = ["httpx (>=0.22.0)", "itsdangerous", "jinja2", "python-multipart (>=0.0.7)", "pyyaml"]

-[[package]]
-name = "stripe"
-version = "11.6.0"
-description = "Python bindings for the Stripe API"
-optional = false
-python-versions = ">=3.6"
-files = [
-    {file = "stripe-11.6.0-py2.py3-none-any.whl", hash = "sha256:6e6cf09ebb6d5fc2d708401cb8868fd7bff987a6d09a0433caaa92c62f97dbc5"},
-    {file = "stripe-11.6.0.tar.gz", hash = "sha256:0ced7cce23a6cb1a393c86a1f7f9435c9d83ae7cbd556362868caf62cb44a92c"},
-]
-
-[package.dependencies]
-requests = {version = ">=2.20", markers = "python_version >= \"3.0\""}
-typing-extensions = {version = ">=4.5.0", markers = "python_version >= \"3.7\""}
-
 [[package]]
 name = "temporalio"
 version = "1.9.0"
 description = "Temporal.io Python SDK"
 optional = false
 python-versions = "<4.0,>=3.8"
+groups = ["main"]
 files = [
    {file = "temporalio-1.9.0-cp38-abi3-macosx_10_9_x86_64.whl", hash = "sha256:ee941702e8925e2c018b5c2d7b296f811205043654d7f9c4564d7fa6597f1989"},
    {file = "temporalio-1.9.0-cp38-abi3-macosx_11_0_arm64.whl", hash = "sha256:101040090238d97b61d769e009f732409894d8f26596a3827662f2dde2862097"},
@@ -2388,6 +2390,7 @@ version = "0.9.0"
 description = "tiktoken is a fast BPE tokeniser for use with OpenAI's models"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "tiktoken-0.9.0-cp310-cp310-macosx_10_12_x86_64.whl", hash = "sha256:586c16358138b96ea804c034b8acf3f5d3f0258bd2bc3b0227af4af5d622e382"},
    {file = "tiktoken-0.9.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:d9c59ccc528c6c5dd51820b3474402f69d9a9e1d656226848ad68a8d5b2e5108"},
@@ -2435,6 +2438,7 @@ version = "0.21.1"
 description = ""
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "tokenizers-0.21.1-cp39-abi3-macosx_10_12_x86_64.whl", hash = "sha256:e78e413e9e668ad790a29456e677d9d3aa50a9ad311a40905d6861ba7692cf41"},
    {file = "tokenizers-0.21.1-cp39-abi3-macosx_11_0_arm64.whl", hash = "sha256:cd51cd0a91ecc801633829fcd1fda9cf8682ed3477c6243b9a095539de4aecf3"},
@@ -2467,6 +2471,8 @@ version = "2.2.1"
 description = "A lil' TOML parser"
 optional = false
 python-versions = ">=3.8"
+groups = ["dev"]
+markers = "python_version == \"3.10\""
 files = [
    {file = "tomli-2.2.1-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:678e4fa69e4575eb77d103de3df8a895e1591b48e740211bd1067378c69e8249"},
    {file = "tomli-2.2.1-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:023aa114dd824ade0100497eb2318602af309e5a55595f76b626d6d9f3b7b0a6"},
@@ -2508,6 +2514,7 @@ version = "4.67.1"
 description = "Fast, Extensible Progress Meter"
 optional = false
 python-versions = ">=3.7"
+groups = ["main"]
 files = [
    {file = "tqdm-4.67.1-py3-none-any.whl", hash = "sha256:26445eca388f82e72884e0d580d5464cd801a3ea01e63e5601bdff9ba6a48de2"},
    {file = "tqdm-4.67.1.tar.gz", hash = "sha256:f8aef9c52c08c13a65f30ea34f4e5aac3fd1a34959879d7e59e63027286627f2"},
@@ -2529,6 +2536,7 @@ version = "5.29.1.20241207"
 description = "Typing stubs for protobuf"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "types_protobuf-5.29.1.20241207-py3-none-any.whl", hash = "sha256:92893c42083e9b718c678badc0af7a9a1307b92afe1599e5cba5f3d35b668b2f"},
    {file = "types_protobuf-5.29.1.20241207.tar.gz", hash = "sha256:2ebcadb8ab3ef2e3e2f067e0882906d64ba0dc65fc5b0fd7a8b692315b4a0be9"},
@@ -2540,10 +2548,12 @@ version = "4.12.2"
 description = "Backported and Experimental Type Hints for Python 3.8+"
 optional = false
 python-versions = ">=3.8"
+groups = ["main", "dev"]
 files = [
    {file = "typing_extensions-4.12.2-py3-none-any.whl", hash = "sha256:04e5ca0351e0f3f85c6853954072df659d0d13fac324d0072316b67d7794700d"},
    {file = "typing_extensions-4.12.2.tar.gz", hash = "sha256:1a7ead55c7e559dd4dee8856e3a88b41225abfe1ce8df57b7c13915fe121ffb8"},
 ]
+markers = {dev = "python_version == \"3.10\""}

 [[package]]
 name = "tzdata"
@@ -2551,6 +2561,7 @@ version = "2025.1"
 description = "Provider of IANA time zone data"
 optional = false
 python-versions = ">=2"
+groups = ["main"]
 files = [
    {file = "tzdata-2025.1-py2.py3-none-any.whl", hash = "sha256:7e127113816800496f027041c570f50bcd464a020098a3b6b199517772303639"},
    {file = "tzdata-2025.1.tar.gz", hash = "sha256:24894909e88cdb28bd1636c6887801df64cb485bd593f2fd83ef29075a81d694"},
@@ -2562,13 +2573,14 @@ version = "2.3.0"
 description = "HTTP library with thread-safe connection pooling, file post, and more."
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "urllib3-2.3.0-py3-none-any.whl", hash = "sha256:1cee9ad369867bfdbbb48b7dd50374c0967a0bb7710050facf0dd6911440e3df"},
    {file = "urllib3-2.3.0.tar.gz", hash = "sha256:f8c5449b3cf0861679ce7e0503c7b44b5ec981bec0d1d3795a07f1ba96f0204d"},
 ]

 [package.extras]
-brotli = ["brotli (>=1.0.9)", "brotlicffi (>=0.8.0)"]
+brotli = ["brotli (>=1.0.9) ; platform_python_implementation == \"CPython\"", "brotlicffi (>=0.8.0) ; platform_python_implementation != \"CPython\""]
 h2 = ["h2 (>=4,<5)"]
 socks = ["pysocks (>=1.5.6,!=1.5.7,<2.0)"]
 zstd = ["zstandard (>=0.18.0)"]
@@ -2579,6 +2591,7 @@ version = "0.34.0"
 description = "The lightning-fast ASGI server."
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "uvicorn-0.34.0-py3-none-any.whl", hash = "sha256:023dc038422502fa28a09c7a30bf2b6991512da7dcdb8fd35fe57cfc154126f4"},
    {file = "uvicorn-0.34.0.tar.gz", hash = "sha256:404051050cd7e905de2c9a7e61790943440b3416f49cb409f965d9dcd0fa73e9"},
@@ -2590,7 +2603,7 @@ h11 = ">=0.8"
 typing-extensions = {version = ">=4.0", markers = "python_version < \"3.11\""}

 [package.extras]
-standard = ["colorama (>=0.4)", "httptools (>=0.6.3)", "python-dotenv (>=0.13)", "pyyaml (>=5.1)", "uvloop (>=0.14.0,!=0.15.0,!=0.15.1)", "watchfiles (>=0.13)", "websockets (>=10.4)"]
+standard = ["colorama (>=0.4) ; sys_platform == \"win32\"", "httptools (>=0.6.3)", "python-dotenv (>=0.13)", "pyyaml (>=5.1)", "uvloop (>=0.14.0,!=0.15.0,!=0.15.1) ; sys_platform != \"win32\" and sys_platform != \"cygwin\" and platform_python_implementation != \"PyPy\"", "watchfiles (>=0.13)", "websockets (>=10.4)"]

 [[package]]
 name = "xyzservices"
@@ -2598,6 +2611,7 @@ version = "2025.1.0"
 description = "Source of XYZ tiles providers"
 optional = false
 python-versions = ">=3.8"
+groups = ["main"]
 files = [
    {file = "xyzservices-2025.1.0-py3-none-any.whl", hash = "sha256:fa599956c5ab32dad1689960b3bb08fdcdbe0252cc82d84fc60ae415dc648907"},
    {file = "xyzservices-2025.1.0.tar.gz", hash = "sha256:5cdbb0907c20be1be066c6e2dc69c645842d1113a4e83e642065604a21f254ba"},
@@ -2609,6 +2623,7 @@ version = "1.20.0"
 description = "Yet another URL library"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "yarl-1.20.0-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:f1f6670b9ae3daedb325fa55fbe31c22c8228f6e0b513772c2e1c623caa6ab22"},
    {file = "yarl-1.20.0-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:85a231fa250dfa3308f3c7896cc007a47bc76e9e8e8595c20b7426cac4884c62"},
@@ -2727,20 +2742,21 @@ version = "3.21.0"
 description = "Backport of pathlib-compatible object wrapper for zip files"
 optional = false
 python-versions = ">=3.9"
+groups = ["main"]
 files = [
    {file = "zipp-3.21.0-py3-none-any.whl", hash = "sha256:ac1bbe05fd2991f160ebce24ffbac5f6d11d83dc90891255885223d42b3cd931"},
    {file = "zipp-3.21.0.tar.gz", hash = "sha256:2c9958f6430a2040341a52eb608ed6dd93ef4392e02ffe219417c1b28b5dd1f4"},
 ]

 [package.extras]
-check = ["pytest-checkdocs (>=2.4)", "pytest-ruff (>=0.2.1)"]
+check = ["pytest-checkdocs (>=2.4)", "pytest-ruff (>=0.2.1) ; sys_platform != \"cygwin\""]
 cover = ["pytest-cov"]
 doc = ["furo", "jaraco.packaging (>=9.3)", "jaraco.tidelift (>=1.4)", "rst.linker (>=1.9)", "sphinx (>=3.5)", "sphinx-lint"]
 enabler = ["pytest-enabler (>=2.2)"]
-test = ["big-O", "importlib-resources", "jaraco.functools", "jaraco.itertools", "jaraco.test", "more-itertools", "pytest (>=6,!=8.1.*)", "pytest-ignore-flaky"]
+test = ["big-O", "importlib-resources ; python_version < \"3.9\"", "jaraco.functools", "jaraco.itertools", "jaraco.test", "more-itertools", "pytest (>=6,!=8.1.*)", "pytest-ignore-flaky"]
 type = ["pytest-mypy"]

 [metadata]
-lock-version = "2.0"
+lock-version = "2.1"
 python-versions = ">=3.10,<4.0"
-content-hash = "b391df89fabb111e4dd5d65a52a9db3a0bf9d95d5473e77cd0946beb940cf26f"
+content-hash = "d9a981a06732679724e07fb4f74b03fc6290f17a3b85bea3356ef72335be377d"
--- a/prompts/agent_prompt_generators.py
+++ b/prompts/agent_prompt_generators.py
@@ -1,7 +1,6 @@
-import json
-from typing import Optional
-
 from models.tool_definitions import AgentGoal
+from typing import Optional
+import json

 MULTI_GOAL_MODE: bool = None

--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ authors = [
 ]
 readme = "README.md"

-# By default, Poetry will find packages automatically,
+# By default, Poetry will find packages automatically, 
 # but explicitly including them is fine:
 packages = [
  { include = "**/*.py", from = "." }
@@ -38,7 +38,6 @@ uvicorn = "^0.34.0"
 python-dotenv = "^1.0.1"
 requests = "^2.32.3"
 pandas = "^2.2.3"
-stripe = "^11.4.1"
 gtfs-kit = "^10.1.1"

 [tool.poetry.group.dev.dependencies]
@@ -46,7 +45,6 @@ pytest = ">=8.2"
 pytest-asyncio = "^0.26.0"
 black = "^23.7"
 isort = "^5.12"
-mypy = "^1.16.0"

 [build-system]
 requires = ["poetry-core>=1.4.0"]
@@ -57,16 +55,4 @@ asyncio_mode = "auto"
 log_cli = true
 log_cli_level = "INFO"
 log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
-asyncio_default_fixture_loop_scope = "function"
-norecursedirs = ["vibe"]
-
-[tool.mypy]
-python_version = "3.10"
-ignore_missing_imports = true
-check_untyped_defs = true
-namespace_packages = true
-explicit_package_bases = true
-ignore_errors = true
-
-[tool.isort]
-profile = "black"
+asyncio_default_fixture_loop_scope = "function"
--- a/scripts/end_chat.py
+++ b/scripts/end_chat.py
@@ -1,12 +1,12 @@
 import asyncio

-from shared.config import get_temporal_client
+
 from workflows.agent_goal_workflow import AgentGoalWorkflow


 async def main():
    # Create client connected to server at the given address
-    client = await get_temporal_client()
+    client = await Client.connect("localhost:7233")

    workflow_id = "agent-workflow"

--- a/scripts/find_events_test.py
+++ b/scripts/find_events_test.py
@@ -1,6 +1,5 @@
-import json
-
 from tools.search_flights import search_flights
+import json

 # Example usage
 if __name__ == "__main__":
--- a/scripts/flight_api_test.py
+++ b/scripts/flight_api_test.py
@@ -1,6 +1,5 @@
-import json
-
 from tools.search_flights import search_flights
+import json

 if __name__ == "__main__":
    # Suppose user typed "new" for New York, "lon" for London
--- a/scripts/run_legacy_worker.py
+++ b/scripts/run_legacy_worker.py
@@ -1,10 +1,12 @@
 import asyncio
+
 import concurrent.futures

 from temporalio.worker import Worker

 from activities.tool_activities import dynamic_tool_activity
-from shared.config import TEMPORAL_LEGACY_TASK_QUEUE, get_temporal_client
+
+from shared.config import get_temporal_client, TEMPORAL_LEGACY_TASK_QUEUE


 async def main():
@@ -22,9 +24,7 @@ async def main():
            activity_executor=activity_executor,
        )

-        print(
-            f"Starting legacy worker, connecting to task queue: {TEMPORAL_LEGACY_TASK_QUEUE}"
-        )
+        print(f"Starting legacy worker, connecting to task queue: {TEMPORAL_LEGACY_TASK_QUEUE}")
        await worker.run()


--- a/scripts/run_worker.py
+++ b/scripts/run_worker.py
@@ -1,15 +1,16 @@
 import asyncio
 import concurrent.futures
-import logging
 import os
-
 from dotenv import load_dotenv
+import logging
+
 from temporalio.worker import Worker

 from activities.tool_activities import ToolActivities, dynamic_tool_activity
-from shared.config import TEMPORAL_TASK_QUEUE, get_temporal_client
 from workflows.agent_goal_workflow import AgentGoalWorkflow

+from shared.config import get_temporal_client, TEMPORAL_TASK_QUEUE
+

 async def main():
    # Load environment variables
--- a/scripts/send_confirm.py
+++ b/scripts/send_confirm.py
@@ -5,6 +5,7 @@ from shared.config import get_temporal_client


 async def main():
+
    # Connect to Temporal and signal the workflow
    client = await get_temporal_client()

--- a/setup.md
+++ b/setup.md
@@ -47,7 +47,7 @@ If you prefer to run commands manually, follow these steps:

 ### Agent Goal Configuration

-The agent can be configured to pursue different goals using the `AGENT_GOAL` environment variable in your `.env` file. If unset, default is `goal_choose_agent_type`.
+The agent can be configured to pursue different goals using the `AGENT_GOAL` environment variable in your `.env` file. If unset, default is `goal_choose_agent_type`. 

 If the first goal is `goal_choose_agent_type` the agent will support multiple goals using goal categories defined by `GOAL_CATEGORIES` in your .env file. If unset, default is all. We recommend starting with `fin`.
 ```bash
@@ -60,7 +60,7 @@ See the section Goal-Specific Tool Configuration below for tool configuration fo

 Note: We recommend using OpenAI's GPT-4o or Claude 3.5 Sonnet for the best results. There can be significant differences in performance and capabilities between models, especially for complex tasks.

-The agent uses LiteLLM to interact with various LLM providers. Configure the following environment variables in your `.env` file:
+The agent uses LiteLLM to interact with various LLM providers. Configure theqfollowing environment variables in your `.env` file:

 - `LLM_MODEL`: The model to use (e.g., "openai/gpt-4o", "anthropic/claude-3-sonnet", "google/gemini-pro", etc.)
 - `LLM_KEY`: Your API key for the selected provider
@@ -167,7 +167,7 @@ npm install
 npx vite
 ```
 Access the UI at `http://localhost:5173`
-
+  

 ## Goal-Specific Tool Configuration
 Here is configuration guidance for specific goals. Travel and financial goals have configuration & setup as below.
@@ -175,7 +175,7 @@ Here is configuration guidance for specific goals. Travel and financial goals ha
 - `AGENT_GOAL=goal_event_flight_invoice` - Helps users find events, book flights, and arrange train travel with invoice generation
    - This is the scenario in the [original video](https://www.youtube.com/watch?v=GEXllEH2XiQ)

-#### Configuring Agent Goal: goal_event_flight_invoice
+#### Configuring Agent Goal: goal_event_flight_invoice 
 * The agent uses a mock function to search for events. This has zero configuration.
 * By default the agent uses a mock function to search for flights.
    * If you want to use the real flights API, go to `tools/search_flights.py` and replace the `search_flights` function with `search_flights_real_api` that exists in the same file.
@@ -190,9 +190,10 @@ Here is configuration guidance for specific goals. Travel and financial goals ha
 - `AGENT_GOAL=goal_match_train_invoice` - Focuses on Premier League match attendance with train booking and invoice generation
    - This goal was part of [Temporal's Replay 2025 conference keynote demo](https://www.youtube.com/watch?v=YDxAWrIBQNE)
    - Note, there is failure built in to this demo (the train booking step) to show how the agent can handle failures and retry. See Tool Configuration below for details.
-#### Configuring Agent Goal: goal_match_train_invoice
+#### Configuring Agent Goal: goal_match_train_invoice 
 NOTE: This goal was developed for an on-stage demo and has failure (and its resolution) built in to show how the agent can handle failures and retry.
-* Omit `FOOTBALL_DATA_API_KEY` from .env for the `SearchFixtures` tool to automatically return mock Premier League fixtures. Finding a real match requires a key from [Football Data](https://www.football-data.org). Sign up for a free account, then see the 'My Account' page to get your API token.
+* Finding a match requires a key from [Football Data](https://www.football-data.org). Sign up for a free account, then see the 'My Account' page to get your API token. Set `FOOTBALL_DATA_API_KEY` to this value.
+    * If you're lazy go to `tools/search_fixtures.py` and replace the `search_fixtures` function with the mock `search_fixtures_example` that exists in the same file.
 * We use a mock function to search for trains. Start the train API server to use the real API: `python thirdparty/train_api.py`
 * * The train activity is 'enterprise' so it's written in C# and requires a .NET runtime. See the [.NET backend](#net-(enterprise)-backend) section for details on running it.
 * Requires a Stripe key for the `create_invoice` tool. Set this in the `STRIPE_API_KEY` environment variable in .env
@@ -212,15 +213,15 @@ poetry run python thirdparty/train_api.py

 ##### Python Train Legacy Worker
 > Agent Goal: goal_match_train_invoice only
-
+ 
 These are Python activities that fail (raise NotImplemented) to show how Temporal handles a failure. You can run these activities with.
-
+ 
 ```bash
- poetry run python scripts/run_legacy_worker.py
+ poetry run python scripts/run_legacy_worker.py 
 ```
-
+ 
 The activity will fail and be retried infinitely. To rescue the activity (and its corresponding workflows), kill the worker and run the .NET one in the section below.
-
+ 
 ##### .NET (enterprise) Worker ;)
 We have activities written in C# to call the train APIs.
 ```bash
@@ -233,12 +234,12 @@ If you're running your train API above on a different host/port then change the
 #### Goals: FIN - Money Movement and Loan Application
 Make sure you have the mock users you want (such as yourself) in [the account mock data file](./tools/data/customer_account_data.json).

- `AGENT_GOAL=goal_fin_move_money` - This scenario _can_ initiate a secondary workflow to move money. Check out [this repo](https://github.com/temporal-sa/temporal-money-transfer-java) - you'll need to get the worker running and connected to the same account as the agentic worker.
+- `AGENT_GOAL=goal_fin_move_money` - This scenario _can_ initiate a secondary workflow to move money. Check out [this repo](https://github.com/temporal-sa/temporal-money-transfer-java) - you'll need to get the worker running and connected to the same account as the agentic worker. 
 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
 ```
- `AGENT_GOAL=goal_fin_loan_application` - This scenario _can_ initiate a secondary workflow to apply for a loan. Check out [this repo](https://github.com/temporal-sa/temporal-latency-optimization-scenarios) - you'll need to get the worker running and connected to the same account as the agentic worker.
+- `AGENT_GOAL=goal_fin_loan_application` - This scenario _can_ initiate a secondary workflow to apply for a loan. Check out [this repo](https://github.com/temporal-sa/temporal-latency-optimization-scenarios) - you'll need to get the worker running and connected to the same account as the agentic worker. 
 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
--- a/shared/config.py
+++ b/shared/config.py
@@ -1,5 +1,4 @@
 import os
-
 from dotenv import load_dotenv
 from temporalio.client import Client
 from temporalio.service import TLSConfig
@@ -10,16 +9,13 @@ load_dotenv(override=True)
 TEMPORAL_ADDRESS = os.getenv("TEMPORAL_ADDRESS", "localhost:7233")
 TEMPORAL_NAMESPACE = os.getenv("TEMPORAL_NAMESPACE", "default")
 TEMPORAL_TASK_QUEUE = os.getenv("TEMPORAL_TASK_QUEUE", "agent-task-queue")
-TEMPORAL_LEGACY_TASK_QUEUE = os.getenv(
-    "TEMPORAL_LEGACY_TASK_QUEUE", "agent-task-queue-legacy"
-)
+TEMPORAL_LEGACY_TASK_QUEUE = os.getenv("TEMPORAL_LEGACY_TASK_QUEUE", "agent-task-queue-legacy")

 # Authentication settings
 TEMPORAL_TLS_CERT = os.getenv("TEMPORAL_TLS_CERT", "")
 TEMPORAL_TLS_KEY = os.getenv("TEMPORAL_TLS_KEY", "")
 TEMPORAL_API_KEY = os.getenv("TEMPORAL_API_KEY", "")

-
 async def get_temporal_client() -> Client:
    """
    Creates a Temporal client based on environment configuration.
--- a/tests/README.md
+++ b/tests/README.md
@@ -1,350 +0,0 @@
-# Temporal AI Agent - Testing Guide
-
-This directory contains comprehensive tests for the Temporal AI Agent project. The tests cover workflows, activities, and integration scenarios using Temporal's testing framework.
-
-## Test Structure
-
-```
-tests/
-├── README.md                      # This file - testing documentation
-├── conftest.py                    # Test configuration and fixtures
-├── test_agent_goal_workflow.py    # Workflow tests
-├── test_tool_activities.py        # Activity tests
-└── workflowtests/                 # Legacy workflow tests
-    └── agent_goal_workflow_test.py
-```
-
-## Test Types
-
-### 1. Workflow Tests (`test_agent_goal_workflow.py`)
-
-Tests the main `AgentGoalWorkflow` class covering:
-
- **Workflow Initialization**: Basic workflow startup and state management
- **Signal Handling**: Testing user_prompt, confirm, end_chat signals
- **Query Methods**: Testing all workflow query endpoints
- **State Management**: Conversation history, goal changes, tool data
- **Validation Flow**: Prompt validation and error handling
- **Tool Execution Flow**: Confirmation and tool execution cycles
-
-### 2. Activity Tests (`test_tool_activities.py`)
-
-Tests the `ToolActivities` class and `dynamic_tool_activity` function:
-
- **LLM Integration**: Testing agent_toolPlanner with mocked LLM responses
- **Validation Logic**: Testing agent_validatePrompt with various scenarios
- **Environment Configuration**: Testing get_wf_env_vars with different env setups
- **JSON Processing**: Testing response parsing and sanitization
- **Dynamic Tool Execution**: Testing the dynamic activity dispatcher
- **Integration**: End-to-end activity execution in Temporal workers
-
-### 3. Configuration Tests (`conftest.py`)
-
-Provides shared test fixtures and configuration:
-
- **Temporal Environment**: Local and time-skipping test environments
- **Sample Data**: Pre-configured agent goals, conversation history, inputs
- **Test Client**: Configured Temporal client for testing
-
-## Running Tests
-
-### Prerequisites
-
-Ensure you have the required dependencies installed:
-
-```bash
-poetry install --with dev
-```
-
-### Basic Test Execution
-
-Run all tests:
-```bash
-poetry run pytest
-```
-
-Run specific test files:
-```bash
-# Workflow tests only
-poetry run pytest tests/test_agent_goal_workflow.py
-
-# Activity tests only
-poetry run pytest tests/test_tool_activities.py
-
-# Legacy tests
-poetry run pytest tests/workflowtests/
-```
-
-Run with verbose output:
-```bash
-poetry run pytest -v
-```
-
-### Test Environment Options
-
-The tests support different Temporal environments via the `--workflow-environment` flag:
-
-#### Local Environment (Default)
-Uses a local Temporal test server:
-```bash
-poetry run pytest --workflow-environment=local
-```
-
-#### Time-Skipping Environment
-Uses Temporal's time-skipping test environment for faster execution:
-```bash
-poetry run pytest --workflow-environment=time-skipping
-```
-
-#### External Server
-Connect to an existing Temporal server:
-```bash
-poetry run pytest --workflow-environment=localhost:7233
-```
-
-#### Setup Script for AI Agent environments such as OpenAI Codex
-```bash
-export SHELL=/bin/bash
-curl -sSL https://install.python-poetry.org | python3 -
-export PATH="$HOME/.local/bin:$PATH"
-ls
-poetry install --with dev
-cd frontend
-npm install
-cd ..
-
-# Pre-download the temporal test server binary
-poetry run python3 -c "
-import asyncio
-import sys
-from temporalio.testing import WorkflowEnvironment
-
-async def predownload():
-    try:
-        print('Starting test server download...')
-        env = await WorkflowEnvironment.start_time_skipping()
-        print('Test server downloaded and started successfully')
-        await env.shutdown()
-        print('Test server shut down successfully')
-    except Exception as e:
-        print(f'Error during download: {e}')
-        sys.exit(1)
-
-asyncio.run(predownload())
-"
-```
-
-### Filtering Tests
-
-Run tests by pattern:
-```bash
-# Run only validation tests
-poetry run pytest -k "validation"
-
-# Run only workflow tests
-poetry run pytest -k "workflow"
-
-# Run only activity tests
-poetry run pytest -k "activity"
-```
-
-Run tests by marker (if you add custom markers):
-```bash
-# Run only integration tests
-poetry run pytest -m integration
-
-# Skip slow tests
-poetry run pytest -m "not slow"
-```
-
-## Test Configuration
-
-### Test Discovery
-
-The `vibe/` directory is excluded from test collection to avoid conflicts with sample tests. This is configured in `pyproject.toml`:
-
-```toml
-[tool.pytest.ini_options]
-norecursedirs = ["vibe"]
-```
-
-### Environment Variables
-
-Tests respect the following environment variables:
-
- `LLM_MODEL`: Model to use for LLM testing (defaults to "openai/gpt-4")
- `LLM_KEY`: API key for LLM service
- `LLM_BASE_URL`: Custom base URL for LLM service
- `SHOW_CONFIRM`: Whether to show confirmation dialogs
- `AGENT_GOAL`: Default agent goal setting
-
-### Mocking Strategy
-
-The tests use extensive mocking to avoid external dependencies:
-
- **LLM Calls**: Mocked using `unittest.mock` to avoid actual API calls
- **Tool Handlers**: Mocked to test workflow logic without tool execution
- **Environment Variables**: Patched for consistent test environments
-
-## Writing New Tests
-
-### Test Naming Convention
-
- Test files: `test_<module_name>.py`
- Test classes: `Test<ClassName>`
- Test methods: `test_<functionality>_<scenario>`
-
-Example:
-```python
-class TestAgentGoalWorkflow:
-    async def test_user_prompt_signal_valid_input(self, client, sample_combined_input):
-        # Test implementation
-        pass
-```
-
-### Using Fixtures
-
-Leverage the provided fixtures for consistent test data:
-
-```python
-async def test_my_workflow(self, client, sample_agent_goal, sample_conversation_history):
-    # client: Temporal test client
-    # sample_agent_goal: Pre-configured AgentGoal
-    # sample_conversation_history: Sample conversation data
-    pass
-```
-
-### Mocking External Dependencies
-
-Always mock external services:
-
-```python
-@patch('activities.tool_activities.completion')
-async def test_llm_integration(self, mock_completion):
-    mock_completion.return_value.choices[0].message.content = '{"test": "response"}'
-    # Test implementation
-```
-
-### Testing Workflow Signals and Queries
-
-```python
-async def test_workflow_signal(self, client, sample_combined_input):
-    # Start workflow
-    handle = await client.start_workflow(
-        AgentGoalWorkflow.run,
-        sample_combined_input,
-        id=str(uuid.uuid4()),
-        task_queue=task_queue_name,
-    )
-    
-    # Send signal
-    await handle.signal(AgentGoalWorkflow.user_prompt, "test message")
-    
-    # Query state
-    conversation = await handle.query(AgentGoalWorkflow.get_conversation_history)
-    
-    # End workflow
-    await handle.signal(AgentGoalWorkflow.end_chat)
-    result = await handle.result()
-```
-
-## Test Data and Fixtures
-
-### Sample Agent Goal
-
-The `sample_agent_goal` fixture provides a basic agent goal with:
- Goal ID: "test_goal"
- One test tool with a required string argument
- Suitable for most workflow testing scenarios
-
-### Sample Conversation History
-
-The `sample_conversation_history` fixture provides:
- Basic user and agent message exchange
- Proper message format for testing
-
-### Sample Combined Input
-
-The `sample_combined_input` fixture provides:
- Complete workflow input with agent goal and tool params
- Conversation summary and prompt queue
- Ready for workflow execution
-
-## Debugging Tests
-
-### Verbose Logging
-
-Enable detailed logging:
-```bash
-poetry run pytest --log-cli-level=DEBUG -s
-```
-
-### Temporal Web UI
-
-When using local environment, access Temporal Web UI at http://localhost:8233 to inspect workflow executions during tests.
-
-### Test Isolation
-
-Each test uses unique task queue names to prevent interference:
-```python
-task_queue_name = str(uuid.uuid4())
-```
-
-## Continuous Integration
-
-### GitHub Actions Example
-
-```yaml
-name: Test
-on: [push, pull_request]
-jobs:
-  test:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v3
-      - uses: actions/setup-python@v4
-        with:
-          python-version: '3.10'
-      - run: pip install poetry
-      - run: poetry install --with dev
-      - run: poetry run pytest --workflow-environment=time-skipping
-```
-
-### Test Coverage
-
-Generate coverage reports:
-```bash
-poetry add --group dev pytest-cov
-poetry run pytest --cov=workflows --cov=activities --cov-report=html
-```
-
-## Best Practices
-
-1. **Mock External Dependencies**: Always mock LLM calls, file I/O, and network requests
-2. **Use Time-Skipping**: For CI/CD, prefer time-skipping environment for speed
-3. **Unique Identifiers**: Use UUIDs for workflow IDs and task queues
-4. **Clean Shutdown**: Always end workflows properly in tests
-5. **Descriptive Names**: Use clear, descriptive test names
-6. **Test Edge Cases**: Include error scenarios and validation failures
-7. **Keep Tests Fast**: Use mocks to avoid slow external calls
-8. **Isolate Tests**: Ensure tests don't depend on each other
-
-## Troubleshooting
-
-### Common Issues
-
-1. **Workflow Timeout**: Increase timeouts or use time-skipping environment
-2. **Mock Not Working**: Check patch decorators and import paths
-3. **Test Hanging**: Ensure workflows are properly ended with signals
-4. **Environment Issues**: Check environment variable settings
-
-### Getting Help
-
- Check Temporal Python SDK documentation
- Review existing test patterns in the codebase
- Use `poetry run pytest --collect-only` to verify test discovery
- Run with `-v` flag for detailed output
-
-## Legacy Tests
-
-The `workflowtests/` directory contains legacy tests. New tests should be added to the main `tests/` directory following the patterns established in this guide.
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -41,12 +41,7 @@ def event_loop():
 async def env(request) -> AsyncGenerator[WorkflowEnvironment, None]:
    env_type = request.config.getoption("--workflow-environment")
    if env_type == "local":
-        env = await WorkflowEnvironment.start_local(
-            dev_server_extra_args=[
-                "--dynamic-config-value",
-                "frontend.enableExecuteMultiOperation=true",
-            ]
-        )
+        env = await WorkflowEnvironment.start_local()
    elif env_type == "time-skipping":
        env = await WorkflowEnvironment.start_time_skipping()
    else:
@@ -58,54 +53,3 @@ async def env(request) -> AsyncGenerator[WorkflowEnvironment, None]:
@pytest_asyncio.fixture
 async def client(env: WorkflowEnvironment) -> Client:
    return env.client
-
-
-@pytest.fixture
-def sample_agent_goal():
-    """Sample agent goal for testing."""
-    from models.tool_definitions import AgentGoal, ToolArgument, ToolDefinition
-
-    return AgentGoal(
-        id="test_goal",
-        category_tag="test",
-        agent_name="TestAgent",
-        agent_friendly_description="A test agent for testing purposes",
-        description="Test goal for agent testing",
-        tools=[
-            ToolDefinition(
-                name="TestTool",
-                description="A test tool for testing purposes",
-                arguments=[
-                    ToolArgument(
-                        name="test_arg", type="string", description="A test argument"
-                    )
-                ],
-            )
-        ],
-    )
-
-
-@pytest.fixture
-def sample_conversation_history():
-    """Sample conversation history for testing."""
-    return {
-        "messages": [
-            {"actor": "user", "response": "Hello, I need help with testing"},
-            {"actor": "agent", "response": "I can help you with that"},
-        ]
-    }
-
-
-@pytest.fixture
-def sample_combined_input(sample_agent_goal):
-    """Sample combined input for workflow testing."""
-    from collections import deque
-
-    from models.data_types import AgentGoalWorkflowParams, CombinedInput
-
-    tool_params = AgentGoalWorkflowParams(
-        conversation_summary="Test conversation summary",
-        prompt_queue=deque(),  # Start with empty queue for most tests
-    )
-
-    return CombinedInput(agent_goal=sample_agent_goal, tool_params=tool_params)
--- a/tests/test_agent_goal_workflow.py
+++ b/tests/test_agent_goal_workflow.py
@@ -1,543 +0,0 @@
-import uuid
-
-from temporalio import activity
-from temporalio.client import Client
-from temporalio.worker import Worker
-
-from models.data_types import (
-    AgentGoalWorkflowParams,
-    CombinedInput,
-    EnvLookupInput,
-    EnvLookupOutput,
-    ToolPromptInput,
-    ValidationInput,
-    ValidationResult,
-)
-from workflows.agent_goal_workflow import AgentGoalWorkflow
-
-
-class TestAgentGoalWorkflow:
-    """Test cases for AgentGoalWorkflow."""
-
-    async def test_workflow_initialization(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test workflow can be initialized and started."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars],
-        ):
-            # Start workflow but don't wait for completion since it runs indefinitely
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Verify workflow is running
-            assert handle is not None
-
-            # Query the workflow to check initial state
-            conversation_history = await handle.query(
-                AgentGoalWorkflow.get_conversation_history
-            )
-            assert isinstance(conversation_history, dict)
-            assert "messages" in conversation_history
-
-            # Test goal query
-            agent_goal = await handle.query(AgentGoalWorkflow.get_agent_goal)
-            assert agent_goal == sample_combined_input.agent_goal
-
-            # End the workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            result = await handle.result()
-            assert isinstance(result, str)
-
-    async def test_user_prompt_signal(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test user_prompt signal handling."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        @activity.defn(name="agent_validatePrompt")
-        async def mock_agent_validatePrompt(
-            validation_input: ValidationInput,
-        ) -> ValidationResult:
-            return ValidationResult(validationResult=True, validationFailedReason={})
-
-        @activity.defn(name="agent_toolPlanner")
-        async def mock_agent_toolPlanner(input: ToolPromptInput) -> dict:
-            return {"next": "done", "response": "Test response from LLM"}
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[
-                mock_get_wf_env_vars,
-                mock_agent_validatePrompt,
-                mock_agent_toolPlanner,
-            ],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Send user prompt
-            await handle.signal(
-                AgentGoalWorkflow.user_prompt, "Hello, this is a test message"
-            )
-
-            # Wait for workflow to complete (it should end due to "done" next step)
-            result = await handle.result()
-            assert isinstance(result, str)
-
-            # Verify the conversation includes our message
-            import json
-
-            try:
-                conversation_history = json.loads(result.replace("'", '"'))
-            except Exception:
-                # Fallback to eval if json fails
-                conversation_history = eval(result)
-            messages = conversation_history["messages"]
-
-            # Should have our user message and agent response
-            user_messages = [msg for msg in messages if msg["actor"] == "user"]
-            assert len(user_messages) > 0
-            assert any(
-                "Hello, this is a test message" in str(msg["response"])
-                for msg in user_messages
-            )
-
-    async def test_confirm_signal(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test confirm signal handling for tool execution."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        @activity.defn(name="agent_validatePrompt")
-        async def mock_agent_validatePrompt(
-            validation_input: ValidationInput,
-        ) -> ValidationResult:
-            return ValidationResult(validationResult=True, validationFailedReason={})
-
-        @activity.defn(name="agent_toolPlanner")
-        async def mock_agent_toolPlanner(input: ToolPromptInput) -> dict:
-            return {
-                "next": "confirm",
-                "tool": "TestTool",
-                "args": {"test_arg": "test_value"},
-                "response": "Ready to execute tool",
-            }
-
-        @activity.defn(name="TestTool")
-        async def mock_test_tool(args: dict) -> dict:
-            return {"result": "Test tool executed successfully"}
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[
-                mock_get_wf_env_vars,
-                mock_agent_validatePrompt,
-                mock_agent_toolPlanner,
-                mock_test_tool,
-            ],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Send user prompt that will require confirmation
-            await handle.signal(AgentGoalWorkflow.user_prompt, "Execute the test tool")
-
-            # Query to check tool data is set
-            import asyncio
-
-            await asyncio.sleep(0.1)  # Give workflow time to process
-
-            tool_data = await handle.query(AgentGoalWorkflow.get_latest_tool_data)
-            if tool_data:
-                assert tool_data.get("tool") == "TestTool"
-                assert tool_data.get("next") == "confirm"
-
-            # Send confirmation and end chat
-            await handle.signal(AgentGoalWorkflow.confirm)
-            await handle.signal(AgentGoalWorkflow.end_chat)
-
-            result = await handle.result()
-            assert isinstance(result, str)
-
-    async def test_validation_failure(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test workflow handles validation failures correctly."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        @activity.defn(name="agent_validatePrompt")
-        async def mock_agent_validatePrompt(
-            validation_input: ValidationInput,
-        ) -> ValidationResult:
-            return ValidationResult(
-                validationResult=False,
-                validationFailedReason={
-                    "next": "question",
-                    "response": "Your request doesn't make sense in this context",
-                },
-            )
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars, mock_agent_validatePrompt],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Send invalid prompt
-            await handle.signal(
-                AgentGoalWorkflow.user_prompt, "Invalid nonsensical prompt"
-            )
-
-            # Give workflow time to process the prompt
-            import asyncio
-
-            await asyncio.sleep(0.2)
-
-            # End workflow to check conversation
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            result = await handle.result()
-
-            # Verify validation failure message was added
-            import json
-
-            try:
-                conversation_history = json.loads(result.replace("'", '"'))
-            except Exception:
-                # Fallback to eval if json fails
-                conversation_history = eval(result)
-            messages = conversation_history["messages"]
-
-            # Should have validation failure response
-            agent_messages = [msg for msg in messages if msg["actor"] == "agent"]
-            assert len(agent_messages) > 0
-            assert any(
-                "doesn't make sense" in str(msg["response"]) for msg in agent_messages
-            )
-
-    async def test_conversation_summary_initialization(
-        self, client: Client, sample_agent_goal
-    ):
-        """Test workflow initializes with conversation summary."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create input with conversation summary
-        from collections import deque
-
-        tool_params = AgentGoalWorkflowParams(
-            conversation_summary="Previous conversation summary", prompt_queue=deque()
-        )
-        combined_input = CombinedInput(
-            agent_goal=sample_agent_goal, tool_params=tool_params
-        )
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Give workflow time to initialize
-            import asyncio
-
-            await asyncio.sleep(0.1)
-
-            # Query conversation summary
-            summary = await handle.query(AgentGoalWorkflow.get_summary_from_history)
-            assert summary == "Previous conversation summary"
-
-            # Query conversation history - should include summary message
-            conversation_history = await handle.query(
-                AgentGoalWorkflow.get_conversation_history
-            )
-            messages = conversation_history["messages"]
-
-            # Should have conversation_summary message
-            summary_messages = [
-                msg for msg in messages if msg["actor"] == "conversation_summary"
-            ]
-            assert len(summary_messages) == 1
-            assert summary_messages[0]["response"] == "Previous conversation summary"
-
-            # End workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            await handle.result()
-
-    async def test_workflow_queries(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test all workflow query methods."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Give workflow time to initialize
-            import asyncio
-
-            await asyncio.sleep(0.1)
-
-            # Test get_conversation_history query
-            conversation_history = await handle.query(
-                AgentGoalWorkflow.get_conversation_history
-            )
-            assert isinstance(conversation_history, dict)
-            assert "messages" in conversation_history
-
-            # Test get_agent_goal query
-            agent_goal = await handle.query(AgentGoalWorkflow.get_agent_goal)
-            assert agent_goal.id == sample_combined_input.agent_goal.id
-
-            # Test get_summary_from_history query
-            summary = await handle.query(AgentGoalWorkflow.get_summary_from_history)
-            # Summary might be None if not set, so check for that
-            if sample_combined_input.tool_params.conversation_summary:
-                assert summary == sample_combined_input.tool_params.conversation_summary
-            else:
-                assert summary is None
-
-            # Test get_latest_tool_data query (should be None initially)
-            tool_data = await handle.query(AgentGoalWorkflow.get_latest_tool_data)
-            assert tool_data is None
-
-            # End workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            await handle.result()
-
-    async def test_enable_disable_debugging_confirm_signals(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test debugging confirm enable/disable signals."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Test enable debugging confirm signal
-            await handle.signal(AgentGoalWorkflow.enable_debugging_confirm)
-
-            # Test disable debugging confirm signal
-            await handle.signal(AgentGoalWorkflow.disable_debugging_confirm)
-
-            # End workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            result = await handle.result()
-            assert isinstance(result, str)
-
-    async def test_workflow_with_empty_prompt_queue(
-        self, client: Client, sample_agent_goal
-    ):
-        """Test workflow behavior with empty prompt queue."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create input with empty prompt queue
-        from collections import deque
-
-        tool_params = AgentGoalWorkflowParams(
-            conversation_summary=None, prompt_queue=deque()
-        )
-        combined_input = CombinedInput(
-            agent_goal=sample_agent_goal, tool_params=tool_params
-        )
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[mock_get_wf_env_vars],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Give workflow time to initialize
-            import asyncio
-
-            await asyncio.sleep(0.1)
-
-            # Query initial state
-            conversation_history = await handle.query(
-                AgentGoalWorkflow.get_conversation_history
-            )
-            assert isinstance(conversation_history, dict)
-            assert "messages" in conversation_history
-
-            # Should have no messages initially (empty prompt queue, no summary)
-            messages = conversation_history["messages"]
-            assert len(messages) == 0
-
-            # End workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            result = await handle.result()
-            assert isinstance(result, str)
-
-    async def test_multiple_user_prompts(
-        self, client: Client, sample_combined_input: CombinedInput
-    ):
-        """Test workflow handling multiple user prompts in sequence."""
-        task_queue_name = str(uuid.uuid4())
-
-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        @activity.defn(name="agent_validatePrompt")
-        async def mock_agent_validatePrompt(
-            validation_input: ValidationInput,
-        ) -> ValidationResult:
-            return ValidationResult(validationResult=True, validationFailedReason={})
-
-        @activity.defn(name="agent_toolPlanner")
-        async def mock_agent_toolPlanner(input: ToolPromptInput) -> dict:
-            # Keep workflow running for multiple prompts
-            return {"next": "question", "response": f"Processed: {input.prompt}"}
-
-        async with Worker(
-            client,
-            task_queue=task_queue_name,
-            workflows=[AgentGoalWorkflow],
-            activities=[
-                mock_get_wf_env_vars,
-                mock_agent_validatePrompt,
-                mock_agent_toolPlanner,
-            ],
-        ):
-            handle = await client.start_workflow(
-                AgentGoalWorkflow.run,
-                sample_combined_input,
-                id=str(uuid.uuid4()),
-                task_queue=task_queue_name,
-            )
-
-            # Send multiple prompts
-            await handle.signal(AgentGoalWorkflow.user_prompt, "First message")
-            import asyncio
-
-            await asyncio.sleep(0.1)
-
-            await handle.signal(AgentGoalWorkflow.user_prompt, "Second message")
-            await asyncio.sleep(0.1)
-
-            await handle.signal(AgentGoalWorkflow.user_prompt, "Third message")
-            await asyncio.sleep(0.1)
-
-            # End workflow
-            await handle.signal(AgentGoalWorkflow.end_chat)
-            result = await handle.result()
-            assert isinstance(result, str)
-
-            # Parse result and verify multiple messages
-            import json
-
-            try:
-                conversation_history = json.loads(result.replace("'", '"'))
-            except Exception:
-                conversation_history = eval(result)
-            messages = conversation_history["messages"]
-
-            # Should have at least one user message (timing dependent)
-            user_messages = [msg for msg in messages if msg["actor"] == "user"]
-            assert len(user_messages) >= 1
-
-            # Verify at least the first message was processed
-            message_texts = [str(msg["response"]) for msg in user_messages]
-            assert any("First message" in text for text in message_texts)
--- a/tests/test_tool_activities.py
+++ b/tests/test_tool_activities.py
@@ -1,445 +0,0 @@
-import json
-import os
-from unittest.mock import AsyncMock, MagicMock, patch
-
-import pytest
-from temporalio.client import Client
-from temporalio.testing import ActivityEnvironment
-
-from activities.tool_activities import ToolActivities, dynamic_tool_activity
-from models.data_types import (
-    EnvLookupInput,
-    EnvLookupOutput,
-    ToolPromptInput,
-    ValidationInput,
-    ValidationResult,
-)
-
-
-class TestToolActivities:
-    """Test cases for ToolActivities."""
-
-    def setup_method(self):
-        """Set up test environment for each test."""
-        self.tool_activities = ToolActivities()
-
-    @pytest.mark.asyncio
-    async def test_agent_validatePrompt_valid_prompt(
-        self, sample_agent_goal, sample_conversation_history
-    ):
-        """Test agent_validatePrompt with a valid prompt."""
-        validation_input = ValidationInput(
-            prompt="I need help with the test tool",
-            conversation_history=sample_conversation_history,
-            agent_goal=sample_agent_goal,
-        )
-
-        # Mock the agent_toolPlanner to return a valid response
-        mock_response = {"validationResult": True, "validationFailedReason": {}}
-
-        with patch.object(
-            self.tool_activities, "agent_toolPlanner", new_callable=AsyncMock
-        ) as mock_planner:
-            mock_planner.return_value = mock_response
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.agent_validatePrompt, validation_input
-            )
-
-            assert isinstance(result, ValidationResult)
-            assert result.validationResult is True
-            assert result.validationFailedReason == {}
-
-            # Verify the mock was called with correct parameters
-            mock_planner.assert_called_once()
-
-    @pytest.mark.asyncio
-    async def test_agent_validatePrompt_invalid_prompt(
-        self, sample_agent_goal, sample_conversation_history
-    ):
-        """Test agent_validatePrompt with an invalid prompt."""
-        validation_input = ValidationInput(
-            prompt="asdfghjkl nonsense",
-            conversation_history=sample_conversation_history,
-            agent_goal=sample_agent_goal,
-        )
-
-        # Mock the agent_toolPlanner to return an invalid response
-        mock_response = {
-            "validationResult": False,
-            "validationFailedReason": {
-                "next": "question",
-                "response": "Your request doesn't make sense in this context",
-            },
-        }
-
-        with patch.object(
-            self.tool_activities, "agent_toolPlanner", new_callable=AsyncMock
-        ) as mock_planner:
-            mock_planner.return_value = mock_response
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.agent_validatePrompt, validation_input
-            )
-
-            assert isinstance(result, ValidationResult)
-            assert result.validationResult is False
-            assert "doesn't make sense" in str(result.validationFailedReason)
-
-    @pytest.mark.asyncio
-    async def test_agent_toolPlanner_success(self):
-        """Test agent_toolPlanner with successful LLM response."""
-        prompt_input = ToolPromptInput(
-            prompt="Test prompt", context_instructions="Test context instructions"
-        )
-
-        # Mock the completion function
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[
-            0
-        ].message.content = (
-            '{"next": "confirm", "tool": "TestTool", "response": "Test response"}'
-        )
-
-        with patch("activities.tool_activities.completion") as mock_completion:
-            mock_completion.return_value = mock_response
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.agent_toolPlanner, prompt_input
-            )
-
-            assert isinstance(result, dict)
-            assert result["next"] == "confirm"
-            assert result["tool"] == "TestTool"
-            assert result["response"] == "Test response"
-
-            # Verify completion was called with correct parameters
-            mock_completion.assert_called_once()
-            call_args = mock_completion.call_args[1]
-            assert call_args["model"] == self.tool_activities.llm_model
-            assert len(call_args["messages"]) == 2
-            assert call_args["messages"][0]["role"] == "system"
-            assert call_args["messages"][1]["role"] == "user"
-
-    @pytest.mark.asyncio
-    async def test_agent_toolPlanner_with_custom_base_url(self):
-        """Test agent_toolPlanner with custom base URL configuration."""
-        # Set up tool activities with custom base URL
-        with patch.dict(os.environ, {"LLM_BASE_URL": "https://custom.endpoint.com"}):
-            tool_activities = ToolActivities()
-
-            prompt_input = ToolPromptInput(
-                prompt="Test prompt", context_instructions="Test context instructions"
-            )
-
-            mock_response = MagicMock()
-            mock_response.choices = [MagicMock()]
-            mock_response.choices[
-                0
-            ].message.content = '{"next": "done", "response": "Test"}'
-
-            with patch("activities.tool_activities.completion") as mock_completion:
-                mock_completion.return_value = mock_response
-
-                activity_env = ActivityEnvironment()
-                await activity_env.run(tool_activities.agent_toolPlanner, prompt_input)
-
-                # Verify base_url was included in the call
-                call_args = mock_completion.call_args[1]
-                assert "base_url" in call_args
-                assert call_args["base_url"] == "https://custom.endpoint.com"
-
-    @pytest.mark.asyncio
-    async def test_agent_toolPlanner_json_parsing_error(self):
-        """Test agent_toolPlanner handles JSON parsing errors."""
-        prompt_input = ToolPromptInput(
-            prompt="Test prompt", context_instructions="Test context instructions"
-        )
-
-        # Mock the completion function to return invalid JSON
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[0].message.content = "Invalid JSON response"
-
-        with patch("activities.tool_activities.completion") as mock_completion:
-            mock_completion.return_value = mock_response
-
-            activity_env = ActivityEnvironment()
-            with pytest.raises(Exception):  # Should raise JSON parsing error
-                await activity_env.run(
-                    self.tool_activities.agent_toolPlanner, prompt_input
-                )
-
-    @pytest.mark.asyncio
-    async def test_get_wf_env_vars_default_values(self):
-        """Test get_wf_env_vars with default values."""
-        env_input = EnvLookupInput(
-            show_confirm_env_var_name="SHOW_CONFIRM", show_confirm_default=True
-        )
-
-        # Clear environment variables
-        with patch.dict(os.environ, {}, clear=True):
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.get_wf_env_vars, env_input
-            )
-
-            assert isinstance(result, EnvLookupOutput)
-            assert result.show_confirm is True  # default value
-            assert result.multi_goal_mode is True  # default value
-
-    @pytest.mark.asyncio
-    async def test_get_wf_env_vars_custom_values(self):
-        """Test get_wf_env_vars with custom environment values."""
-        env_input = EnvLookupInput(
-            show_confirm_env_var_name="SHOW_CONFIRM", show_confirm_default=True
-        )
-
-        # Set environment variables
-        with patch.dict(
-            os.environ, {"SHOW_CONFIRM": "false", "AGENT_GOAL": "specific_goal"}
-        ):
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.get_wf_env_vars, env_input
-            )
-
-            assert isinstance(result, EnvLookupOutput)
-            assert result.show_confirm is False  # from env var
-            assert result.multi_goal_mode is False  # from env var
-
-    def test_sanitize_json_response(self):
-        """Test JSON response sanitization."""
-        # Test with markdown code blocks
-        response_with_markdown = '```json\n{"test": "value"}\n```'
-        sanitized = self.tool_activities.sanitize_json_response(response_with_markdown)
-        assert sanitized == '{"test": "value"}'
-
-        # Test with extra whitespace
-        response_with_whitespace = '  \n{"test": "value"}  \n'
-        sanitized = self.tool_activities.sanitize_json_response(
-            response_with_whitespace
-        )
-        assert sanitized == '{"test": "value"}'
-
-    def test_parse_json_response_success(self):
-        """Test successful JSON parsing."""
-        json_string = '{"next": "confirm", "tool": "TestTool"}'
-        result = self.tool_activities.parse_json_response(json_string)
-
-        assert isinstance(result, dict)
-        assert result["next"] == "confirm"
-        assert result["tool"] == "TestTool"
-
-    def test_parse_json_response_failure(self):
-        """Test JSON parsing with invalid JSON."""
-        invalid_json = "Not valid JSON"
-
-        with pytest.raises(Exception):  # Should raise JSON parsing error
-            self.tool_activities.parse_json_response(invalid_json)
-
-
-class TestDynamicToolActivity:
-    """Test cases for dynamic_tool_activity."""
-
-    @pytest.mark.asyncio
-    async def test_dynamic_tool_activity_sync_handler(self):
-        """Test dynamic tool activity with synchronous handler."""
-        # Mock the activity info and payload converter
-        mock_info = MagicMock()
-        mock_info.activity_type = "TestTool"
-
-        mock_payload_converter = MagicMock()
-        mock_payload = MagicMock()
-        mock_payload.payload = b'{"test_arg": "test_value"}'
-        mock_payload_converter.from_payload.return_value = {"test_arg": "test_value"}
-
-        # Mock the handler function
-        def mock_handler(args):
-            return {"result": f"Handled {args['test_arg']}"}
-
-        with patch("temporalio.activity.info", return_value=mock_info), patch(
-            "temporalio.activity.payload_converter", return_value=mock_payload_converter
-        ), patch("tools.get_handler", return_value=mock_handler):
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(dynamic_tool_activity, [mock_payload])
-
-            assert isinstance(result, dict)
-            assert result["result"] == "Handled test_value"
-
-    @pytest.mark.asyncio
-    async def test_dynamic_tool_activity_async_handler(self):
-        """Test dynamic tool activity with asynchronous handler."""
-        # Mock the activity info and payload converter
-        mock_info = MagicMock()
-        mock_info.activity_type = "AsyncTestTool"
-
-        mock_payload_converter = MagicMock()
-        mock_payload = MagicMock()
-        mock_payload.payload = b'{"test_arg": "async_test"}'
-        mock_payload_converter.from_payload.return_value = {"test_arg": "async_test"}
-
-        # Mock the async handler function
-        async def mock_async_handler(args):
-            return {"async_result": f"Async handled {args['test_arg']}"}
-
-        with patch("temporalio.activity.info", return_value=mock_info), patch(
-            "temporalio.activity.payload_converter", return_value=mock_payload_converter
-        ), patch("tools.get_handler", return_value=mock_async_handler):
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(dynamic_tool_activity, [mock_payload])
-
-            assert isinstance(result, dict)
-            assert result["async_result"] == "Async handled async_test"
-
-
-class TestToolActivitiesIntegration:
-    """Integration tests for ToolActivities in a real Temporal environment."""
-
-    @pytest.mark.asyncio
-    async def test_activities_in_worker(self, client: Client):
-        """Test activities can be registered and executed in a worker."""
-        # task_queue_name = str(uuid.uuid4())
-        tool_activities = ToolActivities()
-
-        # Test get_wf_env_vars activity using ActivityEnvironment
-        env_input = EnvLookupInput(
-            show_confirm_env_var_name="TEST_CONFIRM", show_confirm_default=False
-        )
-
-        activity_env = ActivityEnvironment()
-        result = await activity_env.run(tool_activities.get_wf_env_vars, env_input)
-
-        assert isinstance(result, EnvLookupOutput)
-        assert isinstance(result.show_confirm, bool)
-        assert isinstance(result.multi_goal_mode, bool)
-
-
-class TestEdgeCases:
-    """Test edge cases and error handling."""
-
-    def setup_method(self):
-        """Set up test environment for each test."""
-        self.tool_activities = ToolActivities()
-
-    @pytest.mark.asyncio
-    async def test_agent_validatePrompt_with_empty_conversation_history(
-        self, sample_agent_goal
-    ):
-        """Test validation with empty conversation history."""
-        validation_input = ValidationInput(
-            prompt="Test prompt",
-            conversation_history={"messages": []},
-            agent_goal=sample_agent_goal,
-        )
-
-        mock_response = {"validationResult": True, "validationFailedReason": {}}
-
-        with patch.object(
-            self.tool_activities, "agent_toolPlanner", new_callable=AsyncMock
-        ) as mock_planner:
-            mock_planner.return_value = mock_response
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.agent_validatePrompt, validation_input
-            )
-
-            assert isinstance(result, ValidationResult)
-            assert result.validationResult
-            assert result.validationFailedReason == {}
-
-    @pytest.mark.asyncio
-    async def test_agent_toolPlanner_with_long_prompt(self):
-        """Test toolPlanner with very long prompt."""
-        long_prompt = "This is a very long prompt " * 100
-        tool_prompt_input = ToolPromptInput(
-            prompt=long_prompt, context_instructions="Test context instructions"
-        )
-
-        # Mock the completion response
-        mock_response = MagicMock()
-        mock_response.choices = [MagicMock()]
-        mock_response.choices[
-            0
-        ].message.content = '{"next": "done", "response": "Processed long prompt"}'
-
-        with patch("activities.tool_activities.completion", return_value=mock_response):
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.agent_toolPlanner, tool_prompt_input
-            )
-
-            assert isinstance(result, dict)
-            assert result["next"] == "done"
-            assert "Processed long prompt" in result["response"]
-
-    @pytest.mark.asyncio
-    async def test_sanitize_json_with_various_formats(self):
-        """Test JSON sanitization with various input formats."""
-        # Test markdown code blocks
-        markdown_json = '```json\n{"test": "value"}\n```'
-        result = self.tool_activities.sanitize_json_response(markdown_json)
-        assert result == '{"test": "value"}'
-
-        # Test with extra whitespace
-        whitespace_json = '   \n  {"test": "value"}  \n  '
-        result = self.tool_activities.sanitize_json_response(whitespace_json)
-        assert result == '{"test": "value"}'
-
-        # Test already clean JSON
-        clean_json = '{"test": "value"}'
-        result = self.tool_activities.sanitize_json_response(clean_json)
-        assert result == '{"test": "value"}'
-
-    @pytest.mark.asyncio
-    async def test_parse_json_response_with_invalid_json(self):
-        """Test JSON parsing with invalid JSON."""
-        with pytest.raises(json.JSONDecodeError):
-            self.tool_activities.parse_json_response("Invalid JSON {test: value")
-
-    @pytest.mark.asyncio
-    async def test_get_wf_env_vars_with_various_env_values(self):
-        """Test environment variable parsing with different values."""
-        # Test with "true" string
-        with patch.dict(os.environ, {"TEST_CONFIRM": "true"}):
-            env_input = EnvLookupInput(
-                show_confirm_env_var_name="TEST_CONFIRM", show_confirm_default=False
-            )
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.get_wf_env_vars, env_input
-            )
-
-            assert result.show_confirm
-
-        # Test with "false" string
-        with patch.dict(os.environ, {"TEST_CONFIRM": "false"}):
-            env_input = EnvLookupInput(
-                show_confirm_env_var_name="TEST_CONFIRM", show_confirm_default=True
-            )
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.get_wf_env_vars, env_input
-            )
-
-            assert not result.show_confirm
-
-        # Test with missing env var (should use default)
-        with patch.dict(os.environ, {}, clear=True):
-            env_input = EnvLookupInput(
-                show_confirm_env_var_name="MISSING_VAR", show_confirm_default=True
-            )
-
-            activity_env = ActivityEnvironment()
-            result = await activity_env.run(
-                self.tool_activities.get_wf_env_vars, env_input
-            )
-
-            assert result.show_confirm
--- a/tests/workflowtests/agent_goal_workflow_test.py
+++ b/tests/workflowtests/agent_goal_workflow_test.py
@@ -1,22 +1,15 @@
-import concurrent.futures
-import uuid
-from contextlib import contextmanager
-
-from temporalio import activity
 from temporalio.client import Client, WorkflowExecutionStatus
 from temporalio.worker import Worker
-
+import concurrent.futures
+from temporalio.testing import WorkflowEnvironment
 from api.main import get_initial_agent_goal
-from models.data_types import (
-    AgentGoalWorkflowParams,
-    CombinedInput,
-    EnvLookupInput,
-    EnvLookupOutput,
-    ToolPromptInput,
-    ValidationInput,
-    ValidationResult,
-)
+from models.data_types import AgentGoalWorkflowParams, CombinedInput
 from workflows.agent_goal_workflow import AgentGoalWorkflow
+from activities.tool_activities import ToolActivities, dynamic_tool_activity
+from unittest.mock import patch
+from dotenv import load_dotenv
+import os
+from contextlib import contextmanager


@contextmanager
@@ -26,49 +19,31 @@ def my_context():
    print("Cleanup")


-async def test_flight_booking(client: Client):
-    # load_dotenv("test_flights_single.env")

+async def test_flight_booking(client: Client):
+
+    #load_dotenv("test_flights_single.env")
+    
    with my_context() as value:
        print(f"Working with {value}")
-
+    
+        
        # Create the test environment
-        # env = await WorkflowEnvironment.start_local()
-        # client = env.client
-        task_queue_name = str(uuid.uuid4())
-        workflow_id = str(uuid.uuid4())
+        #env = await WorkflowEnvironment.start_local()
+        #client = env.client
+        task_queue_name = "agent-ai-workflow"
+        workflow_id = "agent-workflow"

-        # Create mock activity functions with proper signatures
-        @activity.defn(name="get_wf_env_vars")
-        async def mock_get_wf_env_vars(input: EnvLookupInput) -> EnvLookupOutput:
-            return EnvLookupOutput(show_confirm=True, multi_goal_mode=True)
-
-        @activity.defn(name="agent_validatePrompt")
-        async def mock_agent_validatePrompt(
-            validation_input: ValidationInput,
-        ) -> ValidationResult:
-            return ValidationResult(validationResult=True, validationFailedReason={})
-
-        @activity.defn(name="agent_toolPlanner")
-        async def mock_agent_toolPlanner(input: ToolPromptInput) -> dict:
-            return {"next": "done", "response": "Test response from LLM"}
-
-        with concurrent.futures.ThreadPoolExecutor(
-            max_workers=100
-        ) as activity_executor:
+        with concurrent.futures.ThreadPoolExecutor(max_workers=100) as activity_executor:        
            worker = Worker(
-                client,
+                client, 
                task_queue=task_queue_name,
                workflows=[AgentGoalWorkflow],
-                activities=[
-                    mock_get_wf_env_vars,
-                    mock_agent_validatePrompt,
-                    mock_agent_toolPlanner,
-                ],
+                activities=[ToolActivities.agent_validatePrompt, ToolActivities.agent_toolPlanner, ToolActivities.get_wf_env_vars, dynamic_tool_activity],
                activity_executor=activity_executor,
            )

-            async with worker:
+            async with worker:                 
                initial_agent_goal = get_initial_agent_goal()
                # Create combined input
                combined_input = CombinedInput(
@@ -76,36 +51,30 @@ async def test_flight_booking(client: Client):
                    agent_goal=initial_agent_goal,
                )

-                prompt = "Hello!"
+                prompt="Hello!"

-                # async with Worker(client, task_queue=task_queue_name, workflows=[AgentGoalWorkflow], activities=[ToolActivities.agent_validatePrompt, ToolActivities.agent_toolPlanner, dynamic_tool_activity]):
+                #async with Worker(client, task_queue=task_queue_name, workflows=[AgentGoalWorkflow], activities=[ToolActivities.agent_validatePrompt, ToolActivities.agent_toolPlanner, dynamic_tool_activity]):

                # todo set goal categories for scenarios
                handle = await client.start_workflow(
                    AgentGoalWorkflow.run,
                    combined_input,
-                    id=workflow_id,
+                    id=workflow_id, 
                    task_queue=task_queue_name,
                    start_signal="user_prompt",
                    start_signal_args=[prompt],
                )
                # todo send signals to simulate user input
                # await handle.signal(AgentGoalWorkflow.user_prompt, "book flights") # for multi-goal
-                await handle.signal(
-                    AgentGoalWorkflow.user_prompt, "sydney in september"
-                )
-                assert (
-                    WorkflowExecutionStatus.RUNNING == (await handle.describe()).status
-                )
+                await handle.signal(AgentGoalWorkflow.user_prompt, "sydney in september")
+                assert WorkflowExecutionStatus.RUNNING == (await handle.describe()).status

-                # assert ["Hello, user1", "Hello, user2"] == await handle.result()
-                await handle.signal(
-                    AgentGoalWorkflow.user_prompt, "I'm all set, end conversation"
-                )
-
-                # assert WorkflowExecutionStatus.COMPLETED == (await handle.describe()).status
+                
+                #assert ["Hello, user1", "Hello, user2"] == await handle.result()
+                await handle.signal(AgentGoalWorkflow.user_prompt, "I'm all set, end conversation")
+                
+                #assert WorkflowExecutionStatus.COMPLETED == (await handle.describe()).status

                result = await handle.result()
-                print(f"Workflow result: {result}")
-                # todo dump workflow history for analysis optional
-                # todo assert result is good
+                #todo dump workflow history for analysis optional
+                #todo assert result is good
--- a/thirdparty/train_api.py
+++ b/thirdparty/train_api.py
@@ -1,9 +1,9 @@
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from urllib.parse import parse_qs, urlparse
 import json
+import time
 import random
 import string
-import time
-from http.server import BaseHTTPRequestHandler, HTTPServer
-from urllib.parse import parse_qs, urlparse


 def parse_datetime(datetime_str):
@@ -213,4 +213,4 @@ def run_server():


 if __name__ == "__main__":
-    run_server()
+    run_server()
--- a/todo.md
+++ b/todo.md
@@ -1,33 +1,8 @@
 # todo list
+[x] take steve's confirm box changes https://temporaltechnologies.slack.com/archives/D062SV8KEEM/p1745251279164319 <br />
+[ ] consider adding goal categories to goal picker

-## General Agent Enhancements
-
-[ ] MCP: There is a plan to add MCP (Model Context Protocol) to the agent. This really really really needs to be done and is scheduled to be done by @steveandroulakis some time in June 2025.
-
-[ ] Google's A2A is emerging as the standard way to hand off agents to other agents. We should examine implementing this soon.
-
-[ ] Custom metrics/tracing is important for AI specific aspects such as number of LLM calls, number of bad LLM responses that require retrying, number of bad chat outcomes. We should add this.
-
-[ ] Evals are very important in agents. We want to be able to 'judge' the agent's performance both in dev and production (AIOps). This will help us improve our agent's performance over time in a targeted fashion.
-
-[ ] Dynamically switch LLMs on persistent failures: <br />
-    - detect failure in the activity using failurecount <br />
-    - activity switches to secondary LLM defined in .env
-    - activity reports switch to workflow
-
-[ ] Collapse history/summarize chat after goal finished <br />
-
-[ ] Write tests<br />
-
-[ ] non-retry the api key error - "Invalid API Key provided: sk_test_**J..." and "AuthenticationError" <br />
-
-[ ] add visual feedback when workflow starting <br />
-
-[ ] enable user to list agents at any time - like end conversation - probably with a next step<br />
-
-## Ideas for more goals and tools
-
-[ ] Add fintech goals <br />
+[ ] adding fintech goals <br />
 - Fraud Detection and Prevention - The AI monitors transactions across accounts, flagging suspicious activities (e.g., unusual spending patterns or login attempts) and autonomously freezing accounts or notifying customers and compliance teams.<br />
 - Personalized Financial Advice - An AI agent analyzes a customer’s financial data (e.g., income, spending habits, savings, investments) and provides tailored advice, such as budgeting tips, investment options, or debt repayment strategies.<br />
 - Portfolio Management and Rebalancing - The AI monitors a customer’s investment portfolio, rebalancing it automatically based on market trends, risk tolerance, and financial goals (e.g., shifting assets between stocks, bonds, or crypto).<br />
@@ -37,3 +12,21 @@
    [ ] tool is maybe a new tool asking the LLM to advise

 [ ] for demo simulate failure  - add utilities/simulated failures from pipeline demo <br />
+
+[ ] LLM failure->autoswitch: <br />
+    - detect failure in the activity using failurecount <br />
+    - activity switches to secondary LLM defined in .env
+    - activity reports switch to workflow
+
+[ ] for demo simulate failure  - add utilities/simulated failures from pipeline demo <br />
+
+[ ] expand [tests](./tests/agent_goal_workflow_test.py)<br />
+[ ] collapse history/summarize after goal finished <br />
+[ ] add aws bedrock <br />
+
+[ ] 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 <br />
+- Insight into the agent’s performance <br />
+[ ] non-retry the api key error - "Invalid API Key provided: sk_test_**J..." and "AuthenticationError" <br />
+[ ] add visual feedback when workflow starting <br />
+[ ] enable user to list agents at any time - like end conversation - probably with a next step<br />
+ - with changing "'Next should only be "pick-new-goal" if all tools have been run (use the system prompt to figure that out).'" in [prompt_generators](./prompts/agent_prompt_generators.py).
--- a/tools/init.py
+++ b/tools/init.py
@@ -1,24 +1,29 @@
-from .change_goal import change_goal
+from .search_fixtures import search_fixtures
+from .search_flights import search_flights
+from .search_trains import search_trains
+from .search_trains import book_trains
 from .create_invoice import create_invoice
-from .ecommerce.get_order import get_order
-from .ecommerce.list_orders import list_orders
-from .ecommerce.track_package import track_package
+from .find_events import find_events
+from .list_agents import list_agents
+from .change_goal import change_goal
+from .transfer_control import transfer_control
+
+from .hr.current_pto import current_pto
+from .hr.book_pto import book_pto
+from .hr.future_pto_calc import future_pto_calc
+from .hr.checkpaybankstatus import checkpaybankstatus
+
 from .fin.check_account_valid import check_account_valid
 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 .find_events import find_events
+
+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
-from .hr.book_pto import book_pto
-from .hr.checkpaybankstatus import checkpaybankstatus
-from .hr.current_pto import current_pto
-from .hr.future_pto_calc import future_pto_calc
-from .list_agents import list_agents
-from .search_fixtures import search_fixtures
-from .search_flights import search_flights
-from .search_trains import book_trains, search_trains
-from .transfer_control import transfer_control


 def get_handler(tool_name: str):
--- a/tools/change_goal.py
+++ b/tools/change_goal.py
@@ -1,8 +1,9 @@
 def change_goal(args: dict) -> dict:
+
    new_goal = args.get("goalID")
    if new_goal is None:
        new_goal = "goal_choose_agent_type"

    return {
        "new_goal": new_goal,
-    }
+    }
--- a/tools/create_invoice.py
+++ b/tools/create_invoice.py
@@ -1,5 +1,4 @@
 import os
-
 import stripe
 from dotenv import load_dotenv

--- a/tools/ecommerce/get_order.py
+++ b/tools/ecommerce/get_order.py
@@ -1,18 +1,16 @@
-import json
 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"
-    )
+    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"]
@@ -20,6 +18,6 @@ def get_order(args: dict) -> dict:
    for order in order_list:
        if order["id"] == order_id:
            return order
-
+        
    return_msg = "Order " + order_id + " not found."
-    return {"error": return_msg}
+    return {"error": return_msg}
--- a/tools/ecommerce/list_orders.py
+++ b/tools/ecommerce/list_orders.py
@@ -1,20 +1,17 @@
-import json
 from pathlib import Path
-
+import json

 def sorting(e):
-    return e["order_date"]
-
+  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"
-    )
+    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"]
@@ -27,6 +24,7 @@ def list_orders(args: dict) -> dict:
    if len(rtn_order_list) > 0:
        rtn_order_list.sort(key=sorting)
        return {"orders": rtn_order_list}
-    else:
+    else:    
        return_msg = "No orders for customer " + email_address + " found."
        return {"error": return_msg}
+
--- a/tools/ecommerce/track_package.py
+++ b/tools/ecommerce/track_package.py
@@ -1,59 +1,49 @@
 import http
-import json
 import os
+import json
+
 from pathlib import Path

-
-# Send back dummy data in the correct format - to use the real API, 1) change this to be track_package_fake and 2) change the below track_package_real to be track_package
+#Send back dummy data in the correct format - to use the real API, 1) change this to be track_package_fake and 2) change the below track_package_real to be track_package 
 def track_package(args: dict) -> dict:
+    
    tracking_id = args.get("tracking_id")
-    file_path = (
-        Path(__file__).resolve().parent.parent / "data" / "dummy_tracking_data.json"
-    )
+    file_path = Path(__file__).resolve().parent.parent / "data" / "dummy_tracking_data.json"
    if not file_path.exists():
        return {"error": "Data file not found."}
-
+    
    with open(file_path, "r") as file:
        data = json.load(file)
    package_list = data["packages"]

    for package in package_list:
        if package["TrackingNumber"] == tracking_id:
-            scheduled_delivery_date = package["ScheduledDeliveryDate"]
-            carrier = package["Carrier"]
-            status_summary = package["StatusSummary"]
-            tracking_details = package.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,
-            }
+                scheduled_delivery_date = package["ScheduledDeliveryDate"]
+                carrier = package["Carrier"]
+                status_summary = package["StatusSummary"]
+                tracking_details = package.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
+                }
+ 
    return_msg = "Package not found with tracking info " + tracking_id
    return {"error": return_msg}

-
-"""Format of response:
+'''Format of response:
 {
    "TrackingNumber": "",
    "Delivered": false,
@@ -104,10 +94,9 @@ def track_package(args: dict) -> dict:
        }
    ]
 }
-"""
-
-
+'''
 def track_package_real(args: dict) -> dict:
+
    tracking_id = args.get("tracking_id")

    api_key = os.getenv("RAPIDAPI_KEY")
@@ -138,17 +127,11 @@ def track_package_real(args: dict) -> dict:
    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
-    ):
+    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}"
-        )
+        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}"

@@ -157,5 +140,5 @@ def track_package_real(args: dict) -> dict:
        "carrier": carrier,
        "status_summary": status_summary,
        "tracking_link": tracking_link,
-        "last_tracking_update": last_tracking_update,
-    }
+        "last_tracking_update": last_tracking_update
+    }
--- a/tools/fin/check_account_valid.py
+++ b/tools/fin/check_account_valid.py
@@ -1,31 +1,24 @@
-import json
 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 check_account_valid(args: dict) -> dict:
+    
    email = args.get("email")
    account_id = args.get("account_id")

-    file_path = (
-        Path(__file__).resolve().parent.parent / "data" / "customer_account_data.json"
-    )
+    file_path = Path(__file__).resolve().parent.parent / "data" / "customer_account_data.json"
    if not file_path.exists():
        return {"error": "Data file not found."}
-
+    
    with open(file_path, "r") as file:
        data = json.load(file)
    account_list = data["accounts"]

    for account in account_list:
        if account["email"] == email or account["account_id"] == account_id:
-            return {"status": "account valid"}
-
-    return_msg = (
-        "Account not found with email address "
-        + email
-        + " or account ID: "
-        + account_id
-    )
-    return {"error": return_msg}
+            return{"status": "account valid"}
+        
+    return_msg = "Account not found with email address " + email + " or account ID: " + account_id
+    return {"error": return_msg}
--- a/tools/fin/get_account_balances.py
+++ b/tools/fin/get_account_balances.py
@@ -1,33 +1,23 @@
-import json
 from pathlib import Path
-
+import json

 # this is made to demonstrate functionality but it could just as durably be an API call
 # this assumes it's a valid account - use check_account_valid() to verify that first
 def get_account_balance(args: dict) -> dict:
+    
    account_key = args.get("email_address_or_account_ID")

-    file_path = (
-        Path(__file__).resolve().parent.parent / "data" / "customer_account_data.json"
-    )
+    file_path = Path(__file__).resolve().parent.parent / "data" / "customer_account_data.json"
    if not file_path.exists():
        return {"error": "Data file not found."}
-
+    
    with open(file_path, "r") as file:
        data = json.load(file)
    account_list = data["accounts"]

    for account in account_list:
        if account["email"] == account_key or account["account_id"] == account_key:
-            return {
-                "name": account["name"],
-                "email": account["email"],
-                "account_id": account["account_id"],
-                "checking_balance": account["checking_balance"],
-                "savings_balance": account["savings_balance"],
-                "bitcoin_balance": account["bitcoin_balance"],
-                "account_creation_date": account["account_creation_date"],
-            }
-
+            return{ "name": account["name"], "email": account["email"], "account_id": account["account_id"], "checking_balance": account["checking_balance"], "savings_balance": account["savings_balance"], "bitcoin_balance": account["bitcoin_balance"], "account_creation_date": account["account_creation_date"] }
+        
    return_msg = "Account not found with for " + account_key
-    return {"error": return_msg}
+    return {"error": return_msg}
--- a/tools/fin/move_money.py
+++ b/tools/fin/move_money.py
@@ -1,12 +1,16 @@
-import json
 import os
-from dataclasses import dataclass
 from pathlib import Path
-
+import json
+from temporalio.client import Client
+from dataclasses import dataclass
+from typing import Optional
+import asyncio
 from temporalio.exceptions import WorkflowAlreadyStartedError
-
 from shared.config import get_temporal_client

+
+from enum import Enum, auto
+
 # enums for the java enum
 # class ExecutionScenarios(Enum):
 #     HAPPY_PATH = 0
@@ -28,6 +32,7 @@ class MoneyMovementWorkflowParameterObj:
 # this is made to demonstrate functionality but it could just as durably be an API call
 # this assumes it's a valid account - use check_account_valid() to verify that first
 async def move_money(args: dict) -> dict:
+
    account_key = args.get("email_address_or_account_ID")
    account_type: str = args.get("accounttype")
    amount = args.get("amount")
@@ -96,6 +101,7 @@ async def move_money(args: dict) -> dict:
 async def start_workflow(
    amount_cents: int, from_account_name: str, to_account_name: str
 ) -> str:
+
    start_real_workflow = os.getenv("FIN_START_REAL_WORKFLOW")
    if start_real_workflow is not None and start_real_workflow.lower() == "false":
        START_REAL_WORKFLOW = False
@@ -122,7 +128,7 @@ async def start_workflow(
                task_queue="MoneyTransferJava",  # Task queue name
            )
            return handle.id
-        except WorkflowAlreadyStartedError:
+        except WorkflowAlreadyStartedError as e:
            existing_handle = client.get_workflow_handle(workflow_id=workflow_id)
            return existing_handle.id
    else:
--- a/tools/fin/submit_loan_application.py
+++ b/tools/fin/submit_loan_application.py
@@ -1,10 +1,18 @@
+from datetime import date, timedelta
 import os
-from dataclasses import dataclass
-from datetime import date
-
+from pathlib import Path
+import json
+from temporalio.client import (
+    Client,
+    WithStartWorkflowOperation,
+    WorkflowHandle,
+    WorkflowUpdateFailedError,
+)
 from temporalio import common
-from temporalio.client import WithStartWorkflowOperation, WorkflowUpdateFailedError
-
+from dataclasses import dataclass
+from typing import Optional
+import asyncio
+from temporalio.exceptions import WorkflowAlreadyStartedError
 from shared.config import get_temporal_client


@@ -16,55 +24,39 @@ class TransactionRequest:
    sourceAccount: str
    targetAccount: str

-
@dataclass
 class TxResult:
    transactionId: str
    status: str

-
-# demonstrate starting a workflow and early return pattern while the workflow continues
+#demonstrate starting a workflow and early return pattern while the workflow continues
 async def submit_loan_application(args: dict) -> dict:
    account_key = args.get("email_address_or_account_ID")
    amount = args.get("amount")

-    loan_status: dict = await start_workflow(amount=amount, account_name=account_key)
+    loan_status: dict = await start_workflow(amount=amount,account_name=account_key)

    if loan_status.get("error") is None:
-        return {
-            "status": loan_status.get("loan_application_status"),
-            "detailed_status": loan_status.get("application_details"),
-            "next_step": loan_status.get("advisement"),
-            "confirmation_id": loan_status.get("transaction_id"),
-        }
+        return {'status': loan_status.get("loan_application_status"), 'detailed_status': loan_status.get("application_details"), 'next_step': loan_status.get("advisement"), 'confirmation_id': loan_status.get("transaction_id")}  
    else:
        print(loan_status)
        return loan_status
-
+    

 # Async function to start workflow
-async def start_workflow(
-    amount: str,
-    account_name: str,
-) -> dict:
+async def start_workflow(amount: str, account_name: str, )-> dict:
+ 
    start_real_workflow = os.getenv("FIN_START_REAL_WORKFLOW")
    if start_real_workflow is not None and start_real_workflow.lower() == "false":
-        # START_REAL_WORKFLOW = False
-        return {
-            "loan_application_status": "applied",
-            "application_details": "loan application is submitted and initial validation is complete",
-            "transaction_id": "APPLICATION" + account_name,
-            "advisement": "You'll receive a confirmation for final approval in three business days",
-        }
+        START_REAL_WORKFLOW = False
+        return {'loan_application_status': "applied", 'application_details': "loan application is submitted and initial validation is complete",'transaction_id': "APPLICATION"+account_name, 'advisement': "You'll receive a confirmation for final approval in three business days", }  
    else:
-        # START_REAL_WORKFLOW = True
-        # Connect to Temporal
+        START_REAL_WORKFLOW = True
+         # Connect to Temporal 
        client = await get_temporal_client()
-
+    
        # Define the workflow ID and task queue
-        workflow_id = (
-            "LOAN_APPLICATION-" + account_name + "-" + date.today().strftime("%Y-%m-%d")
-        )
+        workflow_id = "LOAN_APPLICATION-"+account_name+"-"+date.today().strftime('%Y-%m-%d')
        task_queue = "LatencyOptimizationTEST"

        # Create a TransactionRequest (matching the Java workflow's expected input)
@@ -91,27 +83,21 @@ async def start_workflow(
                )
            )
        except WorkflowUpdateFailedError:
-            print("aww man got exception WorkflowUpdateFailedError")
+            print("aww man got exception WorkflowUpdateFailedError" )
            tx_result = None
            return_msg = "Loan could not be processed for " + account_name
            return {"error": return_msg}

        workflow_handle = await start_op.workflow_handle()
-        print(f"Workflow started with ID: {workflow_handle.id}")
        print(tx_result)

-        print(
-            f"Update result: Transaction ID = {tx_result.transactionId}, Message = {tx_result.status}"
-        )
+        print(f"Update result: Transaction ID = {tx_result.transactionId}, Message = {tx_result.status}")

        # Optionally, wait for the workflow to complete and get the final result
        # final_result = await handle.result()
        # print(f"Workflow completed with result: {final_result}")

-        # return {'status': loan_status.get("loan_status"), 'detailed_status': loan_status.get("results"), 'next_step': loan_status.get("advisement"), 'confirmation_id': loan_status.get("workflowID")}
-        return {
-            "loan_application_status": "applied",
-            "application_details": "loan application is submitted and initial validation is complete",
-            "transaction_id": tx_result.transactionId,
-            "advisement": "You'll receive a confirmation for final approval in three business days",
-        }
+
+        # return {'status': loan_status.get("loan_status"), 'detailed_status': loan_status.get("results"), 'next_step': loan_status.get("advisement"), 'confirmation_id': loan_status.get("workflowID")}  
+        return {'loan_application_status': "applied", 'application_details': "loan application is submitted and initial validation is complete",'transaction_id': tx_result.transactionId, 'advisement': "You'll receive a confirmation for final approval in three business days", }  
+    
--- a/tools/find_events.py
+++ b/tools/find_events.py
@@ -1,6 +1,6 @@
-import json
 from datetime import datetime
 from pathlib import Path
+import json


 def find_events(args: dict) -> dict:
--- a/tools/give_hint.py
+++ b/tools/give_hint.py
@@ -1,10 +1,10 @@
 TREASURE_LOCATION = {
-    "address": "300 Lenora",
-    "city": "Seattle",
-    "state_full": "Washington",
-    "state_abbrev": "WA",
-    "zip": "98121",
-    "country": "USA",
+        "address": "300 Lenora",
+        "city": "Seattle",
+        "state_full": "Washington",
+        "state_abbrev": "WA",
+        "zip": "98121",
+        "country": "USA"
 }

 HINTS = [
@@ -12,8 +12,8 @@ HINTS = [
    "state of " + TREASURE_LOCATION["state_full"],
    "city of " + TREASURE_LOCATION["city"],
    "at a company HQ",
-    "The company's tech traces its roots to a project called Cadence",  # thanks, Grok
-    "The company offers a tool that lets developers write code as if it's running forever, no matter what crashes",  # thanks, Grok
+    "The company's tech traces its roots to a project called Cadence", #thanks, Grok
+    "The company offers a tool that lets developers write code as if it's running forever, no matter what crashes", #thanks, Grok
 ]
 ''' Additional Grok provided hints about Temporal:
 "This company was founded by two engineers who previously worked on a system named after a South American river at Uber."
@@ -26,14 +26,16 @@ HINTS = [
 "They’re backed by big venture capital names like Sequoia, betting on their vision for reliable software."
 "The company’s name might remind you of a word for something fleeting, yet their tech is built to last."'''

-
 def give_hint(args: dict) -> dict:
    hint_total = args.get("hint_total")
    if hint_total is None:
        hint_total = 0
-
+    
    index = hint_total % len(HINTS)
    hint_text = HINTS[index]

    hint_total = hint_total + 1
-    return {"hint_number": hint_total, "hint": hint_text}
+    return {
+        "hint_number": hint_total,
+        "hint": hint_text 
+    }
--- a/tools/goal_registry.py
+++ b/tools/goal_registry.py
@@ -1,8 +1,7 @@
 import os
 from typing import List
-
-import tools.tool_registry as tool_registry
 from models.tool_definitions import AgentGoal
+import tools.tool_registry as tool_registry

 # Turn on Silly Mode - this should be a description of the persona you'd like the bot to have and can be a single word or a phrase.
 # Example if you want the bot to be a specific person, like Mario or Christopher Walken, or to describe a specific tone:
@@ -115,10 +114,10 @@ goal_match_train_invoice = AgentGoal(
    ],
    description="The user wants to book a trip to a city in the UK around the dates of a premier league match. "
    "Help the user find a premier league match to attend, search and book trains for that match and offers to invoice them for the cost of train tickets. "
-    "The user lives in London. Premier league fixtures may be mocked data, so don't worry about valid season dates and teams. "
+    "The user lives in London. "
    "Gather args for these tools in order, ensuring you move the user from one tool to the next: "
    "1. SearchFixtures: Search for fixtures for a team within a specified date range. The user might ask questions about the matches dates and locations to decide on where to go. "
-    "2. SearchTrains: Search for trains to the city of the match. Ensure you list them for the customer to choose from "
+    "2. SearchTrains: Search for trains to the city of the match and list them for the customer to choose from "
    "3. BookTrains: Book the train tickets, used to invoice the user for the cost of the train tickets "
    "4. CreateInvoice: Invoices the user for the cost of train tickets, with total and details inferred from the conversation history ",
    starter_prompt=starter_prompt_generic,
@@ -311,7 +310,7 @@ goal_fin_check_account_balances = AgentGoal(
 )

 # this tool checks account balances, and uses ./data/customer_account_data.json as dummy data
-# it also uses a separate workflow/tool, see ./SETUP.md for details
+# it also uses a separate workflow/tool, see ./setup.md for details
 goal_fin_move_money = AgentGoal(
    id="goal_fin_move_money",
    category_tag="fin",
@@ -351,7 +350,7 @@ goal_fin_move_money = AgentGoal(
 )

 # this starts a loan approval process
-# it also uses a separate workflow/tool, see ./SETUP.md for details
+# it also uses a separate workflow/tool, see ./setup.md for details
 goal_fin_loan_application = AgentGoal(
    id="goal_fin_loan_application",
    category_tag="fin",
@@ -490,6 +489,6 @@ if multi_goal_mode:
            if tool.name == "ListAgents":
                list_agents_found = True
                continue
-        if list_agents_found is False:
+        if list_agents_found == False:
            goal.tools.append(tool_registry.list_agents_tool)
            continue
--- a/tools/guess_location.py
+++ b/tools/guess_location.py
@@ -1,7 +1,7 @@
 from .give_hint import TREASURE_LOCATION

-
 def guess_location(args: dict) -> dict:
+    
    guess_address = args.get("address").lower()
    guess_city = args.get("city").lower()
    guess_state = args.get("state").lower()
@@ -11,12 +11,8 @@ def guess_location(args: dict) -> dict:
    else:
        compare_state = TREASURE_LOCATION.get("state_full").lower()

-    # Check for the street address to be included in the guess to account for "st" vs "street" or leaving Street off entirely
-    if (
-        TREASURE_LOCATION.get("address").lower() in guess_address
-        and TREASURE_LOCATION.get("city").lower() == guess_city
-        and compare_state == guess_state
-    ):
+    #Check for the street address to be included in the guess to account for "st" vs "street" or leaving Street off entirely
+    if TREASURE_LOCATION.get("address").lower() in guess_address and TREASURE_LOCATION.get("city").lower() == guess_city and compare_state == guess_state:
        return {"treasure_found": "True"}
    else:
-        return {"treasure_found": "False"}
+        return {"treasure_found": "False"}
--- a/tools/hr/book_pto.py
+++ b/tools/hr/book_pto.py
@@ -1,10 +1,11 @@
 def book_pto(args: dict) -> dict:
+    
    email = args.get("email")
    start_date = args.get("start_date")
    end_date = args.get("end_date")

-    print(
-        f"[BookPTO] Totally would send an email confirmation of PTO from {start_date} to {end_date} to {email} here!"
-    )
+    print(f"[BookPTO] Totally would send an email confirmation of PTO from {start_date} to {end_date} to {email} here!")

-    return {"status": "success"}
+    return {
+        "status": "success"
+    }
--- a/tools/hr/checkpaybankstatus.py
+++ b/tools/hr/checkpaybankstatus.py
@@ -1,4 +1,9 @@
+from pathlib import Path
+import json
+
+
 def checkpaybankstatus(args: dict) -> dict:
+    
    email = args.get("email")

    if email == "grinch@grinch.com":
@@ -7,4 +12,4 @@ def checkpaybankstatus(args: dict) -> dict:

    # could do logic here or look up data but for now everyone but the grinch is getting paid
    return_msg = "connected"
-    return {"status": return_msg}
+    return {"status": return_msg}
--- a/tools/hr/current_pto.py
+++ b/tools/hr/current_pto.py
@@ -1,27 +1,26 @@
-import json
 from pathlib import Path
+import json


 def current_pto(args: dict) -> dict:
+    
    email = args.get("email")

-    file_path = (
-        Path(__file__).resolve().parent.parent / "data" / "employee_pto_data.json"
-    )
+    file_path = Path(__file__).resolve().parent.parent / "data" / "employee_pto_data.json"
    if not file_path.exists():
        return {"error": "Data file not found."}
-
+    
    data = json.load(open(file_path))
    employee_list = data["theCompany"]["employees"]

    for employee in employee_list:
        if employee["email"] == email:
            num_hours = int(employee["currentPTOHrs"])
-            num_days = float(num_hours / 8)
+            num_days = float(num_hours/8)
            return {
                "num_hours": num_hours,
                "num_days": num_days,
            }
-
+        
    return_msg = "Employee not found with email address " + email
-    return {"error": return_msg}
+    return {"error": return_msg}
--- a/tools/hr/future_pto_calc.py
+++ b/tools/hr/future_pto_calc.py
@@ -1,59 +1,43 @@
 import json
-from datetime import date, datetime
-from pathlib import Path
-
 import pandas
+from pathlib import Path
+from datetime import date, datetime
 from dateutil.relativedelta import relativedelta


 def future_pto_calc(args: dict) -> dict:
-    file_path = (
-        Path(__file__).resolve().parent.parent / "data" / "employee_pto_data.json"
-    )
+    
+    file_path = Path(__file__).resolve().parent.parent / "data" / "employee_pto_data.json"
    if not file_path.exists():
        return {"error": "Data file not found."}
-
+    
    start_date = datetime.strptime(args.get("start_date"), "%Y-%m-%d").date()
    end_date = datetime.strptime(args.get("end_date"), "%Y-%m-%d").date()
    email = args.get("email")

-    # Next, set up the ability to calculate how much PTO will be added to the user's total by the start of the PTO request
+    #Next, set up the ability to calculate how much PTO will be added to the user's total by the start of the PTO request
    today = date.today()

    if today > start_date:
-        return_msg = (
-            "PTO start date " + args.get("start_date") + "cannot be in the past"
-        )
+        return_msg = "PTO start date " + args.get("start_date") + "cannot be in the past"
        return {"error": return_msg}
-
+    
    if end_date < start_date:
-        return_msg = (
-            "PTO end date "
-            + args.get("end_date")
-            + " must be after PTO start date "
-            + args.get("start_date")
-        )
+        return_msg = "PTO end date " + args.get("end_date") + " must be after PTO start date " + args.get("start_date")
        return {"error": return_msg}
-
-    # Get the number of business days, and then business hours (assume 8 hr biz day), included in the PTO request
-    biz_days_of_request = len(
-        pandas.bdate_range(start=start_date, end=end_date, inclusive="both")
-    )
+    
+    #Get the number of business days, and then business hours (assume 8 hr biz day), included in the PTO request
+    biz_days_of_request = len(pandas.bdate_range(start=start_date, end=end_date, inclusive="both"))
    if biz_days_of_request == 0:
-        return_msg = (
-            "There are no business days between "
-            + args.get("start_date")
-            + " and "
-            + args.get("end_date")
-        )
+        return_msg = "There are no business days between " + args.get("start_date") + " and " + args.get("end_date")
        return {"error": return_msg}
    biz_hours_of_request = biz_days_of_request * 8
-
-    # Assume PTO is added on the first of every month - month math compares rolling dates, so compare the PTO request with the first day of the current month.
+    
+    #Assume PTO is added on the first of every month - month math compares rolling dates, so compare the PTO request with the first day of the current month.
    today_first_of_month = date(today.year, today.month, 1)
    time_difference = relativedelta(start_date, today_first_of_month)
    months_to_accrue = time_difference.years * 12 + time_difference.months
-
+    
    data = json.load(open(file_path))
    employee_list = data["theCompany"]["employees"]

@@ -63,14 +47,12 @@ def future_pto_calc(args: dict) -> dict:
        if employee["email"] == email:
            current_pto_hours = int(employee["currentPTOHrs"])
            hrs_added_per_month = int(employee["hrsAddedPerMonth"])
-            pto_available_at_start = current_pto_hours + (
-                months_to_accrue * hrs_added_per_month
-            )
+            pto_available_at_start = current_pto_hours + (months_to_accrue * hrs_added_per_month)
            pto_hrs_remaining_after = pto_available_at_start - biz_hours_of_request
            if pto_hrs_remaining_after >= 0:
                enough_pto = True
            return {
-                "enough_pto": enough_pto,
+                "enough_pto": enough_pto, 
                "pto_hrs_remaining_after": str(pto_hrs_remaining_after),
            }

--- a/tools/list_agents.py
+++ b/tools/list_agents.py
@@ -1,23 +1,19 @@
 import os
-
 import tools.goal_registry as goals

-
 def list_agents(args: dict) -> dict:
+
    goal_categories_start = os.getenv("GOAL_CATEGORIES")
    if goal_categories_start is None:
-        goal_categories = ["all"]  # default to 'all' categories
+        goal_categories = ["all"] # default to 'all' categories
    else:
-        goal_categories_start.strip().lower()  # handle extra spaces or non-lowercase
+        goal_categories_start.strip().lower() # handle extra spaces or non-lowercase
        goal_categories = goal_categories_start.split(",")

    # if multi-goal-mode, add agent_selection as a goal (defaults to True)
-    if "agent_selection" not in goal_categories:
-        first_goal_value = os.getenv("AGENT_GOAL")
-        if (
-            first_goal_value is None
-            or first_goal_value.lower() == "goal_choose_agent_type"
-        ):
+    if "agent_selection" not in goal_categories :
+        first_goal_value = os.getenv("AGENT_GOAL")        
+        if first_goal_value is None or first_goal_value.lower() == "goal_choose_agent_type":
            goal_categories.append("agent_selection")

    # always show goals labeled as "system," like the goal chooser
@@ -37,7 +33,7 @@ def list_agents(args: dict) -> dict:
                        "goal_id": goal.id,
                        "agent_description": goal.agent_friendly_description,
                    }
-                )
+            )
    return {
        "agents": agents,
    }
--- a/tools/search_fixtures.py
+++ b/tools/search_fixtures.py
@@ -1,264 +1,64 @@
 import os
-import random
-from datetime import date, datetime, timedelta
-
 import requests
+from datetime import datetime, timedelta
 from dotenv import load_dotenv

-PREMIER_LEAGUE_CLUBS_DATA = [
-    {"name": "Arsenal FC", "stadium": "Emirates Stadium"},
-    {"name": "Aston Villa FC", "stadium": "Villa Park"},
-    {"name": "AFC Bournemouth", "stadium": "Vitality Stadium"},
-    {"name": "Brentford FC", "stadium": "Gtech Community Stadium"},
-    {"name": "Brighton & Hove Albion FC", "stadium": "American Express Stadium"},
-    {"name": "Chelsea FC", "stadium": "Stamford Bridge"},
-    {"name": "Crystal Palace FC", "stadium": "Selhurst Park"},
-    {"name": "Everton FC", "stadium": "Goodison Park"},
-    {"name": "Fulham FC", "stadium": "Craven Cottage"},
-    {"name": "Ipswich Town FC", "stadium": "Portman Road"},
-    {"name": "Leicester City FC", "stadium": "King Power Stadium"},
-    {"name": "Liverpool FC", "stadium": "Anfield"},
-    {"name": "Manchester City FC", "stadium": "Etihad Stadium"},
-    {"name": "Manchester United FC", "stadium": "Old Trafford"},
-    {"name": "Newcastle United FC", "stadium": "St James' Park"},
-    {"name": "Nottingham Forest FC", "stadium": "City Ground"},
-    {"name": "Southampton FC", "stadium": "St Mary's Stadium"},
-    {"name": "Tottenham Hotspur FC", "stadium": "Tottenham Hotspur Stadium"},
-    {"name": "West Ham United FC", "stadium": "London Stadium"},
-    {"name": "Wolverhampton Wanderers FC", "stadium": "Molineux Stadium"},
-]
-
-
-def get_future_matches(
-    team_name: str,
-    all_clubs_data: list,
-    num_matches: int = 12,
-    date_from: date = None,
-    date_to: date = None,
-) -> list:
-    """Generate a set of future Premier League matches for ``team_name``.
-
-    This is a purely mocked schedule. It returns up to ``num_matches``
-    fixtures, respecting the ``date_from`` and ``date_to`` constraints.
-    Matches are typically on Saturdays or Sundays.
-    """
-    matches = []
-
-    team_details = next((c for c in all_clubs_data if c["name"] == team_name), None)
-    if not team_details:
-        return []
-
-    opponents_pool = [c for c in all_clubs_data if c["name"] != team_name]
-    if not opponents_pool:
-        return []
-
-    # Determine the maximum number of matches we can generate based on opponents
-    # and the requested num_matches
-    num_actual_matches_to_generate = min(num_matches, len(opponents_pool))
-    if num_actual_matches_to_generate == 0:
-        return []
-
-    # Shuffle opponents once and pick them sequentially
-    random.shuffle(opponents_pool)  # Shuffle in place
-
-    # Determine the initial Saturday for match week consideration
-    today_date = date.today()
-    # Default to next Saturday
-    current_match_week_saturday = today_date + timedelta(
-        days=(5 - today_date.weekday() + 7) % 7
-    )
-
-    # If today is Saturday and it's late evening, or if today is Sunday,
-    # advance to the following Saturday.
-    now_time = datetime.now().time()
-    if (
-        today_date.weekday() == 5
-        and now_time > datetime.strptime("20:00", "%H:%M").time()
-    ) or (today_date.weekday() == 6):
-        current_match_week_saturday += timedelta(days=7)
-
-    # If date_from is specified, ensure our starting Saturday is not before it.
-    if date_from:
-        if current_match_week_saturday < date_from:
-            current_match_week_saturday = date_from
-        # Align current_match_week_saturday to be a Saturday on or after the potentially adjusted date
-        current_match_week_saturday += timedelta(
-            days=(5 - current_match_week_saturday.weekday() + 7) % 7
-        )
-
-    opponent_idx = 0
-    while len(matches) < num_actual_matches_to_generate and opponent_idx < len(
-        opponents_pool
-    ):
-        # If the current week's Saturday is already past date_to, stop.
-        if date_to and current_match_week_saturday > date_to:
-            break
-
-        opponent_details = opponents_pool[opponent_idx]
-        is_saturday_game = random.choice([True, True, False])
-        actual_match_date = None
-        kick_off_time = ""
-
-        if is_saturday_game:
-            actual_match_date = current_match_week_saturday
-            kick_off_time = random.choice(["12:30", "15:00", "17:30"])
-        else:  # Sunday game
-            actual_match_date = current_match_week_saturday + timedelta(days=1)
-            kick_off_time = random.choice(["14:00", "16:30"])
-
-        # Check if this specific match date is within the date_to constraint
-        if date_to and actual_match_date > date_to:
-            # If this game is too late, try the next week if possible.
-            # (This mainly affects Sunday games if Saturday was the last valid day)
-            current_match_week_saturday += timedelta(days=7)
-            continue  # Skip adding this match, try next week.
-
-        match_datetime_gmt = (
-            f"{actual_match_date.strftime('%Y-%m-%d')} {kick_off_time} GMT"
-        )
-        is_home_match = random.choice([True, False])
-
-        if is_home_match:
-            team1_name = team_details["name"]
-            team2_name = opponent_details["name"]
-            stadium_name = team_details["stadium"]
-        else:
-            team1_name = opponent_details["name"]
-            team2_name = team_details["name"]
-            stadium_name = opponent_details["stadium"]
-
-        matches.append(
-            {
-                "team1": team1_name,
-                "team2": team2_name,
-                "stadium": stadium_name,
-                "datetime_gmt": match_datetime_gmt,
-            }
-        )
-        opponent_idx += 1
-        current_match_week_saturday += timedelta(
-            days=7
-        )  # Advance to next week's Saturday
-
-    return matches
-
-
 BASE_URL = "https://api.football-data.org/v4"


 def search_fixtures(args: dict) -> dict:
    load_dotenv(override=True)
-    api_key = os.getenv("FOOTBALL_DATA_API_KEY")
+    api_key = os.getenv("FOOTBALL_DATA_API_KEY", "YOUR_DEFAULT_KEY")

    team_name = args.get("team")
    date_from_str = args.get("date_from")
    date_to_str = args.get("date_to")
-
-    if not team_name:
-        return {"error": "Team name is required."}
-
-    parsed_date_from = None
-    if date_from_str:
-        try:
-            parsed_date_from = datetime.strptime(date_from_str, "%Y-%m-%d").date()
-        except ValueError:
-            return {
-                "error": f"Invalid date_from: '{date_from_str}'. Expected format YYYY-MM-DD."
-            }
-
-    parsed_date_to = None
-    if date_to_str:
-        try:
-            parsed_date_to = datetime.strptime(date_to_str, "%Y-%m-%d").date()
-        except ValueError:
-            return {
-                "error": f"Invalid date_to: '{date_to_str}'. Expected format YYYY-MM-DD."
-            }
-
-    if parsed_date_from and parsed_date_to and parsed_date_from > parsed_date_to:
-        return {"error": "date_from cannot be after date_to."}
-
-    # If no API key, fall back to mocked data
-    if not api_key:
-        # Use the parsed date objects (which can be None)
-        fixtures = get_future_matches(
-            team_name,
-            PREMIER_LEAGUE_CLUBS_DATA,
-            date_from=parsed_date_from,
-            date_to=parsed_date_to,
-            # num_matches can be passed explicitly if needed, otherwise defaults to 12
-        )
-        if not fixtures:
-            # Check if the team name itself was invalid, as get_future_matches returns [] for that too
-            team_details_check = next(
-                (c for c in PREMIER_LEAGUE_CLUBS_DATA if c["name"] == team_name), None
-            )
-            if not team_details_check:
-                return {"error": f"Team '{team_name}' not found in mocked data."}
-            # If team is valid, an empty fixtures list means no matches fit the criteria (e.g., date range)
-        return {"fixtures": fixtures}
-
-    # API Key is present, proceed with API logic
-    # The API requires both date_from and date_to
-    if not parsed_date_from or not parsed_date_to:
-        return {
-            "error": "Both date_from and date_to (YYYY-MM-DD) are required for API search."
-        }
-
    headers = {"X-Auth-Token": api_key}
-    # For API calls, team name matching might be case-insensitive or require specific handling
-    # The existing logic uses team_name.lower() for the API search path later.
+    team_name = team_name.lower()
+
+    try:
+        date_from = datetime.strptime(date_from_str, "%Y-%m-%d")
+        date_to = datetime.strptime(date_to_str, "%Y-%m-%d")
+    except ValueError:
+        return {
+            "error": "Invalid date provided. Expected format YYYY-MM-DD for both date_from and date_to."
+        }

    # Fetch team ID
    teams_response = requests.get(f"{BASE_URL}/competitions/PL/teams", headers=headers)
    if teams_response.status_code != 200:
-        return {
-            "error": f"Failed to fetch teams data from API (status {teams_response.status_code})."
-        }
+        return {"error": "Failed to fetch teams data."}

    teams_data = teams_response.json()
    team_id = None
-    # Using lower() for comparison, assuming API team names might have varied casing
-    # or the input team_name might not be exact.
-    # The `ToolDefinition` lists exact names, so direct match might also be an option.
-    for team_api_data in teams_data.get("teams", []):
-        if team_name.lower() in team_api_data.get("name", "").lower():
-            team_id = team_api_data["id"]
+    for team in teams_data["teams"]:
+        if team_name in team["name"].lower():
+            team_id = team["id"]
            break

    if not team_id:
-        return {"error": f"Team '{team_name}' not found via API."}
+        return {"error": "Team not found."}

-    date_from_formatted = parsed_date_from.strftime("%Y-%m-%d")
-    date_to_formatted = parsed_date_to.strftime("%Y-%m-%d")
+    date_from_formatted = date_from.strftime("%Y-%m-%d")
+    date_to_formatted = date_to.strftime("%Y-%m-%d")
    fixtures_url = f"{BASE_URL}/teams/{team_id}/matches?dateFrom={date_from_formatted}&dateTo={date_to_formatted}"
-    # print(fixtures_url) # Keep for debugging if necessary
+    print(fixtures_url)

    fixtures_response = requests.get(fixtures_url, headers=headers)
    if fixtures_response.status_code != 200:
-        return {
-            "error": f"Failed to fetch fixtures data from API (status {fixtures_response.status_code})."
-        }
+        return {"error": "Failed to fetch fixtures data."}

    fixtures_data = fixtures_response.json()
    matching_fixtures = []

    for match in fixtures_data.get("matches", []):
-        # Ensure match datetime parsing is robust
-        try:
-            match_datetime_utc = datetime.strptime(
-                match["utcDate"], "%Y-%m-%dT%H:%M:%SZ"
-            )
-        except (ValueError, TypeError):
-            # Skip malformed match entries or log an error
-            continue
-
-        if match.get("competition", {}).get("code") == "PL":
+        match_datetime = datetime.strptime(match["utcDate"], "%Y-%m-%dT%H:%M:%SZ")
+        if match["competition"]["code"] == "PL":
            matching_fixtures.append(
                {
-                    "date": match_datetime_utc.strftime("%Y-%m-%d"),
-                    "homeTeam": match.get("homeTeam", {}).get("name", "N/A"),
-                    "awayTeam": match.get("awayTeam", {}).get("name", "N/A"),
+                    "date": match_datetime.strftime("%Y-%m-%d"),
+                    "homeTeam": match["homeTeam"]["name"],
+                    "awayTeam": match["awayTeam"]["name"],
                }
            )

@@ -282,69 +82,34 @@ def search_fixtures_example(args: dict) -> dict:

    # Validate dates
    try:
-        # Ensure date strings are not None before parsing
-        if date_from_str is None or date_to_str is None:
-            raise ValueError("Date strings cannot be None")
-        date_from_obj = datetime.strptime(date_from_str, "%Y-%m-%d")
-        date_to_obj = datetime.strptime(date_to_str, "%Y-%m-%d")
+        date_from = datetime.strptime(date_from_str, "%Y-%m-%d")
+        date_to = datetime.strptime(date_to_str, "%Y-%m-%d")
    except ValueError:
        return {
            "error": "Invalid date provided. Expected format YYYY-MM-DD for both date_from and date_to."
        }

    # Calculate 3 reasonable fixture dates within the given range
-    date_range = (date_to_obj - date_from_obj).days
-    if date_range < 0:  # date_from is after date_to
-        return {"fixtures": []}  # No fixtures possible
-
-    fixture_dates_timestamps = []
+    date_range = (date_to - date_from).days
    if date_range < 21:
-        # If range is less than 3 weeks, use evenly spaced fixtures if possible
-        if date_range >= 2:  # Need at least some gap for 3 fixtures
-            fixture_dates_timestamps = [
-                date_from_obj
-                + timedelta(days=max(0, date_range // 4)),  # Closer to start
-                date_from_obj + timedelta(days=max(1, date_range // 2)),  # Middle
-                date_to_obj - timedelta(days=max(0, date_range // 4)),  # Closer to end
-            ]
-        elif date_range == 1:  # Only two days
-            fixture_dates_timestamps = [date_from_obj, date_to_obj]
-        elif date_range == 0:  # Only one day
-            fixture_dates_timestamps = [date_from_obj]
-        else:  # date_range is negative, handled above, or 0 (single day)
-            fixture_dates_timestamps = [date_from_obj] if date_range == 0 else []
-
+        # If range is less than 3 weeks, use evenly spaced fixtures
+        fixture_dates = [
+            date_from + timedelta(days=max(1, date_range // 3)),
+            date_from + timedelta(days=max(2, date_range * 2 // 3)),
+            date_to - timedelta(days=min(2, date_range // 4)),
+        ]
    else:
-        # Otherwise space them out by weeks, ensuring they are within the bounds
-        d1 = date_from_obj + timedelta(days=7)
-        d2 = date_from_obj + timedelta(days=14)
-        d3 = date_to_obj - timedelta(days=7)  # Potential third game from the end
+        # Otherwise space them out by weeks
+        fixture_dates = [
+            date_from + timedelta(days=7),
+            date_from + timedelta(days=14),
+            date_to - timedelta(days=7),
+        ]

-        fixture_dates_timestamps.append(d1)
-        if d2 <= date_to_obj and d2 > d1:  # ensure d2 is valid and distinct
-            fixture_dates_timestamps.append(d2)
-        if (
-            d3 >= date_from_obj and d3 > d2 and d3 <= date_to_obj
-        ):  # ensure d3 is valid and distinct
-            fixture_dates_timestamps.append(d3)
-        elif (
-            d3 < date_from_obj and len(fixture_dates_timestamps) < 3
-        ):  # if d3 is too early, try using date_to_obj itself if distinct
-            if date_to_obj not in fixture_dates_timestamps:
-                fixture_dates_timestamps.append(date_to_obj)
-
-    # Ensure unique dates and sort, then take up to 3.
-    fixture_dates_timestamps = sorted(
-        list(
-            set(
-                f_date
-                for f_date in fixture_dates_timestamps
-                if date_from_obj <= f_date <= date_to_obj
-            )
-        )
-    )
-    fixture_dates_final = fixture_dates_timestamps[:3]
+    # Ensure we only have 3 dates
+    fixture_dates = fixture_dates[:3]

+    # Expanded pool of opponent teams to avoid team playing against itself
    all_opponents = [
        "Manchester United FC",
        "Leicester City FC",
@@ -355,35 +120,35 @@ def search_fixtures_example(args: dict) -> dict:
        "Tottenham Hotspur FC",
        "West Ham United FC",
        "Everton FC",
-        "Generic Opponent A",
-        "Generic Opponent B",
-        "Generic Opponent C",  # Fallbacks
    ]

+    # Select opponents that aren't the same as the requested team
    available_opponents = [
        team for team in all_opponents if team.lower() != team_name.lower()
    ]

-    # Ensure we have enough opponents for the number of fixtures we'll generate
-    if len(available_opponents) < len(fixture_dates_final):
-        needed = len(fixture_dates_final) - len(available_opponents)
-        for i in range(needed):
-            available_opponents.append(f"Placeholder Opponent {i+1}")
+    # Ensure we have at least 3 opponents
+    if len(available_opponents) < 3:
+        # Add generic opponents if needed
+        additional_teams = [f"Opponent {i} FC" for i in range(1, 4)]
+        available_opponents.extend(additional_teams)

-    opponents = available_opponents[: len(fixture_dates_final)]
+    # Take only the first 3 opponents
+    opponents = available_opponents[:3]

+    # Generate fixtures - always exactly 3
    fixtures = []
-    for i, fixture_date_obj in enumerate(fixture_dates_final):
-        if i >= len(opponents):  # Should not happen with the logic above
-            break
-        date_str = fixture_date_obj.strftime("%Y-%m-%d")
-        if i % 2 == 0:  # Home game
-            fixtures.append(
-                {"date": date_str, "homeTeam": team_name, "awayTeam": opponents[i]}
-            )
-        else:  # Away game
+    for i, fixture_date in enumerate(fixture_dates):
+        date_str = fixture_date.strftime("%Y-%m-%d")
+
+        # Alternate between home and away games
+        if i % 2 == 0:
            fixtures.append(
                {"date": date_str, "homeTeam": opponents[i], "awayTeam": team_name}
            )
+        else:
+            fixtures.append(
+                {"date": date_str, "homeTeam": team_name, "awayTeam": opponents[i]}
+            )

    return {"fixtures": fixtures}
--- a/tools/search_flights.py
+++ b/tools/search_flights.py
@@ -1,9 +1,8 @@
-import http.client
-import json
 import os
-import urllib.parse
-
+import json
+import http.client
 from dotenv import load_dotenv
+import urllib.parse


 def search_airport(query: str) -> list:
--- a/tools/tool_registry.py
+++ b/tools/tool_registry.py
@@ -1,4 +1,4 @@
-from models.tool_definitions import ToolArgument, ToolDefinition
+from models.tool_definitions import ToolDefinition, ToolArgument

 # ----- System tools -----
 list_agents_tool = ToolDefinition(
@@ -90,7 +90,7 @@ search_flights_tool = ToolDefinition(

 search_trains_tool = ToolDefinition(
    name="SearchTrains",
-    description="Search for trains between two English cities. Returns a list of train information for the user to choose from. Present the list to the user.",
+    description="Search for trains between two English cities. Returns a list of train information for the user to choose from.",
    arguments=[
        ToolArgument(
            name="origin",
@@ -156,7 +156,7 @@ create_invoice_tool = ToolDefinition(

 search_fixtures_tool = ToolDefinition(
    name="SearchFixtures",
-    description="Search for upcoming fixtures for a given team within a date range inferred from the user's description. Ignore valid premier league dates. Valid teams this season are Arsenal FC, Aston Villa FC, AFC Bournemouth, Brentford FC, Brighton & Hove Albion FC, Chelsea FC, Crystal Palace FC, Everton FC, Fulham FC, Ipswich Town FC, Leicester City FC, Liverpool FC, Manchester City FC, Manchester United FC, Newcastle United FC, Nottingham Forest FC, Southampton FC, Tottenham Hotspur FC, West Ham United FC, Wolverhampton Wanderers FC",
+    description="Search for upcoming fixtures for a given team within a date range inferred from the user's description. Valid teams this 24/25 season are Arsenal FC, Aston Villa FC, AFC Bournemouth, Brentford FC, Brighton & Hove Albion FC, Chelsea FC, Crystal Palace FC, Everton FC, Fulham FC, Ipswich Town FC, Leicester City FC, Liverpool FC, Manchester City FC, Manchester United FC, Newcastle United FC, Nottingham Forest FC, Southampton FC, Tottenham Hotspur FC, West Ham United FC, Wolverhampton Wanderers FC",
    arguments=[
        ToolArgument(
            name="team",
--- a/tools/transfer_control.py
+++ b/tools/transfer_control.py
@@ -1,7 +1,7 @@
 import shared.config

-
 def transfer_control(args: dict) -> dict:
+
    return {
        "new_goal": shared.config.AGENT_GOAL,
-    }
+    }
--- a/workflows/agent_goal_workflow.py
+++ b/workflows/agent_goal_workflow.py
@@ -1,35 +1,31 @@
 from collections import deque
 from datetime import timedelta
-from typing import Any, Deque, Dict, List, Optional, TypedDict, Union
+from typing import Dict, Any, Union, List, Optional, Deque, TypedDict

-from temporalio import workflow
 from temporalio.common import RetryPolicy
+from temporalio import workflow

-from models.data_types import (
-    ConversationHistory,
-    EnvLookupInput,
-    EnvLookupOutput,
-    NextStep,
-    ValidationInput,
-)
+from models.data_types import ConversationHistory, EnvLookupOutput, NextStep, ValidationInput, EnvLookupInput
 from models.tool_definitions import AgentGoal
+from workflows.workflow_helpers import LLM_ACTIVITY_START_TO_CLOSE_TIMEOUT, \
+    LLM_ACTIVITY_SCHEDULE_TO_CLOSE_TIMEOUT
 from workflows import workflow_helpers as helpers
-from workflows.workflow_helpers import (
-    LLM_ACTIVITY_SCHEDULE_TO_CLOSE_TIMEOUT,
-    LLM_ACTIVITY_START_TO_CLOSE_TIMEOUT,
-)

 with workflow.unsafe.imports_passed_through():
    from activities.tool_activities import ToolActivities
-    from models.data_types import CombinedInput, ToolPromptInput
-    from prompts.agent_prompt_generators import generate_genai_prompt
+    from prompts.agent_prompt_generators import (
+        generate_genai_prompt
+    )
+    from models.data_types import (
+        CombinedInput,
+        ToolPromptInput,
+    )
    from tools.goal_registry import goal_list

 # Constants
 MAX_TURNS_BEFORE_CONTINUE = 250

-
-# ToolData as part of the workflow is what's accessible to the UI - see LLMResponse.jsx for example
+#ToolData as part of the workflow is what's accessible to the UI - see LLMResponse.jsx for example
 class ToolData(TypedDict, total=False):
    next: NextStep
    tool: str
@@ -37,7 +33,6 @@ class ToolData(TypedDict, total=False):
    response: str
    force_confirm: bool = True

-
@workflow.defn
 class AgentGoalWorkflow:
    """Workflow that manages tool execution with user confirmation and conversation history."""
@@ -48,21 +43,16 @@ class AgentGoalWorkflow:
        self.conversation_summary: Optional[str] = None
        self.chat_ended: bool = False
        self.tool_data: Optional[ToolData] = None
-        self.confirmed: bool = (
-            False  # indicates that we have confirmation to proceed to run tool
-        )
+        self.confirmed: bool = False # indicates that we have confirmation to proceed to run tool
        self.tool_results: List[Dict[str, Any]] = []
        self.goal: AgentGoal = {"tools": []}
-        self.show_tool_args_confirmation: bool = (
-            True  # set from env file in activity lookup_wf_env_settings
-        )
-        self.multi_goal_mode: bool = (
-            False  # set from env file in activity lookup_wf_env_settings
-        )
+        self.show_tool_args_confirmation: bool = True # set from env file in activity lookup_wf_env_settings
+        self.multi_goal_mode: bool = False # set from env file in activity lookup_wf_env_settings

    # see ../api/main.py#temporal_client.start_workflow() for how the input parameters are set
    @workflow.run
    async def run(self, combined_input: CombinedInput) -> str:
+
        """Main workflow execution method."""
        # setup phase, starts with blank tool_params and agent_goal prompt as defined in tools/goal_registry.py
        params = combined_input.tool_params
@@ -78,12 +68,12 @@ class AgentGoalWorkflow:
        if params and params.prompt_queue:
            self.prompt_queue.extend(params.prompt_queue)

-        waiting_for_confirm = False
+        waiting_for_confirm = False 
        current_tool = None

        # This is the main interactive loop. Main responsibilities:
        #   - Selecting and changing goals as directed by the user
-        #   - reacting to user input (from signals)
+        #   - reacting to user input (from signals) 
        #   - validating user input to make sure it makes sense with the current goal and tools
        #   - calling the LLM through activities to determine next steps and prompts
        #   - executing the selected tools via activities
@@ -97,7 +87,7 @@ class AgentGoalWorkflow:
            if self.chat_should_end():
                return f"{self.conversation_history}"

-            # Execute the tool
+            # Execute the tool 
            if self.ready_for_tool_execution(waiting_for_confirm, current_tool):
                waiting_for_confirm = await self.execute_tool(current_tool)
                continue
@@ -106,12 +96,10 @@ class AgentGoalWorkflow:
            if self.prompt_queue:
                # get most recent prompt
                prompt = self.prompt_queue.popleft()
-                workflow.logger.info(
-                    f"workflow step: processing message on the prompt queue, message is {prompt}"
-                )
-
+                workflow.logger.info(f"workflow step: processing message on the prompt queue, message is {prompt}")
+                
                # Validate user-provided prompts
-                if self.is_user_prompt(prompt):
+                if self.is_user_prompt(prompt): 
                    self.add_message("user", prompt)

                    # Validate the prompt before proceeding
@@ -132,25 +120,18 @@ class AgentGoalWorkflow:

                    # If validation fails, provide that feedback to the user - i.e., "your words make no sense, puny human" end this iteration of processing
                    if not validation_result.validationResult:
-                        workflow.logger.warning(
-                            f"Prompt validation failed: {validation_result.validationFailedReason}"
-                        )
-                        self.add_message(
-                            "agent", validation_result.validationFailedReason
-                        )
+                        workflow.logger.warning(f"Prompt validation failed: {validation_result.validationFailedReason}")
+                        self.add_message("agent", validation_result.validationFailedReason)
                        continue

                # If valid, proceed with generating the context and prompt
                context_instructions = generate_genai_prompt(
-                    agent_goal=self.goal,
-                    conversation_history=self.conversation_history,
-                    multi_goal_mode=self.multi_goal_mode,
-                    raw_json=self.tool_data,
-                )
-
-                prompt_input = ToolPromptInput(
-                    prompt=prompt, context_instructions=context_instructions
-                )
+                    agent_goal=self.goal, 
+                    conversation_history = self.conversation_history, 
+                    multi_goal_mode=self.multi_goal_mode, 
+                    raw_json=self.tool_data)
+                
+                prompt_input = ToolPromptInput(prompt=prompt, context_instructions=context_instructions)

                # connect to LLM and execute to get next steps
                tool_data = await workflow.execute_activity_method(
@@ -170,24 +151,20 @@ class AgentGoalWorkflow:
                next_step = tool_data.get("next")
                current_tool = tool_data.get("tool")

-                workflow.logger.info(
-                    f"next_step: {next_step}, current tool is {current_tool}"
-                )
+                workflow.logger.info(f"next_step: {next_step}, current tool is {current_tool}")

                # make sure we're ready to run the tool & have everything we need
                if next_step == "confirm" and current_tool:
                    args = tool_data.get("args", {})
-                    # if we're missing arguments, ask for them
-                    if await helpers.handle_missing_args(
-                        current_tool, args, tool_data, self.prompt_queue
-                    ):
+                    # if we're missing arguments, ask for them 
+                    if await helpers.handle_missing_args(current_tool, args, tool_data, self.prompt_queue):
                        continue

                    waiting_for_confirm = True

-                    # We have needed arguments, if we want to force the user to confirm, set that up
+                    # We have needed arguments, if we want to force the user to confirm, set that up                    
                    if self.show_tool_args_confirmation:
-                        self.confirmed = False  # set that we're not confirmed
+                        self.confirmed = False # set that we're not confirmed 
                        workflow.logger.info("Waiting for user confirm signal...")
                    # if we have all needed arguments (handled above) and not holding for a debugging confirm, proceed:
                    else:
@@ -197,11 +174,14 @@ class AgentGoalWorkflow:
                    workflow.logger.info("All steps completed. Resetting goal.")
                    self.change_goal("goal_choose_agent_type")

+
+                
                # else if the next step is to be done with the conversation such as if the user requests it via asking to "end conversation"
                elif next_step == "done":
+                    
                    self.add_message("agent", tool_data)

-                    # here we could send conversation to AI for analysis
+                    #here we could send conversation to AI for analysis

                    # end the workflow
                    return str(self.conversation_history)
@@ -212,10 +192,10 @@ class AgentGoalWorkflow:
                    self.prompt_queue,
                    self.goal,
                    MAX_TURNS_BEFORE_CONTINUE,
-                    self.add_message,
+                    self.add_message
                )

-    # Signal that comes from api/main.py via a post to /send-prompt
+    #Signal that comes from api/main.py via a post to /send-prompt
    @workflow.signal
    async def user_prompt(self, prompt: str) -> None:
        """Signal handler for receiving user prompts."""
@@ -225,28 +205,28 @@ class AgentGoalWorkflow:
            return
        self.prompt_queue.append(prompt)

-    # Signal that comes from api/main.py via a post to /confirm
+    #Signal that comes from api/main.py via a post to /confirm
    @workflow.signal
    async def confirm(self) -> None:
        """Signal handler for user confirmation of tool execution."""
        workflow.logger.info("Received user signal: confirmation")
        self.confirmed = True

-    # Signal that comes from api/main.py via a post to /end-chat
+    #Signal that comes from api/main.py via a post to /end-chat
    @workflow.signal
    async def end_chat(self) -> None:
        """Signal handler for ending the chat session."""
        workflow.logger.info("signal received: end_chat")
        self.chat_ended = True

-    # Signal that can be sent from Temporal Workflow UI to enable debugging confirm and override .env setting
+    #Signal that can be sent from Temporal Workflow UI to enable debugging confirm and override .env setting
    @workflow.signal
    async def enable_debugging_confirm(self) -> None:
        """Signal handler for enabling debugging confirm UI & associated logic."""
        workflow.logger.info("signal received: enable_debugging_confirm")
        self.enable_debugging_confirm = True

-    # Signal that can be sent from Temporal Workflow UI to disable debugging confirm and override .env setting
+    #Signal that can be sent from Temporal Workflow UI to disable debugging confirm and override .env setting
    @workflow.signal
    async def disable_debugging_confirm(self) -> None:
        """Signal handler for disabling debugging confirm UI & associated logic."""
@@ -257,7 +237,7 @@ class AgentGoalWorkflow:
    def get_conversation_history(self) -> ConversationHistory:
        """Query handler to retrieve the full conversation history."""
        return self.conversation_history
-
+    
    @workflow.query
    def get_agent_goal(self) -> AgentGoal:
        """Query handler to retrieve the current goal of the agent."""
@@ -265,7 +245,7 @@ class AgentGoalWorkflow:

    @workflow.query
    def get_summary_from_history(self) -> Optional[str]:
-        """Query handler to retrieve the conversation summary if available.
+        """Query handler to retrieve the conversation summary if available. 
        Used only for continue as new of the workflow."""
        return self.conversation_summary

@@ -292,9 +272,9 @@ class AgentGoalWorkflow:
        )

    def change_goal(self, goal: str) -> None:
-        """Change the goal (usually on request of the user).
-
-        Args:
+        """ Change the goal (usually on request of the user).
+        
+        Args: 
            goal: goal to change to)
        """
        if goal is not None:
@@ -303,9 +283,8 @@ class AgentGoalWorkflow:
                    self.goal = listed_goal
                    workflow.logger.info("Changed goal to " + goal)
            if goal is None:
-                workflow.logger.warning(
-                    "Goal not set after goal reset, probably bad."
-                )  # if this happens, there's probably a problem with the goal list
+                workflow.logger.warning("Goal not set after goal reset, probably bad.") # if this happens, there's probably a problem with the goal list
+        

    # workflow function that defines if chat should end
    def chat_should_end(self) -> bool:
@@ -314,11 +293,9 @@ class AgentGoalWorkflow:
            return True
        else:
            return False
-
+    
    # define if we're ready for tool execution
-    def ready_for_tool_execution(
-        self, waiting_for_confirm: bool, current_tool: Any
-    ) -> bool:
+    def ready_for_tool_execution(self, waiting_for_confirm: bool, current_tool: Any) -> bool:
        if self.confirmed and waiting_for_confirm and current_tool and self.tool_data:
            return True
        else:
@@ -327,19 +304,19 @@ class AgentGoalWorkflow:
    # LLM-tagged prompts start with "###"
    # all others are from the user
    def is_user_prompt(self, prompt) -> bool:
-        if prompt.startswith("###"):
-            return False
-        else:
-            return True
-
+         if prompt.startswith("###"):
+             return False
+         else:
+             return True
+         
    # look up env settings in an activity so they're part of history
-    async def lookup_wf_env_settings(self, combined_input: CombinedInput) -> None:
+    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,
+            show_confirm_env_var_name = "SHOW_CONFIRM", 
+            show_confirm_default = True,
        )
-        env_output: EnvLookupOutput = await workflow.execute_activity_method(
-            ToolActivities.get_wf_env_vars,
+        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,
            retry_policy=RetryPolicy(
@@ -348,13 +325,11 @@ class AgentGoalWorkflow:
        )
        self.show_tool_args_confirmation = env_output.show_confirm
        self.multi_goal_mode = env_output.multi_goal_mode
-
+    
    # execute the tool - return False if we're not waiting for confirm anymore (always the case if it works successfully)
-    #
-    async def execute_tool(self, current_tool: str) -> bool:
-        workflow.logger.info(
-            f"workflow step: user has confirmed, executing the tool {current_tool}"
-        )
+    # 
+    async def execute_tool(self, current_tool: str)->bool:
+        workflow.logger.info(f"workflow step: user has confirmed, executing the tool {current_tool}")
        self.confirmed = False
        waiting_for_confirm = False
        confirmed_tool_data = self.tool_data.copy()
@@ -367,27 +342,21 @@ class AgentGoalWorkflow:
            self.tool_data,
            self.tool_results,
            self.add_message,
-            self.prompt_queue,
+            self.prompt_queue
        )

        # set new goal if we should
        if len(self.tool_results) > 0:
-            if (
-                "ChangeGoal" in self.tool_results[-1].values()
-                and "new_goal" in self.tool_results[-1].keys()
-            ):
+            if "ChangeGoal" in self.tool_results[-1].values() and "new_goal" in self.tool_results[-1].keys():
                new_goal = self.tool_results[-1].get("new_goal")
                self.change_goal(new_goal)
-            elif (
-                "ListAgents" in self.tool_results[-1].values()
-                and self.goal.id != "goal_choose_agent_type"
-            ):
+            elif "ListAgents" in self.tool_results[-1].values() and self.goal.id != "goal_choose_agent_type":
                self.change_goal("goal_choose_agent_type")
        return waiting_for_confirm
-
+        
    # debugging helper - drop this in various places in the workflow to get status
    # 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:
+    def print_useful_workflow_vars(self, status_or_step:str) -> None:
        print(f"***{status_or_step}:***")
        if self.goal:
            print(f"current goal: {self.goal.id}")
@@ -398,3 +367,4 @@ class AgentGoalWorkflow:
        else:
            print("no tool data initialized yet")
        print(f"self.confirmed: {self.confirmed}")
+    
--- a/workflows/workflow_helpers.py
+++ b/workflows/workflow_helpers.py
@@ -1,9 +1,8 @@
 from datetime import timedelta
-from typing import Any, Deque, Dict
-
+from typing import Dict, Any, Deque
 from temporalio import workflow
-from temporalio.common import RetryPolicy
 from temporalio.exceptions import ActivityError
+from temporalio.common import RetryPolicy

 from models.data_types import ConversationHistory, ToolPromptInput
 from prompts.agent_prompt_generators import (
Author	SHA1	Message	Date
Steve Androulakis	8baa398a53	setup update	2025-05-26 14:33:32 -07:00
Steve Androulakis	617c258d4b	clean up setup.md	2025-05-26 14:28:45 -07:00
Jonathan Muszkat	e9bf3a4c16	bump dependencies	2025-05-26 14:20:28 -07:00
Jonathan Muszkat	ea62bb6d83	chore: make start-dev feedback from: https://github.com/temporal-community/temporal-ai-agent/issues/31	2025-05-26 14:20:28 -07:00
Jonathan Muszkat	e67b8a56ef	update	2025-05-26 14:20:28 -07:00
Jonathan Muszkat	dcb6271c23	feat: LiteLLM integration	2025-05-26 14:20:28 -07:00