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

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

Browse Source
This commit is contained in:
Laine
2025-03-05 17:24:18 -05:00
parent 6accc1f2e6
commit d09db9f11f
3 changed files with 51 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/main.py
View File

@@ -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")