Move where goal is set, make dummy data default for create_invoice

api/main.py

@@ -23,12 +23,21 @@ load_dotenv()
 
 def get_agent_goal():
     """Get the agent goal from environment variables."""
-    goal_name = os.getenv("AGENT_GOAL", "goal_match_train_invoice")
     goals = {
         "goal_match_train_invoice": goal_match_train_invoice,
         "goal_event_flight_invoice": goal_event_flight_invoice,
     }
-    return goals.get(goal_name, goal_event_flight_invoice)
+# Agent Goal Configuration
+#AGENT_GOAL=goal_event_flight_invoice  
+#AGENT_GOAL=goal_match_train_invoice
+
+    #goal_name = os.getenv("AGENT_GOAL")
+    goal_name = "goal_event_flight_invoice"
+    if goal_name is not None:
+        return goals.get(goal_name)
+    else:
+        #if no goal is set in the env file, default to event/flight use case
+        return goals.get("goal_event_flight_invoice", goal_event_flight_invoice)
 
 
 @app.on_event("startup")

tools/create_invoice.py

@@ -4,7 +4,7 @@ from dotenv import load_dotenv
 
 load_dotenv(override=True)  # Load environment variables from a .env file
 
-stripe.api_key = os.getenv("STRIPE_API_KEY", "YOUR_DEFAULT_KEY")
+stripe.api_key = os.getenv("STRIPE_API_KEY")
 
 
 def ensure_customer_exists(
@@ -26,41 +26,49 @@ def ensure_customer_exists(
 
 def create_invoice(args: dict) -> dict:
     """Create and finalize a Stripe invoice."""
-    # Find or create customer
+    # If an API key exists in the env file, find or create customer
-    customer_id = ensure_customer_exists(
+    if stripe.api_key is not None:
-        args.get("customer_id"), args.get("email", "default@example.com")
+        customer_id = ensure_customer_exists(
-    )
+            args.get("customer_id"), args.get("email", "default@example.com")
+        )
 
-    # Get amount and convert to cents
+        # Get amount and convert to cents
-    amount = args.get("amount", 200.00)  # Default to $200.00
+        amount = args.get("amount", 200.00)  # Default to $200.00
-    try:
+        try:
-        amount_cents = int(float(amount) * 100)
+            amount_cents = int(float(amount) * 100)
-    except (TypeError, ValueError):
+        except (TypeError, ValueError):
-        return {"error": "Invalid amount provided. Please confirm the amount."}
+            return {"error": "Invalid amount provided. Please confirm the amount."}
 
-    # Create an invoice item
+        # Create an invoice item
-    stripe.InvoiceItem.create(
+        stripe.InvoiceItem.create(
-        customer=customer_id,
+            customer=customer_id,
-        amount=amount_cents,
+            amount=amount_cents,
-        currency="gbp",
+            currency="gbp",
-        description=args.get("tripDetails", "Service Invoice"),
+            description=args.get("tripDetails", "Service Invoice"),
-    )
+        )
 
-    # Create and finalize the invoice
+        # Create and finalize the invoice
-    invoice = stripe.Invoice.create(
+        invoice = stripe.Invoice.create(
-        customer=customer_id,
+            customer=customer_id,
-        collection_method="send_invoice",  # Invoice is sent to the customer
+            collection_method="send_invoice",  # Invoice is sent to the customer
-        days_until_due=args.get("days_until_due", 7),  # Default due date: 7 days
+            days_until_due=args.get("days_until_due", 7),  # Default due date: 7 days
-        pending_invoice_items_behavior="include",  # No pending invoice items
+            pending_invoice_items_behavior="include",  # No pending invoice items
-    )
+        )
-    finalized_invoice = stripe.Invoice.finalize_invoice(invoice.id)
+        finalized_invoice = stripe.Invoice.finalize_invoice(invoice.id)
-
-    return {
-        "invoiceStatus": finalized_invoice.status,
-        "invoiceURL": finalized_invoice.hosted_invoice_url,
-        "reference": finalized_invoice.number,
-    }
 
+        return {
+            "invoiceStatus": finalized_invoice.status,
+            "invoiceURL": finalized_invoice.hosted_invoice_url,
+            "reference": finalized_invoice.number,
+        }
+    # if no API key is in the env file, return dummy info
+    else:
+        print("[CreateInvoice] Creating invoice with:", args)
+        return {
+            "invoiceStatus": "generated",
+            "invoiceURL": "https://pay.example.com/invoice/12345",
+            "reference": "INV-12345",
+        }
 
 def create_invoice_example(args: dict) -> dict:
     """

tools/goal_registry.py

@@ -51,7 +51,6 @@ goal_match_train_invoice = AgentGoal(
     ),
 )
 
-# unused
 goal_event_flight_invoice = AgentGoal(
     tools=[
         find_events_tool,