agentx-python 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

agentx/integrations/google_adk.py

@@ -75,6 +75,9 @@ class AgentXADKPlugin(BasePlugin):
         self._session_id = session_id
         # invocation_id → accumulated run state
         self._runs: Dict[str, Dict[str, Any]] = {}
+        # invocation_id → pre-buffered user input text
+        # (on_user_message_callback fires *before* before_run_callback)
+        self._pending_inputs: Dict[str, str] = {}
         # id(tool_context) → start time float
         self._tool_starts: Dict[int, float] = {}
 
@@ -82,30 +85,29 @@ class AgentXADKPlugin(BasePlugin):
     # Run lifecycle
     # ------------------------------------------------------------------
 
+    async def on_user_message_callback(
+        self, *, invocation_context: Any, user_message: Any
+    ) -> None:
+        # Called *before* before_run_callback, so self._runs doesn't exist yet.
+        # Buffer the input and consume it in before_run_callback.
+        inv_id = invocation_context.invocation_id
+        text = _content_to_text(user_message)
+        if text:
+            self._pending_inputs[inv_id] = text
+
     async def before_run_callback(self, *, invocation_context: Any) -> None:
         inv_id = invocation_context.invocation_id
         agent_name = getattr(invocation_context.agent, "name", None) or self._agent_name
         self._runs[inv_id] = {
             "start": time.time(),
             "name": agent_name,
-            "input": None,
+            "input": self._pending_inputs.pop(inv_id, None),
             "output": None,
             "model": None,
             "tool_calls": [],
             "error": None,
         }
 
-    async def on_user_message_callback(
-        self, *, invocation_context: Any, user_message: Any
-    ) -> None:
-        inv_id = invocation_context.invocation_id
-        state = self._runs.get(inv_id)
-        if state is None:
-            return
-        text = _content_to_text(user_message)
-        if text and state["input"] is None:
-            state["input"] = text
-
     async def after_run_callback(self, *, invocation_context: Any) -> None:
         inv_id = invocation_context.invocation_id
         state = self._runs.pop(inv_id, None)

agentx/version.py

@@ -1 +1 @@
-VERSION = "0.5.1"
+VERSION = "0.5.2"

{agentx_python-0.5.1.dist-info → agentx_python-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.5.1
+Version: 0.5.2
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team

{agentx_python-0.5.1.dist-info → agentx_python-0.5.2.dist-info}/RECORD

@@ -2,7 +2,7 @@ agentx/__init__.py,sha256=4iMiGLNU4S-I-WyX0Z7l0BFUY_krVJ3seMQPqyO-ReA,602
 agentx/agentx.py,sha256=tw3CUnkvTM1eopoO7u8TgKnAz9t0HNSxrVcjIJsytAg,3469
 agentx/exceptions.py,sha256=TKooc1_pQ2P_4DqpLfZx_ap-6cfrjFLW8JS2qaGx_1M,1237
 agentx/util.py,sha256=yOV3VVdc0KUbw5pEkSRiO2lZzEtL1vOsnB_f5tFk948,665
-agentx/version.py,sha256=Rm_tMXSiQe1ocDOCUKKIk-I7ZEL21NXSJBaSKX1ooCs,18
+agentx/version.py,sha256=H_2wzkwt5P7u9zKSCOnQry4NWS9DJ3fCbX50CHsFMtA,18
 agentx/evaluations/__init__.py,sha256=OmdCHiBO3Qcsck0UkOyVqSvfRs6nXbSi-2FXzyRfsog,89
 agentx/evaluations/_term.py,sha256=S4blZgH66wHvkt2hR29h8JE5xnakBeD7dlZdUW9Wfoo,2065
 agentx/evaluations/client.py,sha256=3mC0uSBRk9uO4F3a7kIMM_UUsyypF99_97E5MSpYFzM,7736
@@ -20,7 +20,7 @@ agentx/evaluations/adapters/raw.py,sha256=FkDq_mdf21-xt5EHnGyIGcS6VZxkEkFcmM9VTd
 agentx/integrations/__init__.py,sha256=xpgFqfqdMqXSdCSucQoPRaewj9w5upSmkcaQtBKrtRE,545
 agentx/integrations/anthropic.py,sha256=VzRdeICYm_zfgNBy6PEu6bYSUa9iKuSoRX6HSivWQpM,4802
 agentx/integrations/crewai.py,sha256=o7tgFE1I2T7fxOc2lG7aY1nAAwihejV3y-EJII6z1j0,3488
-agentx/integrations/google_adk.py,sha256=270HRdOuHo_4AWULoN0oyJ9XlrBKTeb4Ipjk5EZxzrY,6938
+agentx/integrations/google_adk.py,sha256=ydpor3DAdGN1LbaWsZHL_f-L48mJ3s7ckfgQLxHXnEo,7200
 agentx/integrations/google_genai.py,sha256=fNKyoqegBYhxAeADYuRaI5K6-p1h6ntAdpBaNhFziQQ,5147
 agentx/integrations/langchain.py,sha256=hSy2aRi0QzVo0eDpIRRacZg7ulJG-XqY_tH7C5YvNd0,12848
 agentx/integrations/openai_agents.py,sha256=JscJELxs8sTj3toi9ixY5TOpH-lCVBtpgxYgahIaezE,8162
@@ -32,8 +32,8 @@ agentx/tracing/__init__.py,sha256=l2mIRILE-wWrCD0fekgrIOQUEvtDObzdrkgpQUtXroU,40
 agentx/tracing/ci_types.py,sha256=zFVcZGvc1qbVDdOlEPQ1i5R6terqJHxCzFF4ZjouxxI,1423
 agentx/tracing/ingest_client.py,sha256=8aLabvWecgZsn0pkajZkI1it2TOk5boP0kC3o6TO52o,11487
 agentx/tracing/tracer.py,sha256=nJILoeQg6HV4SR05xJPy6EiSiZCpkp4yFBTrN1BO_1w,16951
-agentx_python-0.5.1.dist-info/licenses/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
-agentx_python-0.5.1.dist-info/METADATA,sha256=gq6txrb4Xjdar5TAzk-aHMiON2zM-i7CA4njI8UFLK8,6901
-agentx_python-0.5.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-agentx_python-0.5.1.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
-agentx_python-0.5.1.dist-info/RECORD,,
+agentx_python-0.5.2.dist-info/licenses/LICENSE,sha256=6ZbiPNFmv3xBb44LGhAa3PZYK0ROAztsd5LRFZDlGFE,1074
+agentx_python-0.5.2.dist-info/METADATA,sha256=hgVDJgwMhUSMtriS7IkHO-F7-G97r1sAse6ztHH6b4k,6901
+agentx_python-0.5.2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentx_python-0.5.2.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.5.2.dist-info/RECORD,,