fix(agent): remove wrongly initialized agent_id from agent's context

Signed-off-by: Valentin De Matos <vltn.dematos@gmail.com>
Thytu · Dec 24, 2024 · 8fbab7a · 8fbab7a
1 parent 3c9a483
commit 8fbab7a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/agentarium/Agent.py b/agentarium/Agent.py
@@ -102,7 +102,10 @@ def __init__(self, **kwargs):
         if not Agent._allow_init:
             raise RuntimeError("Agent instances should be created using Agent.create_agent()")
 
-        self.agent_id = kwargs.get("agent_id") if "agent_id" in kwargs and kwargs["agent_id"] != DefaultValue.NOT_PROVIDED else faker.uuid4()
+        if "agent_id" in kwargs:
+            self.agent_id = kwargs.pop("agent_id")
+
+        self.agent_id = self.agent_id if self.agent_id!= DefaultValue.NOT_PROVIDED else faker.uuid4()
         self.agent_informations: dict = kwargs or {}
 
         if "gender" not in kwargs or kwargs["gender"] == DefaultValue.NOT_PROVIDED:

diff --git a/agentarium/__init__.py b/agentarium/__init__.py
@@ -3,7 +3,7 @@
 from .AgentInteractionManager import AgentInteractionManager
 from .Interaction import Interaction
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 
 __all__ = [
     "Agent",

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "agentarium"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
     { name = "thytu" },
 ]