moved to decorators

AgentOps-AI · Jul 31, 2024 · 525d946 · 525d946
1 parent 07056da
commit 525d946
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/agentops/decorators.py b/agentops/decorators.py
@@ -156,6 +156,11 @@ def decorator(obj):
 
             def new_init(self, *args, **kwargs):
                 try:
+                    kwarg_name = kwargs.get("agentops_name", None)
+                    if kwarg_name is not None:
+                        self.agent_ops_agent_name = kwarg_name
+                        del kwargs["agentops_name"]
+
                     original_init(self, *args, **kwargs)
 
                     if not Client().is_initialized: