fix: imports (oupsy)

Signed-off-by: Valentin De Matos <[email protected]>

Author: Thytu

agentarium/CheckpointManager.py

@@ -83,7 +83,7 @@ def load(self) -> None:
         Load a simulation from a checkpoint.
         """
 
-        from agentarium.agent import Agent
+        from agentarium import Agent
         env_data = pickle.load(open(self.path, "rb"))
 
         self._state = OrderedDict(env_data["state"])

agentarium/__init__.py

@@ -1,9 +1,9 @@
-from .agent import Agent
+from .Agent import Agent
 # from .Environment import Environment
 from .AgentInteractionManager import AgentInteractionManager
 from .Interaction import Interaction
 
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 
 __all__ = [
     "Agent",

examples/1_basic_chat/demo.py

@@ -1,8 +1,8 @@
-from agentarium.Agent import Agent
+from agentarium import Agent
 
 # Create some agents
-alice_agent = Agent(name="Alice", occupation="Software Engineer")
-bob_agent = Agent(name="Bob", occupation="Data Scientist")
+alice_agent = Agent.create_agent(name="Alice", occupation="Software Engineer")
+bob_agent = Agent.create_agent(name="Bob", occupation="Data Scientist")
 
 alice_agent.talk_to(bob_agent, "Hello Bob! I heard you're working on some interesting data science projects.")
 bob_agent.talk_to(alice_agent, "Hi Alice! Yes, I'm currently working on a machine learning model for natural language processing.")

examples/2_checkpointing/demo.py

@@ -1,4 +1,4 @@
-from agentarium.agent import Agent
+from agentarium import Agent
 from agentarium.CheckpointManager import CheckpointManager
 
 

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "agentarium"
-version = "0.2.3"
+version = "0.2.4"
 authors = [
     { name = "thytu" },
 ]