Skip to content

Commit dfeb12f

Browse files
fix: correct import paths in test_crew_agent_parser.py
1 parent 2d5b775 commit dfeb12f

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tests/agents/test_crew_agent_parser.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
import pytest
22

3-
from crewai.agents.crew_agent_executor import (
3+
from crewai.agents import parser
4+
from crewai.agents.parser import (
45
AgentAction,
56
AgentFinish,
6-
OutputParserException,
77
)
8-
from crewai.agents import parser
8+
from crewai.agents.parser import (
9+
OutputParserError as OutputParserException,
10+
)
911

1012

1113
def test_valid_action_parsing_special_characters():
@@ -348,9 +350,9 @@ def test_integration_valid_and_invalid():
348350
for part in parts:
349351
try:
350352
result = parser.parse(part.strip())
351-
results.append(result)
352353
except OutputParserException as e:
353-
results.append(e)
354+
result = e
355+
results.append(result)
354356

355357
assert isinstance(results[0], AgentAction)
356358
assert isinstance(results[1], AgentFinish)

0 commit comments

Comments
 (0)