From 3e11b07d1d771f928fce6d5a2ec134c447112cfe Mon Sep 17 00:00:00 2001 From: pk673 <123758881+pk673@users.noreply.github.com> Date: Wed, 22 May 2024 13:04:50 -0700 Subject: [PATCH] logger fix (#2659) Co-authored-by: HRUSHIKESH DOKALA <96101829+Hk669@users.noreply.github.com> Co-authored-by: Eric Zhu --- autogen/agentchat/contrib/gpt_assistant_agent.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/autogen/agentchat/contrib/gpt_assistant_agent.py b/autogen/agentchat/contrib/gpt_assistant_agent.py index 417bbcdac..40a28bfbc 100644 --- a/autogen/agentchat/contrib/gpt_assistant_agent.py +++ b/autogen/agentchat/contrib/gpt_assistant_agent.py @@ -11,6 +11,7 @@ from autogen import OpenAIWrapper from autogen.agentchat.agent import Agent from autogen.agentchat.assistant_agent import AssistantAgent, ConversableAgent from autogen.oai.openai_utils import create_gpt_assistant, retrieve_assistants_by_name, update_gpt_assistant +from autogen.runtime_logging import log_new_agent, logging_enabled logger = logging.getLogger(__name__) @@ -65,6 +66,8 @@ class GPTAssistantAgent(ConversableAgent): super().__init__( name=name, system_message=instructions, human_input_mode="NEVER", llm_config=openai_client_cfg, **kwargs ) + if logging_enabled(): + log_new_agent(self, locals()) # GPTAssistantAgent's azure_deployment param may cause NotFoundError (404) in client.beta.assistants.list() # See: https://github.com/microsoft/autogen/pull/1721