mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
# test_fix_2402.py
|
|
from crewai import Agent, Task, Crew
|
|
|
|
# Case 1: Only system_template provided
|
|
agent1 = Agent(
|
|
role="Test Role",
|
|
goal="Test Goal",
|
|
backstory="Test Backstory",
|
|
system_template="You are a test agent...",
|
|
# prompt_template is intentionally missing
|
|
)
|
|
|
|
# Case 2: system_template and prompt_template provided, but response_template missing
|
|
agent2 = Agent(
|
|
role="Test Role",
|
|
goal="Test Goal",
|
|
backstory="Test Backstory",
|
|
system_template="You are a test agent...",
|
|
prompt_template="This is a test prompt...",
|
|
# response_template is intentionally missing
|
|
)
|
|
|
|
# Create tasks and crews
|
|
task1 = Task(description="Test task 1", agent=agent1, expected_output="Test output 1")
|
|
task2 = Task(description="Test task 2", agent=agent2, expected_output="Test output 2")
|
|
|
|
crew1 = Crew(agents=[agent1], tasks=[task1])
|
|
crew2 = Crew(agents=[agent2], tasks=[task2])
|
|
|
|
print("Testing agent with only system_template...")
|
|
try:
|
|
agent1.execute_task(task1)
|
|
print("Success! No error was raised.")
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
print("\nTesting agent with missing response_template...")
|
|
try:
|
|
agent2.execute_task(task2)
|
|
print("Success! No error was raised.")
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|