mojentic 0.6.0__py3-none-any.whl → 0.6.2__py3-none-any.whl

mojentic/tracer/tracer_system_spec.py

@@ -18,7 +18,7 @@ class DescribeTracerSystem:
     """
     Tests for the TracerSystem class.
     """
-    
+
     def should_initialize_with_default_event_store(self):
         """
         Given no event store
@@ -27,12 +27,12 @@ class DescribeTracerSystem:
         """
         # Given / When
         tracer_system = TracerSystem()
-        
+
         # Then
         assert tracer_system.event_store is not None
         assert isinstance(tracer_system.event_store, EventStore)
         assert tracer_system.enabled is True
-    
+
     def should_initialize_with_provided_event_store(self):
         """
         Given an event store
@@ -41,13 +41,13 @@ class DescribeTracerSystem:
         """
         # Given
         event_store = EventStore()
-        
+
         # When
         tracer_system = TracerSystem(event_store=event_store)
-        
+
         # Then
         assert tracer_system.event_store is event_store
-    
+
     def should_record_llm_call(self):
         """
         Given an enabled tracer system
@@ -56,11 +56,12 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
+
         # When
         messages = [{"role": "system", "content": "You are a helpful assistant."}]
-        tracer_system.record_llm_call("test-model", messages, 0.7)
-        
+        correlation_id = "test-correlation-id"
+        tracer_system.record_llm_call("test-model", messages, 0.7, correlation_id=correlation_id)
+
         # Then
         events = tracer_system.get_events(event_type=LLMCallTracerEvent)
         assert len(events) == 1
@@ -68,7 +69,7 @@ class DescribeTracerSystem:
         assert event.model == "test-model"
         assert event.messages == messages
         assert event.temperature == 0.7
-        
+
     def should_record_llm_response(self):
         """
         Given an enabled tracer system
@@ -77,14 +78,16 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
+
         # When
+        correlation_id = "test-correlation-id"
         tracer_system.record_llm_response(
             "test-model",
             "This is a test response",
-            call_duration_ms=150.5
+            call_duration_ms=150.5,
+            correlation_id=correlation_id
         )
-        
+
         # Then
         events = tracer_system.get_events(event_type=LLMResponseTracerEvent)
         assert len(events) == 1
@@ -92,7 +95,7 @@ class DescribeTracerSystem:
         assert event.model == "test-model"
         assert event.content == "This is a test response"
         assert event.call_duration_ms == 150.5
-        
+
     def should_record_tool_call(self):
         """
         Given an enabled tracer system
@@ -101,16 +104,18 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
+
         # When
         arguments = {"query": "test query"}
+        correlation_id = "test-correlation-id"
         tracer_system.record_tool_call(
             "test-tool",
             arguments,
             "test result",
-            "TestAgent"
+            "TestAgent",
+            correlation_id=correlation_id
         )
-        
+
         # Then
         events = tracer_system.get_events(event_type=ToolCallTracerEvent)
         assert len(events) == 1
@@ -119,7 +124,7 @@ class DescribeTracerSystem:
         assert event.arguments == arguments
         assert event.result == "test result"
         assert event.caller == "TestAgent"
-        
+
     def should_record_agent_interaction(self):
         """
         Given an enabled tracer system
@@ -128,15 +133,17 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
+
         # When
+        correlation_id = "test-correlation-id"
         tracer_system.record_agent_interaction(
             "AgentA",
             "AgentB",
             "RequestEvent",
-            "12345"
+            "12345",
+            correlation_id=correlation_id
         )
-        
+
         # Then
         events = tracer_system.get_events(event_type=AgentInteractionTracerEvent)
         assert len(events) == 1
@@ -145,7 +152,7 @@ class DescribeTracerSystem:
         assert event.to_agent == "AgentB"
         assert event.event_type == "RequestEvent"
         assert event.event_id == "12345"
