fix: rm unnecessary change

Author: zhjuzi

oasis/environment/env.py

@@ -55,6 +55,7 @@ def __init__(
         platform: Union[DefaultPlatformType, Platform],
         database_path: str = None,
         semaphore: int = 128,
+        profile_update_enabled: bool = True,
     ) -> None:
         r"""Init the oasis environment.
 
@@ -120,7 +121,7 @@ def __init__(
         # Initialize user profile management
         self.user_profiles: Dict[int, Dict[str, Any]] = {}  # Store user profile data
         self.user_profile_agent: PreferenceAgent = None  # User profile analysis agent
-        self.profile_update_enabled: bool = True  # Whether to enable dynamic profile updates
+        self.profile_update_enabled: bool = profile_update_enabled  # Whether to enable dynamic profile updates
         self.user_profile_file: str = "user_profile.json"  # User profile storage file
 
     async def reset(self) -> None:
@@ -161,14 +162,12 @@ async def _load_user_profiles(self) -> None:
                 if isinstance(data, dict) and "user_profiles" in data:
                     # New format: data with timestamps
                     raw_profiles = data["user_profiles"]
-                    # 确保所有 key 都是字符串类型
                     self.user_profiles = {str(k): v for k, v in raw_profiles.items()}
                     self._profile_update_count = data.get("update_count", 0)
                     env_log.info(f"Loaded {len(self.user_profiles)} user profiles from {self.user_profile_file} (last updated: {data.get('last_updated', 'unknown')}, update #{self._profile_update_count})")
                     env_log.info(f"Profile keys loaded: {list(self.user_profiles.keys())}")
                 else:
                     # Old format: direct user profile data
-                    # 确保所有 key 都是字符串类型
                     self.user_profiles = {str(k): v for k, v in data.items()}
                     self._profile_update_count = 0
                     env_log.info(f"Loaded {len(self.user_profiles)} user profiles from {self.user_profile_file} (legacy format)")

oasis/social_agent/agent.py

@@ -132,12 +132,10 @@ async def perform_action_by_llm(self):
                 f"platform environments. Notice that don't limit your "
                 f"actions for example to just like the posts. "
                 f"Here is your social media environment: {env_prompt}"))
-        #print(f'user_info: {self.user_info}')
-        #print(f'env_prompt: {env_prompt}')
         try:
-            # agent_log.info(
-            #     f"Agent {self.social_agent_id} observing environment: "
-            #     f"{env_prompt}")
+            agent_log.info(
+                f"Agent {self.social_agent_id} observing environment: "
+                f"{env_prompt}")
             response = await self.astep(user_msg)
             for tool_call in response.info['tool_calls']:
                 action_name = tool_call.tool_name

oasis/social_agent/agents_generator.py

@@ -359,7 +359,7 @@ async def generate_controllable_agents(
             }},
             recsys_type="reddit",
         )
-
+        # All controllable agent_ids come before LLM agent_ids
         agent = SocialAgent(agent_id=i,
                             user_info=user_info,
                             channel=channel,
@@ -378,7 +378,7 @@ async def generate_controllable_agents(
     for i in range(control_user_num):
         for j in range(control_user_num):
             agent = agent_graph.get_agent(i)
-
+            # All controllable agents follow each other
             if i != j:
                 user_id = agent_user_id_mapping[j]
                 await agent.env.action.follow(user_id)
@@ -407,7 +407,7 @@ async def gen_control_agents_with_data(
             },
             recsys_type="reddit",
         )
-
+        # All controllable agent_ids come before LLM agent_ids
         agent = SocialAgent(
             agent_id=i,
             user_info=user_info,

oasis/social_platform/database.py

@@ -89,13 +89,6 @@ def create_db(db_path: str | None = None):
     if db_path is None:
         db_path = get_db_path()
 
-    # If the database file already exists, return the connection and cursor directly
-    if osp.exists(db_path):
-        print(f"Database file {db_path} already exists, skipping creation.")
-        conn = sqlite3.connect(db_path)
-        cursor = conn.cursor()
-        return conn, cursor
-
     # Connect to the database:
     print("db_path", db_path)
     conn = sqlite3.connect(db_path)