andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 23:58:15 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
Brandon Hancock
2024-12-10 16:25:32 -05:00
parent bc3fd789d9
commit d5e06cbda6
2 changed files with 0 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/crewai/agents/crew_agent_executor.py
View File

@@ -87,20 +87,15 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
self.llm.stop = self.stop self.llm.stop = self.stop
def invoke(self, inputs: Dict[str, str]) -> Dict[str, Any]: def invoke(self, inputs: Dict[str, str]) -> Dict[str, Any]:
print("prompt: ", self.prompt)
print("inputs: ", inputs)
if "system" in self.prompt: if "system" in self.prompt:
system_prompt = self._format_prompt(self.prompt.get("system", ""), inputs) system_prompt = self._format_prompt(self.prompt.get("system", ""), inputs)
print("system_prompt: ", system_prompt)
user_prompt = self._format_prompt(self.prompt.get("user", ""), inputs) user_prompt = self._format_prompt(self.prompt.get("user", ""), inputs)
print("user_prompt: ", user_prompt)
self.messages.append(self._format_msg(system_prompt, role="system")) self.messages.append(self._format_msg(system_prompt, role="system"))
self.messages.append(self._format_msg(user_prompt)) self.messages.append(self._format_msg(user_prompt))
else: else:
user_prompt = self._format_prompt(self.prompt.get("prompt", ""), inputs) user_prompt = self._format_prompt(self.prompt.get("prompt", ""), inputs)
self.messages.append(self._format_msg(user_prompt)) self.messages.append(self._format_msg(user_prompt))
print("total messages at invoke: ", len(self.messages))
self._show_start_logs() self._show_start_logs()
self.ask_for_human_input = bool(inputs.get("ask_for_human_input", False)) self.ask_for_human_input = bool(inputs.get("ask_for_human_input", False))
@@ -417,12 +412,7 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
AgentFinish: The final output after incorporating human feedback. AgentFinish: The final output after incorporating human feedback.
""" """
while self.ask_for_human_input: while self.ask_for_human_input:
print("Messages at human feedback:")
for idx, message in enumerate(self.messages, start=1):
print(f"Message {idx}: {message}")
print("Total messages at human feedback: ", len(self.messages))
human_feedback = self._ask_human_input(formatted_answer.output) human_feedback = self._ask_human_input(formatted_answer.output)
print("Human feedback: ", human_feedback)
if self.crew and self.crew._train: if self.crew and self.crew._train:
self._handle_crew_training_output(formatted_answer, human_feedback) self._handle_crew_training_output(formatted_answer, human_feedback)
@@ -473,9 +463,6 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
self.ask_for_human_input = True self.ask_for_human_input = True
# Add human feedback to messages # Add human feedback to messages
self.messages.append(self._format_msg(f"Feedback: {human_feedback}")) self.messages.append(self._format_msg(f"Feedback: {human_feedback}"))
print("Messages after human feedback:")
for idx, message in enumerate(self.messages, start=1):
print(f"Message {idx}: {message}")
# Invoke the loop again with updated messages # Invoke the loop again with updated messages
formatted_answer = self._invoke_loop() formatted_answer = self._invoke_loop()

6
src/crewai/flow/flow.py
View File

@@ -167,8 +167,6 @@ class Flow(Generic[T], metaclass=FlowMeta):
_routers: Dict[str, str] = {} _routers: Dict[str, str] = {}
_router_paths: Dict[str, List[str]] = {} _router_paths: Dict[str, List[str]] = {}
initial_state: Union[Type[T], T, None] = None initial_state: Union[Type[T], T, None] = None
# Define a single event emitter signal
event_emitter = Signal("event_emitter") event_emitter = Signal("event_emitter")
def __class_getitem__(cls: Type["Flow"], item: Type[T]) -> Type["Flow"]: def __class_getitem__(cls: Type["Flow"], item: Type[T]) -> Type["Flow"]:
@@ -264,7 +262,6 @@ class Flow(Generic[T], metaclass=FlowMeta):
Returns: Returns:
The final output from the flow execution. The final output from the flow execution.
""" """
# Emit flow_started event
self.event_emitter.send( self.event_emitter.send(
self, self,
event=FlowStartedEvent( event=FlowStartedEvent(
@@ -306,7 +303,6 @@ class Flow(Generic[T], metaclass=FlowMeta):
# Determine the final output (from the last executed method) # Determine the final output (from the last executed method)
final_output = self._method_outputs[-1] if self._method_outputs else None final_output = self._method_outputs[-1] if self._method_outputs else None
# Emit flow_finished event
self.event_emitter.send( self.event_emitter.send(
self, self,
event=FlowFinishedEvent( event=FlowFinishedEvent(
@@ -380,7 +376,6 @@ class Flow(Generic[T], metaclass=FlowMeta):
try: try:
method = self._methods[listener_name] method = self._methods[listener_name]
# Emit method_execution_started event
self.event_emitter.send( self.event_emitter.send(
self, self,
event=MethodExecutionStartedEvent( event=MethodExecutionStartedEvent(
@@ -405,7 +400,6 @@ class Flow(Generic[T], metaclass=FlowMeta):
# If listener does not expect parameters, call without arguments # If listener does not expect parameters, call without arguments
listener_result = await self._execute_method(listener_name, method) listener_result = await self._execute_method(listener_name, method)
# Emit method_execution_finished event
self.event_emitter.send( self.event_emitter.send(
self, self,
event=MethodExecutionFinishedEvent( event=MethodExecutionFinishedEvent(