-        
+
     def should_not_record_when_disabled(self):
         """
         Given a disabled tracer system
@@ -154,16 +161,17 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem(enabled=False)
-        
+
         # When
-        tracer_system.record_llm_call("test-model", [])
-        tracer_system.record_llm_response("test-model", "response")
-        tracer_system.record_tool_call("test-tool", {}, "result")
-        tracer_system.record_agent_interaction("AgentA", "AgentB", "Event")
-        
+        correlation_id = "test-correlation-id"
+        tracer_system.record_llm_call("test-model", [], correlation_id=correlation_id)
+        tracer_system.record_llm_response("test-model", "response", correlation_id=correlation_id)
+        tracer_system.record_tool_call("test-tool", {}, "result", correlation_id=correlation_id)
+        tracer_system.record_agent_interaction("AgentA", "AgentB", "Event", correlation_id=correlation_id)
+
         # Then
         assert len(tracer_system.get_events()) == 0
-        
+
     def should_filter_events_by_time_range(self):
         """
         Given several tracer events with different timestamps
@@ -172,26 +180,27 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
+
         # Create events with specific timestamps
         now = time.time()
+        correlation_id = "test-correlation-id"
         tracer_system.event_store.store(
-            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now - 100, model="model1", messages=[])
+            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now - 100, model="model1", messages=[], correlation_id=correlation_id)
         )
         tracer_system.event_store.store(
-            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now - 50, model="model2", messages=[])
+            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now - 50, model="model2", messages=[], correlation_id=correlation_id)
         )
         tracer_system.event_store.store(
-            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now, model="model3", messages=[])
+            LLMCallTracerEvent(source=DescribeTracerSystem, timestamp=now, model="model3", messages=[], correlation_id=correlation_id)
         )
-        
+
         # When
         result = tracer_system.get_events(start_time=now - 75, end_time=now - 25)
-        
+
         # Then
         assert len(result) == 1
         assert result[0].model == "model2"
-        
+
     def should_get_last_n_tracer_events(self):
         """
         Given several tracer events of different types
@@ -200,20 +209,21 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        
-        tracer_system.record_llm_call("model1", [])
-        tracer_system.record_tool_call("tool1", {}, "result1")
-        tracer_system.record_llm_call("model2", [])
-        tracer_system.record_llm_call("model3", [])
-        
+
+        correlation_id = "test-correlation-id"
+        tracer_system.record_llm_call("model1", [], correlation_id=correlation_id)
+        tracer_system.record_tool_call("tool1", {}, "result1", correlation_id=correlation_id)
+        tracer_system.record_llm_call("model2", [], correlation_id=correlation_id)
+        tracer_system.record_llm_call("model3", [], correlation_id=correlation_id)
+
         # When
         result = tracer_system.get_last_n_tracer_events(2, event_type=LLMCallTracerEvent)
-        
+
         # Then
         assert len(result) == 2
         assert result[0].model == "model2"
         assert result[1].model == "model3"
-        
+
     def should_enable_and_disable(self):
         """
         Given a tracer system
@@ -223,19 +233,19 @@ class DescribeTracerSystem:
         # Given
         tracer_system = TracerSystem(enabled=False)
         assert not tracer_system.enabled
-        
+
         # When
         tracer_system.enable()
-        
+
         # Then
         assert tracer_system.enabled
-        
+
         # When
         tracer_system.disable()
-        
+
         # Then
         assert not tracer_system.enabled
-        
+
     def should_clear_events(self):
         """
         Given a tracer system with events
@@ -244,12 +254,13 @@ class DescribeTracerSystem:
         """
         # Given
         tracer_system = TracerSystem()
-        tracer_system.record_llm_call("model1", [])
-        tracer_system.record_tool_call("tool1", {}, "result1")
+        correlation_id = "test-correlation-id"
+        tracer_system.record_llm_call("model1", [], correlation_id=correlation_id)
+        tracer_system.record_tool_call("tool1", {}, "result1", correlation_id=correlation_id)
         assert len(tracer_system.get_events()) == 2
-        
+
         # When
         tracer_system.clear()
-        
+
         # Then
-        assert len(tracer_system.get_events()) == 0
+        assert len(tracer_system.get_events()) == 0

{mojentic-0.6.0.dist-info → mojentic-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mojentic
-Version: 0.6.0
+Version: 0.6.2
 Summary: Mojentic is an agentic framework that aims to provide a simple and flexible way to assemble teams of agents to solve complex problems.
 Author-email: Stacey Vetzal <stacey@vetzal.com>
 Project-URL: Homepage, https://github.com/svetzal/mojentic

{mojentic-0.6.0.dist-info → mojentic-0.6.2.dist-info}/RECORD

@@ -22,7 +22,7 @@ _examples/list_models.py,sha256=8noMpGeXOdX5Pf0NXCt_CRurOKEg_5luhWveGntBhe8,578
 _examples/oversized_embeddings.py,sha256=_z2JoqZn0g7VtRsFVWIkngVqzjhQQvCEUYWVxs1I7MM,284
 _examples/raw.py,sha256=Y2wvgynFuoUs28agE4ijsLYec8VRjiReklqlCH2lERs,442
 _examples/react.py,sha256=VQ-5MmjUXoHzBFPTV_JrocuOkDzZ8oyUUSYLlEToJ_0,939
-_examples/recursive_agent.py,sha256=lFmGXpAeNYmKZSCBsGgnKd8wptzFnDLcjFNcdDNUImU,3018
+_examples/recursive_agent.py,sha256=ZDQ9Ts35JdxK6-LfO5GNmXZM2ru9w3Wpa1HXsmzbmqw,3016
 _examples/routed_send_response.py,sha256=FHhTy7y2f7GY_WAWZYqOnFqF1xeJTBuS5EjfJGsJ3S8,4146
 _examples/simple_llm.py,sha256=Cpjj1S8wXLovta1OcutTqp-fls-zVY4l5TAkbEqpA0s,1231
 _examples/simple_llm_repl.py,sha256=bpk7S1-yQ-oZdRD_0ICV37NQxntgvgjiFD5KdIU40Ig,1406
@@ -31,7 +31,7 @@ _examples/simple_tool.py,sha256=9hRo2YZk7NmprTIkLmEyueCbZ0Ls1Ey4l4eDvvgoWy8,1288
 _examples/solver_chat_session.py,sha256=4mTsgQVJWwujDFX6NUIK6kKr45E09Q4yIkZF-FdoTtg,2148
 _examples/streaming.py,sha256=P1CGvm11yi_tojnha2Qmp_OFB5A3T-F5dWIzxEclNhM,1137
 _examples/tell_user_example.py,sha256=gPGS_3KAHUIFsnvkDXX6u1mVqlsB67zsyNdGuEho60o,1274
-_examples/tracer_demo.py,sha256=IG9llYs9ARFbjkK-acW091S73z0jr2ZS_0GWps88dAQ,3994
+_examples/tracer_demo.py,sha256=X4nZtMX4yPDLF0i-pb0gXDHvjUwgCDJ6Q6RS7lNRlVs,7101
 _examples/working_memory.py,sha256=xbCHb6Y0Um6ai5T9fWAX4oQOcCQJDinsn902qm8q48E,1935
 _examples/react/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 _examples/react/formatters.py,sha256=nbgWve0YGKaRY6P4tXV84PYw8YDN4neefiq7mR2I_tI,1065
@@ -54,13 +54,13 @@ mojentic/agents/base_llm_agent_spec.py,sha256=1kzayCtAZY7oWloaFMh7-NkTtiKihDCLD3
 mojentic/agents/correlation_aggregator_agent.py,sha256=okdq92rMNwIXTgnuwwn3zp4JzSxTh05vGWzTGocOgbU,1134
 mojentic/agents/iterative_problem_solver.py,sha256=RLt8MCM99d-Z26EhC5UnmE0M97EM475VGBVsNio8DSA,4329
 mojentic/agents/output_agent.py,sha256=I9GHcWvQhWxEcvjzj47scKo_QF8_6yR17ZUEHhwC94Y,231
-mojentic/agents/simple_recursive_agent.py,sha256=DcJFttTAYurRIObTL8kfbHUs5uvuqh6prEgA5nmxCLo,9950
+mojentic/agents/simple_recursive_agent.py,sha256=B3QcOeIuamoLp0MsAwCjJLgVCaODsEi03xAAfpMWFRM,9846
 mojentic/context/__init__.py,sha256=MKMP7ViQg8gMtLFvn9pf47XMc5beA5Wx95K4dEw93z8,55
 mojentic/context/shared_working_memory.py,sha256=Zt9MNGErEkDIUAaHvyhEOiTaEobI9l0MV4Z59lQFBr0,396
 mojentic/llm/__init__.py,sha256=mwdPpTRofw7_KSlW6ZQmcM-GSpyISWyI2bxphKpV7A0,180
 mojentic/llm/chat_session.py,sha256=H2gY0mZYVym8jC69VHsmKaRZ9T87Suyw0-TW5r850nA,3992
 mojentic/llm/chat_session_spec.py,sha256=8-jj-EHV2WwWuvo3t8I75kSEAYiG1nR-OEwkkLTi_z0,3872
-mojentic/llm/llm_broker.py,sha256=8h1J6Avue_iJhGzMTVuF3Zmgek--P_Q4iVRJT8-_b14,8823
+mojentic/llm/llm_broker.py,sha256=DFi0zgFYgdOKrJx7DomktnbBQKppi4mB_RcDvXTf9-4,9170
 mojentic/llm/llm_broker_spec.py,sha256=40lzmYm_6Zje6z5MQ7_o3gSBThLsNW_l_1mZTUVll6A,5342
 mojentic/llm/message_composers.py,sha256=Fo9o7UGZOOIYoGI_DyOfP_oMiEiCMQz-zdWdTKtozVk,12108
 mojentic/llm/message_composers_spec.py,sha256=RNW14Zb-kIBWT5Wy9cZQyxHPrcRIaBFjBYCR9N-m0kE,12109
@@ -74,7 +74,7 @@ mojentic/llm/gateways/models.py,sha256=lnGvr3E4C5n15v0aI8Cc0FMOH6GBCrk5_XaEEe3vK
 mojentic/llm/gateways/ollama.py,sha256=629fpZhC0zVCYqj360-PKTT4mQOLec5nzzvfMtS_mLQ,7581
 mojentic/llm/gateways/ollama_messages_adapter.py,sha256=kUN_p2FyN88_trXMcL-Xsn9xPBU7pGKlJwTUEUCf6G4,1404
 mojentic/llm/gateways/ollama_messages_adapter_spec.py,sha256=gVRbWDrHOa1EiZ0CkEWe0pGn-GKRqdGb-x56HBQeYSE,4981
-mojentic/llm/gateways/openai.py,sha256=NQECA41PS21RhXjwQdbH3DAP15WqkEl22UlS1ka_lLM,5434
+mojentic/llm/gateways/openai.py,sha256=G5bdQ0AkMlG-nOCvsw1gziyIerCiA1__RPfFK4M6fXI,5475
 mojentic/llm/gateways/openai_message_adapter_spec.py,sha256=ITBSV5njldV_x0NPgjmg8Okf9KzevQJ8dTXM-t6ubcg,6612
 mojentic/llm/gateways/openai_messages_adapter.py,sha256=9Btcr77RzlqZIdkNuGDGitk9D7SQTv4GsdKXaN2Fuv4,3863
 mojentic/llm/gateways/tokenizer_gateway.py,sha256=ztuqfunlJ6xmyUPPHcC_69-kegiNJD6jdSEde7hDh2w,485
@@ -88,7 +88,7 @@ mojentic/llm/tools/current_datetime.py,sha256=JkIFlBBnvE4CMcgZizqITVTtNfL7zeeLxD
 mojentic/llm/tools/date_resolver.py,sha256=VDxX59PTHo0PstDxaJUo3SF7vkvQoG92bo6j8ABq8y4,2185
 mojentic/llm/tools/date_resolver_spec.py,sha256=OaRvJyhSN8sgi75euk4E5ImaqUmvQdgZKY8u_NOiPWE,1185
 mojentic/llm/tools/file_manager.py,sha256=X8Uw4XtdH_Ol2EB3SN11-GYlC1diJ1cAywU9_EuCjCg,3788
-mojentic/llm/tools/llm_tool.py,sha256=MhPkRkM0GihZ8_82Y7Xn3zhfuA_Ic0zmC9ouHXMWTT0,1660
+mojentic/llm/tools/llm_tool.py,sha256=MV1aexLjiPhcm1B1lvIkSJlokpPfuhF_Amqg7zCa6VQ,1695
 mojentic/llm/tools/llm_tool_spec.py,sha256=OPudmuqSxkJBUaL0-qABri-tvrAyb6QI9Ac5B5iTo3I,2112
 mojentic/llm/tools/organic_web_search.py,sha256=X_WQSSFgeMp5jlm0tnqGOHDIZ2KHDvz3k6GF-XqsuN4,1301
 mojentic/llm/tools/tell_user_tool.py,sha256=bF_PTfF0-skCi_Exw6EPpdfoNUZRJ4MWg-XCe4VYd6s,999
@@ -114,15 +114,15 @@ mojentic/llm/tools/ephemeral_task_manager/start_task_tool_spec.py,sha256=oDeVPPf
 mojentic/tracer/__init__.py,sha256=075nYiRvYTzkzombGY-31Gy0JMPM7Zn645F6_j5-k6I,445
 mojentic/tracer/event_store.py,sha256=eqbze4YRuiMDr8TCtMNHi-U_6-CLgstOHkLmtM5hc6Q,3691
 mojentic/tracer/event_store_spec.py,sha256=OWXWbeMCYU9dfySmbpmDA4KmwLlSaFy4ZpRt6BsTkFE,6522
-mojentic/tracer/null_tracer.py,sha256=nBVs3CiVdbFXoiwtth6eh3m3z1Gggs6yEIBqX4QbW-U,6338
-mojentic/tracer/tracer_events.py,sha256=DWv2PEU1CiUC1UnDScu9_DESEiTRIwpw0EMymFg4FJw,5383
+mojentic/tracer/null_tracer.py,sha256=zvNLKjQSu3na6SWoFa16k9fFyjmSD3Ihw5rTjRUXrSs,6872
+mojentic/tracer/tracer_events.py,sha256=mtONmApybFmbbt_4xN6h-wGrcZZneI3sJJ_8Z68V-tA,5405
 mojentic/tracer/tracer_events_spec.py,sha256=v84P3IjVr0_TGjwJD1w2SSNlTDJj0a4P8tMkDrsbOTA,3549
-mojentic/tracer/tracer_system.py,sha256=iBU4aHNYMSaIMh1q5CHcItufzYE6GXdw2wNtXu4NQRk,9415
-mojentic/tracer/tracer_system_spec.py,sha256=r5_GYEN78t0nJFawKvya8eyIUbmSgenyflPld-7Y448,7968
+mojentic/tracer/tracer_system.py,sha256=7CPy_2tlsHtXQ4DcO5oo52N9a9WS0GH-mjeINzu62ls,9989
+mojentic/tracer/tracer_system_spec.py,sha256=TNm0f9LV__coBx0JGEKyzzNN9mFjCSG_SSrRISO8Xeg,8632
 mojentic/utils/__init__.py,sha256=lqECkkoFvHFttDnafRE1vvh0Dmna_lwupMToP5VvX5k,115
 mojentic/utils/formatting.py,sha256=bPrwwdluXdQ8TsFxfWtHNOeMWKNvAfABSoUnnA1g7c8,947
-mojentic-0.6.0.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
-mojentic-0.6.0.dist-info/METADATA,sha256=jZvs5Umlb93eIs366WDSch4GYJ2x_p6a038wLS7EreY,5383
-mojentic-0.6.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-mojentic-0.6.0.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
-mojentic-0.6.0.dist-info/RECORD,,
+mojentic-0.6.2.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
+mojentic-0.6.2.dist-info/METADATA,sha256=VEeadz_96qq1APeO0zg-Nc2U5nBErrr2q6R1WfmEUlQ,5383
+mojentic-0.6.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mojentic-0.6.2.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
+mojentic-0.6.2.dist-info/RECORD,,

{mojentic-0.6.0.dist-info → mojentic-0.6.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any