Spaces:
Sleeping
Sleeping
Commit ·
22b52d8
1
Parent(s): 6806846
refactor: remove phase comment headers and unused imports
Browse files- src/code_tribunal/agents.py +1 -21
src/code_tribunal/agents.py
CHANGED
|
@@ -1,10 +1,6 @@
|
|
| 1 |
-
"""Courtroom agent definitions
|
| 2 |
-
|
| 3 |
-
import os
|
| 4 |
-
from typing import Callable
|
| 5 |
|
| 6 |
from crewai import Agent, LLM
|
| 7 |
-
from pydantic import BaseModel
|
| 8 |
|
| 9 |
from code_tribunal.config import TribunalConfig
|
| 10 |
|
|
@@ -18,10 +14,6 @@ def _get_llm(config: TribunalConfig) -> LLM:
|
|
| 18 |
)
|
| 19 |
|
| 20 |
|
| 21 |
-
# ---------------------------------------------------------------------------
|
| 22 |
-
# Phase 3: Investigators (with domain-specific tools)
|
| 23 |
-
# ---------------------------------------------------------------------------
|
| 24 |
-
|
| 25 |
def security_investigator(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 26 |
return Agent(
|
| 27 |
role="Security Forensic Investigator",
|
|
@@ -86,10 +78,6 @@ def architecture_investigator(config: TribunalConfig, tools: list | None = None)
|
|
| 86 |
)
|
| 87 |
|
| 88 |
|
| 89 |
-
# ---------------------------------------------------------------------------
|
| 90 |
-
# Phase 4: The Trial (with evidence context tools)
|
| 91 |
-
# ---------------------------------------------------------------------------
|
| 92 |
-
|
| 93 |
def prosecutor(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 94 |
return Agent(
|
| 95 |
role="The Prosecutor",
|
|
@@ -137,10 +125,6 @@ def defense_attorney(config: TribunalConfig, tools: list | None = None) -> Agent
|
|
| 137 |
)
|
| 138 |
|
| 139 |
|
| 140 |
-
# ---------------------------------------------------------------------------
|
| 141 |
-
# Phase 5: Verdict + Report
|
| 142 |
-
# ---------------------------------------------------------------------------
|
| 143 |
-
|
| 144 |
def judge(config: TribunalConfig) -> Agent:
|
| 145 |
return Agent(
|
| 146 |
role="The Judge",
|
|
@@ -182,10 +166,6 @@ def verdict_report_agent(config: TribunalConfig) -> Agent:
|
|
| 182 |
)
|
| 183 |
|
| 184 |
|
| 185 |
-
# ---------------------------------------------------------------------------
|
| 186 |
-
# Interactive Q&A
|
| 187 |
-
# ---------------------------------------------------------------------------
|
| 188 |
-
|
| 189 |
def expert_witness(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 190 |
return Agent(
|
| 191 |
role="Court Expert Witness",
|
|
|
|
| 1 |
+
"""Courtroom agent definitions — 8 agents with tools."""
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
from crewai import Agent, LLM
|
|
|
|
| 4 |
|
| 5 |
from code_tribunal.config import TribunalConfig
|
| 6 |
|
|
|
|
| 14 |
)
|
| 15 |
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
def security_investigator(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 18 |
return Agent(
|
| 19 |
role="Security Forensic Investigator",
|
|
|
|
| 78 |
)
|
| 79 |
|
| 80 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
def prosecutor(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 82 |
return Agent(
|
| 83 |
role="The Prosecutor",
|
|
|
|
| 125 |
)
|
| 126 |
|
| 127 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
def judge(config: TribunalConfig) -> Agent:
|
| 129 |
return Agent(
|
| 130 |
role="The Judge",
|
|
|
|
| 166 |
)
|
| 167 |
|
| 168 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 169 |
def expert_witness(config: TribunalConfig, tools: list | None = None) -> Agent:
|
| 170 |
return Agent(
|
| 171 |
role="Court Expert Witness",
|