Crews as classes

1(env)$ mkdir contract_crew
2(env)$ touch contract_crew/__init__.py
3(env)$ mv config contract_crew/config
4(env)$ rm review_crew.py test_config.py

1# contract_crew/config/tasks.yaml
2
3extract_terms:
4  description: >
5    Read the contract at {contract_path}. List its key terms: parties, dates,
6    term and renewal, fees and payment, service levels, data handling,
7    termination, liability, and any clause that lets one party change the
8    deal on its own.
9  expected_output: >
10    A bulleted list of key terms. Each bullet quotes the contract and ends
11    with its section number.
12  agent: reader
13
14assess_risks:
15  description: >
16    Using the key terms, write a risk memo for {customer}. Pick the five terms
17    most likely to cost them money or catch them out, and say what they
18    should do about each one.
19  expected_output: >
20    A Markdown memo of no more than 350 words: a two-sentence summary, then a
21    numbered list of five risks. Each risk has a section number, why it
22    matters, and a recommended action.
23  agent: risk_analyst
24
25fact_check:
26  description: >
27    Check every claim in the risk memo against the contract at {contract_path}.
28    Fix any number, date, or section reference that does not match the
29    contract, and remove anything the contract does not say.
30  expected_output: >
31    The corrected memo in Markdown, followed by a "Corrections" section that
32    lists each change you made, or says "None".
33  agent: fact_checker
34  output_file: output/review.md

1# contract_crew/crew.py
2
3from crewai import Agent, Crew, Process, Task
4from crewai.project import CrewBase, agent, crew, task
5from crewai_tools import FileReadTool
6
7
8@CrewBase
9class ContractReviewCrew:
10    """Reads a contract, flags the risks, and checks the memo against the source."""
11
12    agents_config = "config/agents.yaml"
13    tasks_config = "config/tasks.yaml"
14
15    def __init__(self):
16        self.read_file = FileReadTool()
17
18    @agent
19    def reader(self) -> Agent:
20        return Agent(config=self.agents_config["reader"], tools=[self.read_file], verbose=True)
21
22    @agent
23    def risk_analyst(self) -> Agent:
24        return Agent(config=self.agents_config["risk_analyst"], verbose=True)
25
26    @agent
27    def fact_checker(self) -> Agent:
28        return Agent(config=self.agents_config["fact_checker"], tools=[self.read_file], verbose=True)
29
30    @task
31    def extract_terms(self) -> Task:
32        return Task(config=self.tasks_config["extract_terms"])
33
34    @task
35    def assess_risks(self) -> Task:
36        return Task(config=self.tasks_config["assess_risks"])
37
38    @task
39    def fact_check(self) -> Task:
40        return Task(config=self.tasks_config["fact_check"])
41
42    @crew
43    def crew(self) -> Crew:
44        return Crew(
45            agents=self.agents,
46            tasks=self.tasks,
47            process=Process.sequential,
48            verbose=True,
49        )

1# main.py
2
3from dotenv import load_dotenv
4
5from contract_crew.crew import ContractReviewCrew
6
7load_dotenv()
8
9
10def main():
11    result = ContractReviewCrew().crew().kickoff(
12        inputs={
13            "contract_path": "contracts/acme-saas-agreement.md",
14            "customer": "Northwind Traders",
15        }
16    )
17    print(result.raw)
18
19
20if __name__ == "__main__":
21    main()

1(env)$ python main.py

1(env)$ mkdir tests
2(env)$ touch tests/__init__.py

1# tests/test_crew.py
2
3from contract_crew.crew import ContractReviewCrew
4
5
6def test_crew_is_wired_up():
7    crew = ContractReviewCrew().crew()
8
9    assert [agent.role for agent in crew.agents] == [
10        "Contract analyst",
11        "Commercial risk analyst",
12        "Contract fact-checker",
13    ]
14    assert [task.agent.role for task in crew.tasks] == [
15        "Contract analyst",
16        "Commercial risk analyst",
17        "Contract fact-checker",
18    ]

1(env)$ python -m pytest
2
3tests/test_crew.py .                                                     [100%]
4
5======================== 1 passed, 11 warnings in 3.37s ========================