agentUniverse 0.0.17.dev2__tar.gz → 0.0.17.dev4__tar.gz

{agentuniverse-0.0.17.dev2 → agentuniverse-0.0.17.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentUniverse
-Version: 0.0.17.dev2
+Version: 0.0.17.dev4
 Summary: agentUniverse is a framework for developing applications powered by multi-agent base on large language model.
 Author: AntGroup
 Author-email: jerry.zzw@antgroup.com

{agentuniverse-0.0.17.dev2 → agentuniverse-0.0.17.dev4}/agentuniverse/base/annotation/trace.py

@@ -187,16 +187,16 @@ def trace_agent(func):
         start_info = get_caller_info()
         pair_id = f"agent_{uuid.uuid4().hex}"
         kwargs['memory_source_info'] = start_info
+        # add invocation chain to the monitor module.
+        Monitor.init_invocation_chain()
+        Monitor.add_invocation_chain({'source': source, 'type': 'agent'})
         ConversationMemoryModule().add_agent_input_info(start_info, self, agent_input, pair_id)
         if tracing is False:
             result = await func(*args, **kwargs)
             ConversationMemoryModule().add_agent_result_info(self, result, start_info, pair_id)
+            Monitor.pop_invocation_chain()
             return result
 
-        # add invocation chain to the monitor module.
-        Monitor.init_invocation_chain()
-        Monitor.add_invocation_chain({'source': source, 'type': 'agent'})
-
         start_time = time.time()
         Monitor().trace_agent_input(source=source, agent_input=agent_input)
 
@@ -230,16 +230,16 @@ def trace_agent(func):
         pair_id = f"agent_{uuid.uuid4().hex}"
         start_info = get_caller_info()
         kwargs['memory_source_info'] = start_info
+        # add invocation chain to the monitor module.
+        Monitor.init_invocation_chain()
+        Monitor.add_invocation_chain({'source': source, 'type': 'agent'})
         ConversationMemoryModule().add_agent_input_info(start_info, self, agent_input, pair_id)
         if tracing is False:
             result = func(*args, **kwargs)
             ConversationMemoryModule().add_agent_result_info(self, result, start_info, pair_id)
+            Monitor.pop_invocation_chain()
             return result
 
-        # add invocation chain to the monitor module.
-        Monitor.init_invocation_chain()
-        Monitor.add_invocation_chain({'source': source, 'type': 'agent'})
-
         start_time = time.time()
         Monitor().trace_agent_input(source=source, agent_input=agent_input)
 

{agentuniverse-0.0.17.dev2 → agentuniverse-0.0.17.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agentUniverse"
-version = "0.0.17.dev2"
+version = "0.0.17.dev4"
 description = "agentUniverse is a framework for developing applications powered by multi-agent base on large language model."
 
 authors = ["AntGroup <jerry.zzw@antgroup.com>"]