Rename compile response into a private operator agents function

This commit is contained in:
Debanjum
2025-05-30 20:31:43 -07:00
parent ded1db642c
commit cecbfe35e2
4 changed files with 7 additions and 7 deletions

View File

@@ -253,7 +253,7 @@ class AnthropicOperatorAgent(OperatorAgent):
) )
return formatted_messages return formatted_messages
def compile_response(self, response_content: list[BetaContentBlock | dict] | str) -> str: def _compile_response(self, response_content: list[BetaContentBlock | dict] | str) -> str:
"""Compile Anthropic response into a single string.""" """Compile Anthropic response into a single string."""
if isinstance(response_content, str): if isinstance(response_content, str):
return response_content return response_content

View File

@@ -79,10 +79,10 @@ class OperatorAgent(ABC):
await self.act(current_state) await self.act(current_state)
if not self.messages: if not self.messages:
return "No actions to summarize." return "No actions to summarize."
return self.compile_response(self.messages[-1].content) return self._compile_response(self.messages[-1].content)
@abstractmethod @abstractmethod
def compile_response(self, response: List | str) -> str: def _compile_response(self, response: List | str) -> str:
pass pass
@abstractmethod @abstractmethod
@@ -102,7 +102,7 @@ class OperatorAgent(ABC):
self.tracer["chat_model"] = self.vision_model.name self.tracer["chat_model"] = self.vision_model.name
if is_promptrace_enabled() and len(self.messages) > 1: if is_promptrace_enabled() and len(self.messages) > 1:
compiled_messages = [ compiled_messages = [
AgentMessage(role=msg.role, content=self.compile_response(msg.content)) for msg in self.messages AgentMessage(role=msg.role, content=self._compile_response(msg.content)) for msg in self.messages
] ]
commit_conversation_trace(compiled_messages[:-1], compiled_messages[-1].content, self.tracer) commit_conversation_trace(compiled_messages[:-1], compiled_messages[-1].content, self.tracer)

View File

@@ -260,7 +260,7 @@ class BinaryOperatorAgent(OperatorAgent):
return summary return summary
def compile_response(self, response_content: str | List) -> str: def _compile_response(self, response_content: str | List) -> str:
"""Compile response content into a string, handling OpenAI message structures.""" """Compile response content into a string, handling OpenAI message structures."""
if isinstance(response_content, str): if isinstance(response_content, str):
return response_content return response_content

View File

@@ -218,8 +218,8 @@ class OpenAIOperatorAgent(OperatorAgent):
) )
return formatted_messages return formatted_messages
def compile_response(self, response_content: str | list[dict | ResponseOutputItem]) -> str: def _compile_response(self, response_content: str | list[dict | ResponseOutputItem]) -> str:
"""Compile the response from model into a single string.""" """Compile the response from model into a single string for prompt tracing."""
# Handle case where response content is a string. # Handle case where response content is a string.
# This is the case when response content is a user query # This is the case when response content is a user query
if isinstance(response_content, str): if isinstance(response_content, str):