code cleanup with black

pull/772/head
Kye Gomez 2 months ago
parent 45357db33a
commit fb5e1d8deb

@ -20,7 +20,7 @@ equity_analyst_1 = Agent(
max_tokens=4000, max_tokens=4000,
saved_state_path="equity_analyst_1.json", saved_state_path="equity_analyst_1.json",
interactive=False, interactive=False,
roles="analyst" roles="analyst",
) )
equity_analyst_2 = Agent( equity_analyst_2 = Agent(
@ -39,11 +39,11 @@ equity_analyst_2 = Agent(
max_tokens=4000, max_tokens=4000,
saved_state_path="equity_analyst_2.json", saved_state_path="equity_analyst_2.json",
interactive=False, interactive=False,
roles="analyst" roles="analyst",
) )
# Run analysis with both analysts # Run analysis with both analysts
equity_analyst_1.talk_to( equity_analyst_1.talk_to(
equity_analyst_2, equity_analyst_2,
"Analyze high growth tech stocks focusing on fundamentals like revenue growth, margins, and market position. Create a detailed analysis table in markdown." "Analyze high growth tech stocks focusing on fundamentals like revenue growth, margins, and market position. Create a detailed analysis table in markdown.",
) )

@ -249,7 +249,9 @@ class GroupChat:
ValueError: If any required components are missing or invalid ValueError: If any required components are missing or invalid
""" """
if len(self.agents) < 2: if len(self.agents) < 2:
raise ValueError("At least two agents are required for a group chat") raise ValueError(
"At least two agents are required for a group chat"
)
if self.speaker_fn is None: if self.speaker_fn is None:
raise ValueError("No speaker function provided") raise ValueError("No speaker function provided")
if self.max_loops <= 0: if self.max_loops <= 0:

Loading…
Cancel
Save