aidmock

var CommonResponses = map[string]string{

	"simple_answer": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "This is a simple answer."}, "human_readable_thought": "Providing straightforward response", "cumulative_summary": "Simple answer provided"}`,

	"complex_answer": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "This is a detailed answer with multiple considerations and nuances."}, "human_readable_thought": "Analyzing multiple factors before answering", "cumulative_summary": "Complex analysis completed and answer provided"}`,

	"tool_sleep": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "sleep"}, "human_readable_thought": "Need to use sleep functionality", "cumulative_summary": "Requesting sleep tool"}`,

	"tool_search": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "search"}, "human_readable_thought": "Need to search for information", "cumulative_summary": "Initiating search"}`,

	"tool_file_operation": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "file_operation"}, "human_readable_thought": "Need to perform file operations", "cumulative_summary": "File operation requested"}`,

	"params_sleep": `{"@action": "call-tool", "params": {"seconds": 0.1}}`,

	"params_search": `{"@action": "call-tool", "params": {"query": "test query", "max_results": 10}}`,

	"satisfaction_yes": `{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Task completed successfully"}`,

	"satisfaction_no": `{"@action": "verify-satisfaction", "user_satisfied": false, "reasoning": "Task needs improvement"}`,

	"plan_exec_simple": `{"@action": "object", "next_action": {"type": "request_plan_and_execution", "plan_request_payload": "Execute simple task"}, "human_readable_thought": "Creating execution plan", "cumulative_summary": "Plan created"}`,

	"clarify_single": `{"@action": "object", "next_action": {"type": "ask_for_clarification", "ask_for_clarification_payload": {"question": "Which option do you prefer?", "options": ["option1", "option2", "option3"]}}, "human_readable_thought": "Need user input", "cumulative_summary": "Requesting clarification"}`,

	"clarify_open": `{"@action": "object", "next_action": {"type": "ask_for_clarification", "ask_for_clarification_payload": {"question": "Please provide more details about your requirement.", "options": []}}, "human_readable_thought": "Need more information", "cumulative_summary": "Requesting details"}`,

	"error_response": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "An error occurred during processing."}, "human_readable_thought": "Error detected", "cumulative_summary": "Error handled"}`,

	"memory_based_answer": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Based on previous context, here is the answer."}, "human_readable_thought": "Retrieving from memory", "cumulative_summary": "Memory-based response provided"}`,

	"tool_create_risk": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "create_risk"}, "human_readable_thought": "Creating security risk entry", "cumulative_summary": "Risk creation initiated"}`,

	"params_risk": `{"@action": "call-tool", "params": {"target": "http://example.com", "title": "Test Risk", "severity": "medium"}}`,

	"tool_write_code": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "write_yaklang_code"}, "human_readable_thought": "Writing Yaklang code", "cumulative_summary": "Code generation started"}`,

	"params_code": `{"@action": "call-tool", "params": {"code": "println(\"Hello World\")", "description": "Simple hello world"}}`,

	"tool_query_doc": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "query_document"}, "human_readable_thought": "Searching documentation", "cumulative_summary": "Document query initiated"}`,

	"tool_blueprint": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "blueprint"}, "human_readable_thought": "Selecting blueprint", "cumulative_summary": "Blueprint selection"}`,

	"multi_tool_start": `{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "first_tool"}, "human_readable_thought": "Starting tool sequence", "cumulative_summary": "Multi-tool execution initiated"}`,

	"task_cancel": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task cancelled as requested."}, "human_readable_thought": "Cancelling task", "cumulative_summary": "Task cancellation processed"}`,

	"task_complete": `{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task completed successfully."}, "human_readable_thought": "Task finished", "cumulative_summary": "Task completion confirmed"}`,
}

var HelloWorldScenario = NewKeywordScenarios()

