From 5fab7fc7619ebb25a851813bb3e593f846305feb Mon Sep 17 00:00:00 2001
From: chenweize1998 <chenweize1998@gmail.com>
Date: Mon, 23 Oct 2023 21:47:51 +0800
Subject: [PATCH] fix: import error in alice home task #76

---
 agentverse/agents/__init__.py                        | 1 +
 agentverse/environments/__init__.py                  | 1 +
 agentverse/environments/simulation_env/reflection.py | 9 +++++----
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/agentverse/agents/__init__.py b/agentverse/agents/__init__.py
index 3c74687dc..4a08e6799 100644
--- a/agentverse/agents/__init__.py
+++ b/agentverse/agents/__init__.py
@@ -7,6 +7,7 @@
 from .base import BaseAgent
 from agentverse.agents.simulation_agent.conversation import ConversationAgent
 from agentverse.agents.simulation_agent.tool import ToolAgent
+from agentverse.agents.simulation_agent.reflection import ReflectionAgent
 from agentverse.agents.simulation_agent.prisoner_dilemma import (
     PoliceAgent,
     PrisonerAgent,
diff --git a/agentverse/environments/__init__.py b/agentverse/environments/__init__.py
index 099ceb1bd..d783a6686 100644
--- a/agentverse/environments/__init__.py
+++ b/agentverse/environments/__init__.py
@@ -9,6 +9,7 @@
 
 # from .basic import PipelineEnvironment
 from .simulation_env.basic import BasicEnvironment
+from .simulation_env.reflection import ReflectionEnvironment
 from .simulation_env.pokemon import PokemonEnvironment
 from .simulation_env.prisoner_dilemma import PrisonerDilemmaEnvironment
 from .simulation_env.sde_team import SdeTeamEnvironment
diff --git a/agentverse/environments/simulation_env/reflection.py b/agentverse/environments/simulation_env/reflection.py
index c7c316398..faf19d79c 100644
--- a/agentverse/environments/simulation_env/reflection.py
+++ b/agentverse/environments/simulation_env/reflection.py
@@ -1,5 +1,4 @@
 import asyncio
-import logging
 from typing import Any, Dict, List
 
 from datetime import datetime as dt
@@ -7,13 +6,15 @@
 
 from pydantic import Field
 
+from agentverse.logging import logger
+from agentverse.environments import env_registry as EnvironmentRegistry
 from agentverse.agents.simulation_agent.conversation import BaseAgent
 
 # from agentverse.environments.simulation_env.rules.base import Rule
 from agentverse.environments.simulation_env.rules.base import SimulationRule as Rule
 from agentverse.message import Message
 
-from . import env_registry as EnvironmentRegistry
+# from . import env_registry as EnvironmentRegistry
 from ..base import BaseEnvironment
 
 from pydantic import validator
@@ -70,7 +71,7 @@ def __init__(self, rule, **kwargs):
     async def step(self) -> List[Message]:
         """Run one step of the environment"""
 
-        logging.log(logging.INFO, f"Tick tock. Current time: {self.current_time}")
+        logger.info(f"Tick tock. Current time: {self.current_time}")
 
         # Get the next agent index
         agent_ids = self.rule.get_next_agent_idx(self)
@@ -107,7 +108,7 @@ async def step(self) -> List[Message]:
     def print_messages(self, messages: List[Message]) -> None:
         for message in messages:
             if message is not None:
-                logging.info(f"{message.sender}: {message.content}")
+                logger.info(f"{message.sender}: {message.content}")
 
     def reset(self) -> None:
         """Reset the environment"""