fix: import error in alice home task #76

OpenBMB · Oct 23, 2023 · 5fab7fc · 5fab7fc
1 parent dc0921b
commit 5fab7fc
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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
@@ -1,19 +1,20 @@
 import asyncio
-import logging
 from typing import Any, Dict, List
 
 from datetime import datetime as dt
 import datetime
 
 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"""