var TestScenarios = map[string]*TestScenarioDefinition{

	"simple_qa": &TestScenarioDefinition{
		Name:        "simple_qa",
		Description: "Simple question and answer flow",
		Steps: []string{
			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Here is the answer to your question."}, "human_readable_thought": "Understanding the question", "cumulative_summary": "Question answered"}`,
		},
	},

	"single_tool_call": &TestScenarioDefinition{
		Name:        "single_tool_call",
		Description: "Single tool call workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "sleep"}, "human_readable_thought": "Need to use sleep tool", "cumulative_summary": "Requesting tool"}`,

			`{"@action": "call-tool", "params": {"seconds": 0.1}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Tool executed successfully"}`,
		},
	},

	"multi_tool_call": &TestScenarioDefinition{
		Name:        "multi_tool_call",
		Description: "Multiple tool call workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "search"}, "human_readable_thought": "Searching first", "cumulative_summary": "Search initiated"}`,
			`{"@action": "call-tool", "params": {"query": "test"}}`,

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "analyze"}, "human_readable_thought": "Analyzing results", "cumulative_summary": "Analysis initiated"}`,
			`{"@action": "call-tool", "params": {"data": "search_results"}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "All tools executed"}`,
		},
	},

	"plan_and_execute": &TestScenarioDefinition{
		Name:        "plan_and_execute",
		Description: "Plan and execute workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "request_plan_and_execution", "plan_request_payload": "Create and execute plan"}, "human_readable_thought": "Planning task", "cumulative_summary": "Plan requested"}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Plan executed successfully"}`,
		},
	},

	"clarification_flow": &TestScenarioDefinition{
		Name:        "clarification_flow",
		Description: "Clarification request workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "ask_for_clarification", "ask_for_clarification_payload": {"question": "Which option?", "options": ["A", "B", "C"]}}, "human_readable_thought": "Need clarification", "cumulative_summary": "Awaiting user input"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Based on your choice, here is the result."}, "human_readable_thought": "Processing user choice", "cumulative_summary": "Choice processed"}`,
		},
	},

	"error_and_retry": &TestScenarioDefinition{
		Name:        "error_and_retry",
		Description: "Error handling and retry workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "risky_tool"}, "human_readable_thought": "Trying tool", "cumulative_summary": "First attempt"}`,
			`{"@action": "call-tool", "params": {"retry": false}}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Error occurred, will retry."}, "human_readable_thought": "Handling error", "cumulative_summary": "Error detected"}`,

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "risky_tool"}, "human_readable_thought": "Retrying", "cumulative_summary": "Retry attempt"}`,
			`{"@action": "call-tool", "params": {"retry": true}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Retry successful"}`,
		},
	},

	"memory_conversation": &TestScenarioDefinition{
		Name:        "memory_conversation",
		Description: "Conversation with memory context",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "First response"}, "human_readable_thought": "Initial answer", "cumulative_summary": "First interaction recorded"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Second response based on context"}, "human_readable_thought": "Using previous context", "cumulative_summary": "Context-aware response"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Third response with full context"}, "human_readable_thought": "Full context utilization", "cumulative_summary": "Complete conversation context maintained"}`,
		},
	},

	"code_generation": &TestScenarioDefinition{
		Name:        "code_generation",
		Description: "Code generation and validation workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "write_yaklang_code"}, "human_readable_thought": "Writing code", "cumulative_summary": "Code generation started"}`,

			`{"@action": "call-tool", "params": {"code": "println(\"test\")", "language": "yaklang"}}`,

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "check_syntax"}, "human_readable_thought": "Validating syntax", "cumulative_summary": "Syntax check initiated"}`,
			`{"@action": "call-tool", "params": {"code": "println(\"test\")"}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Code generated and validated"}`,
		},
	},

	"risk_creation": &TestScenarioDefinition{
		Name:        "risk_creation",
		Description: "Security risk creation workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "create_risk"}, "human_readable_thought": "Creating risk entry", "cumulative_summary": "Risk creation initiated"}`,

			`{"@action": "call-tool", "params": {"target": "http://test.com", "title": "SQL Injection", "severity": "high", "type": "sqli"}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Risk created successfully"}`,
		},
	},

	"knowledge_query": &TestScenarioDefinition{
		Name:        "knowledge_query",
		Description: "Knowledge base query and answer workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "query_knowledge_base"}, "human_readable_thought": "Searching knowledge", "cumulative_summary": "Knowledge query initiated"}`,
			`{"@action": "call-tool", "params": {"query": "How to use yaklang?"}}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Based on knowledge base: Yaklang is used for..."}, "human_readable_thought": "Providing knowledge-based answer", "cumulative_summary": "Answer from knowledge base"}`,
		},
	},

	"blueprint_execution": &TestScenarioDefinition{
		Name:        "blueprint_execution",
		Description: "Blueprint selection and execution workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "blueprint"}, "human_readable_thought": "Selecting blueprint", "cumulative_summary": "Blueprint selection"}`,
			`{"@action": "call-tool", "params": {"blueprint_name": "web_scanner"}}`,

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "modify_blueprint_params"}, "human_readable_thought": "Adjusting parameters", "cumulative_summary": "Parameters modified"}`,
			`{"@action": "call-tool", "params": {"target": "http://example.com", "depth": 3}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "Blueprint executed"}`,
		},
	},

	"task_queue_management": &TestScenarioDefinition{
		Name:        "task_queue_management",
		Description: "Task queue management workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task enqueued"}, "human_readable_thought": "Adding to queue", "cumulative_summary": "Task queued"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Priority adjusted"}, "human_readable_thought": "Jumping queue", "cumulative_summary": "Priority changed"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task dequeued and executing"}, "human_readable_thought": "Processing task", "cumulative_summary": "Task executing"}`,
		},
	},

	"mcp_protocol": &TestScenarioDefinition{
		Name:        "mcp_protocol",
		Description: "MCP protocol tool usage workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "require_tool", "tool_require_payload": "mcp_tool"}, "human_readable_thought": "Using MCP tool", "cumulative_summary": "MCP invoked"}`,

			`{"@action": "call-tool", "params": {"protocol": "mcp", "action": "fetch_resource"}}`,

			`{"@action": "verify-satisfaction", "user_satisfied": true, "reasoning": "MCP tool executed successfully"}`,
		},
	},

	"self_reflection": &TestScenarioDefinition{
		Name:        "self_reflection",
		Description: "Self-reflection and optimization workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Initial approach"}, "human_readable_thought": "First attempt", "cumulative_summary": "Initial response"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "After reflection, better approach"}, "human_readable_thought": "Reflecting on approach", "cumulative_summary": "Improved solution"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Optimized final solution"}, "human_readable_thought": "Optimization complete", "cumulative_summary": "Best solution provided"}`,
		},
	},

	"task_cancellation": &TestScenarioDefinition{
		Name:        "task_cancellation",
		Description: "Task cancellation and cleanup workflow",
		Steps: []string{

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task started"}, "human_readable_thought": "Starting task", "cumulative_summary": "Task initiated"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Cancelling task"}, "human_readable_thought": "Processing cancellation", "cumulative_summary": "Cancellation in progress"}`,

			`{"@action": "object", "next_action": {"type": "directly_answer", "answer_payload": "Task cancelled and cleaned up"}, "human_readable_thought": "Cleanup complete", "cumulative_summary": "Task cancelled"}`,
		},
	},
}