agno 2.3.7__py3-none-any.whl → 2.3.8__py3-none-any.whl

agno/workflow/parallel.py

@@ -1,6 +1,7 @@
 import asyncio
 import warnings
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from contextvars import copy_context
 from copy import deepcopy
 from dataclasses import dataclass
 from typing import Any, AsyncIterator, Awaitable, Callable, Dict, Iterator, List, Optional, Union
@@ -267,8 +268,9 @@ class Parallel:
 
         with ThreadPoolExecutor(max_workers=len(self.steps)) as executor:
             # Submit all tasks with their original indices
+            # Use copy_context().run to propagate context variables to child threads
             future_to_index = {
-                executor.submit(execute_step_with_index, indexed_step): indexed_step[0]
+                executor.submit(copy_context().run, execute_step_with_index, indexed_step): indexed_step[0]
                 for indexed_step in indexed_steps
             }
 
@@ -449,7 +451,11 @@ class Parallel:
 
         with ThreadPoolExecutor(max_workers=len(self.steps)) as executor:
             # Submit all tasks
-            futures = [executor.submit(execute_step_stream_with_index, indexed_step) for indexed_step in indexed_steps]
+            # Use copy_context().run to propagate context variables to child threads
+            futures = [
+                executor.submit(copy_context().run, execute_step_stream_with_index, indexed_step)
+                for indexed_step in indexed_steps
+            ]
 
             # Process events from queue as they arrive
             completed_steps = 0

agno/workflow/step.py

@@ -644,7 +644,7 @@ class Step:
                             session_state=session_state_copy,  # Send a copy to the executor
                             stream=True,
                             stream_events=stream_events,
-                            yield_run_response=True,
+                            yield_run_output=True,
                             run_context=run_context,
                             **kwargs,
                         )
@@ -653,7 +653,7 @@ class Step:
                         for event in response_stream:
                             if isinstance(event, RunOutput) or isinstance(event, TeamRunOutput):
                                 active_executor_run_response = event
-                                break
+                                continue
                             # Only yield executor events if stream_executor_events is True
                             if stream_executor_events:
                                 enriched_event = self._enrich_event_with_context(
@@ -1174,7 +1174,7 @@ class Step:
                             stream=True,
                             stream_events=stream_events,
                             run_context=run_context,
-                            yield_run_response=True,
+                            yield_run_output=True,
                             **kwargs,
                         )
 

{agno-2.3.7.dist-info → agno-2.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.3.7
+Version: 2.3.8
 Summary: Agno: a lightweight library for building Multi-Agent Systems
 Author-email: Ashpreet Bedi <ashpreet@agno.com>
 Project-URL: homepage, https://agno.com
@@ -153,7 +153,7 @@ Requires-Dist: mcp>=1.9.2; extra == "mcp"
 Provides-Extra: mem0
 Requires-Dist: mem0ai; extra == "mem0"
 Provides-Extra: memori
-Requires-Dist: memorisdk==2.3.3; extra == "memori"
+Requires-Dist: memorisdk==3.0.5; extra == "memori"
 Provides-Extra: newspaper
 Requires-Dist: newspaper4k; extra == "newspaper"
 Requires-Dist: lxml_html_clean; extra == "newspaper"

{agno-2.3.7.dist-info → agno-2.3.8.dist-info}/RECORD

@@ -6,7 +6,7 @@ agno/media.py,sha256=eTfYb_pwhX_PCIVPSrW4VYRqmoxKABEF1aZClrVvQ30,16500
 agno/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agno/table.py,sha256=9hHFnInNsrj0ZKtWjGDP5c6kbmNdtQvDDYT2j2CZJ6o,198
 agno/agent/__init__.py,sha256=s7S3FgsjZxuaabzi8L5n4aSH8IZAiZ7XaNNcySGR-EQ,1051
-agno/agent/agent.py,sha256=hxNxn2tyhR8q7gIGujTUiaSRRw9wfnfsr6-i_vodcsY,496743
+agno/agent/agent.py,sha256=ljgnuhq2UTjMC6-hSSnb4eM66x8ItubRpNWvR4tpNtE,497820
 agno/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agno/api/agent.py,sha256=fKlQ62E_C9Rjd7Zus3Gs3R1RG-IhzFV-ICpkb6SLqYc,932
 agno/api/api.py,sha256=gFhVjxJYkQsw8mBl2fhoStMPGlyJ37DJaqgUOwZVvQI,1021
@@ -60,21 +60,21 @@ agno/db/migrations/v1_to_v2.py,sha256=gj8deaEWUxOr0qJyMfjOpV3LxEh-otOSOxDckeUq0q
 agno/db/migrations/versions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agno/db/migrations/versions/v2_3_0.py,sha256=WikuOtgQs_Tq8o3JndCMG6HAMzKvz4GVth9V5V5RuvE,35161
 agno/db/mongo/__init__.py,sha256=NG2WC2jE5PX1kMXTYCVHjhomf5DAJgvzSj45QHdvl2E,466
-agno/db/mongo/async_mongo.py,sha256=7s26a6NH5y6Wdb_8vSRG4cHyk2rzkx6rhbFAbizqV1M,108485
-agno/db/mongo/mongo.py,sha256=zQG3uyiBgJZQlnkBOth9EXc8KedKr14p7gRkyMn3HNM,99320
+agno/db/mongo/async_mongo.py,sha256=0jh36UmtYqeQOyMT9MB-2GE7aqCUEN0jGWJzuv8ZfCo,107154
+agno/db/mongo/mongo.py,sha256=GHeDRw2VhhuTDtLcXpsf2UMU_WlJbJCc6n131Omn3Bw,98457
 agno/db/mongo/schemas.py,sha256=Q3JrMZ3zCKZcHilh5CoQePI5d5dRjHAtjs5QuWrxiWc,3098
 agno/db/mongo/utils.py,sha256=1KrbF1PR_707e9SqXZcquaw_N7mRYm9HjJnvYwBjgG8,9921
 agno/db/mysql/__init__.py,sha256=ohBMZ1E6ctioEF0XX5PjC4LtUQrc6lFkjsE4ojyXA8g,63
-agno/db/mysql/mysql.py,sha256=Ve4ojP3slr_qMxXQ0hi4XDQPtOtgm2jBR4CEg5ao6j8,121370
+agno/db/mysql/mysql.py,sha256=io5bOVEXvcM85DJsJhYUf5w4nCeeF6TtU9ypZAM3eiQ,120209
 agno/db/mysql/schemas.py,sha256=_4p2IWt0zVq-fHuxFhEpWLWxlp3QPKk8xtNMiB1TNOY,9067
 agno/db/mysql/utils.py,sha256=PdqN-SxM-ox8HU9CZyxzvs2D1FE2vdZFVCyFgFcQsyU,12366
 agno/db/postgres/__init__.py,sha256=Ojk00nTCzQFiH2ViD7KIBjgpkTKLRNPCwWnuXMKtNXY,154
-agno/db/postgres/async_postgres.py,sha256=TQTS6u15RNaQFwvgU6lGxo8eJavElw9TlBO6lPq7amw,107237
-agno/db/postgres/postgres.py,sha256=ouC5XrcGL95R30OEEjfdtMIb7jdJTjS_l6xzf1kS2cw,118023
+agno/db/postgres/async_postgres.py,sha256=pZAVNSk41_nD5B4DtUJ1jDX1l2ZDe-tAdkhGR-ru5pQ,106024
+agno/db/postgres/postgres.py,sha256=rbkI4YJQYHqryMrpZOLtP07i8izbcschg7XQZlVI160,116664
 agno/db/postgres/schemas.py,sha256=G0BRG9GcUcO43-TU52nKSEog8v3tdJNwIhEjts9dCns,8405
 agno/db/postgres/utils.py,sha256=UE3UQZ-h7fADAKBsX4BWcDka54YNROEpBrlfTmDvpqc,15471
 agno/db/redis/__init__.py,sha256=rZWeZ4CpVeKP-enVQ-SRoJ777i0rdGNgoNDRS9gsfAc,63
-agno/db/redis/redis.py,sha256=CDv2no3UEycJkKESgsggDyIFfM2rToDp_Gcb2yZuNUA,82524
+agno/db/redis/redis.py,sha256=WgP-f8OfPTUEISY8O8meYbcJsFA3rpmXEthVQZRKdoM,82222
 agno/db/redis/schemas.py,sha256=90rDTGUyN1kYZ2KgeFjyOu1SwqYoTc1ErePxGqFlbj0,5217
 agno/db/redis/utils.py,sha256=vQpLI-V442B6m0q_l8IVEJctLD4P-kawbCm5A_Fnotg,11326
 agno/db/schemas/__init__.py,sha256=g72Zr5_nm00yXHStv4pf9PG9bGLKXEK7Av6YQtrDbCQ,147
@@ -85,12 +85,12 @@ agno/db/schemas/memory.py,sha256=dA7It1cN60dId-GpowmKTORy4Q6r9L49ZpGnnZkXmds,212
 agno/db/schemas/metrics.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agno/db/singlestore/__init__.py,sha256=dufbaod8ZZIeZIVi0hYJQ8Eu2DfIfWdIy00cpqAsx9U,87
 agno/db/singlestore/schemas.py,sha256=gnCpoRAztbPiZOCfx9MGBx1sySijwogajSjxIeGpsaA,8923
-agno/db/singlestore/singlestore.py,sha256=Q0MhS41gCm4xgtoHuLQjvUEHJYurqtlJD7aP5_46rx0,118882
+agno/db/singlestore/singlestore.py,sha256=mShU-mGdOamf82VihF96e1tVjzSNol8Zs94f0S--HLs,118169
 agno/db/singlestore/utils.py,sha256=w2FVFIBpFJK6nKJVt1sgv9N-esmfpGY_8ViFRI69I2I,13677
 agno/db/sqlite/__init__.py,sha256=09V3i4y0-tBjt60--57ivZ__SaaS67GCsDT4Apzv-5Y,138
-agno/db/sqlite/async_sqlite.py,sha256=Amv3zha6FTHCuwAuYUzw0jGoIbNC5s6sv6QTJoMRPEw,123366
+agno/db/sqlite/async_sqlite.py,sha256=62P-HurucmK3pk5aM8vjqHbHuuor5pDSuJW-vCLyE74,121942
 agno/db/sqlite/schemas.py,sha256=1lrIFWEhXYZJIlQYEnOl3XLp3UfsE5ILj2HUPmOf3cg,8265
-agno/db/sqlite/sqlite.py,sha256=3jiDLjs1vM3sTd6526v4JJhrpjL6cMaW9Xik0nTA57k,120340
+agno/db/sqlite/sqlite.py,sha256=XjRJAmClEzYS6itHuPRD6RB9-iTtHgDkBmNLKy2vJOM,119118
 agno/db/sqlite/utils.py,sha256=sgp-KR7A5naTqTORHYHyP_21hG9LffOzCgqbx5S8RfE,15171
 agno/db/surrealdb/__init__.py,sha256=C8qp5-Nx9YnSmgKEtGua-sqG_ntCXONBw1qqnNyKPqI,75
 agno/db/surrealdb/metrics.py,sha256=oKDRyjRQ6KR3HaO8zDHQLVMG7-0NDkOFOKX5I7mD5FA,10336
@@ -181,7 +181,7 @@ agno/memory/strategies/base.py,sha256=13xCiBPtFUrpDaWbiz8zmNGjopT4VSIkcHVxfnvJ4W
 agno/memory/strategies/summarize.py,sha256=4M9zWTsooC3EtHpZoC7Z-yFaQgQoebRMNfZPitdsvB0,7307
 agno/memory/strategies/types.py,sha256=b3N5jOG_dM4AxT7vGagFIc9sqUUjxFtRHSoH4_AhEx8,1225
 agno/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agno/models/base.py,sha256=3ZZpyACrsSjhpK3YOjRaLXR2hQFSQnKGpIKWPVGj9Ww,108826
+agno/models/base.py,sha256=_ldhrYebV77fEzHtnILNILPW2aG4KwMjsT-THnYDwHA,114403
 agno/models/defaults.py,sha256=1_fe4-ZbNriE8BgqxVRVi4KGzEYxYKYsz4hn6CZNEEM,40
 agno/models/message.py,sha256=5-Nvqd6rXNca37yiVVQgBJY4w_gudVawg-ZQHcUX4SE,19763
 agno/models/metrics.py,sha256=81IILXZwGmOTiWK003bi5mg4bM1f4LCWbwyamjFzp18,4500
@@ -247,7 +247,7 @@ agno/models/ollama/chat.py,sha256=Szc8rEWRvQ2CW50V5xAuccX4Ozc1BAV9wUPbFJhY_J8,16
 agno/models/openai/__init__.py,sha256=OssVgQRpsriU6aJZ3lIp_jFuqvX6y78L4Fd3uTlmI3E,225
 agno/models/openai/chat.py,sha256=KZupbtQxrNiHH9m-y6hrldLq0dGe0XYjvShJQsz-Kjw,40096
 agno/models/openai/like.py,sha256=wmw9PfAVqluBs4MMY73dgjelKn1yl5JDKyCRvaNFjFw,745
-agno/models/openai/responses.py,sha256=270XSJx6HaYi_vn9ENxZ_xL22x3tqmVChmHsm4z6coA,46643
+agno/models/openai/responses.py,sha256=SahBoPgh8KFhRIE1QymhNhQoydFHFEmpBWxmFuixT88,46702
 agno/models/openrouter/__init__.py,sha256=ZpZhNyy_EGSXp58uC9e2iyjnxBctql7GaY8rUG-599I,90
 agno/models/openrouter/openrouter.py,sha256=fdFZy-1ZwvDKlO0b9TiXdtNe5ScSU4RJSYw08DojK_4,3284
 agno/models/perplexity/__init__.py,sha256=JNmOElDLwcZ9_Lk5owkEdgwmAhaH3YJ-VJqOI8rgp5c,90
@@ -284,7 +284,7 @@ agno/os/interfaces/base.py,sha256=vXkr1tRjWHTcmBlQFzvQjqURLhObmFtUAx82uij_j48,54
 agno/os/interfaces/a2a/__init__.py,sha256=Fs7--dx9drvtVS9QjsCCm0P7c-hJ7TzU8gNwKTQsZDA,62
 agno/os/interfaces/a2a/a2a.py,sha256=UiTX-GOuXxp1CQzRgtmgSSYNn2iFC_vKYnC37OidygM,1337
 agno/os/interfaces/a2a/router.py,sha256=OpNCHhYoCUyfcuRtiOdvUrUZnxfFB1jhMWraLA85XMg,10985
-agno/os/interfaces/a2a/utils.py,sha256=rTn9x0UGbX53EiAsljf87PmzYwLuvqaobUWpCn35ut0,39285
+agno/os/interfaces/a2a/utils.py,sha256=J_m_l3noqfjjRRQ-kQQMAab1CnvDPh-M1regJ9_o5Yg,39286
 agno/os/interfaces/agui/__init__.py,sha256=1zrGICk4roXUINwSFZfqH6sBsbHmD5KjGYVJMGg4fKQ,66
 agno/os/interfaces/agui/agui.py,sha256=PKGoDDbtQFmEC0zRwZmsjS_5t9bJWJ-ZGwxEQsu9P-U,1415
 agno/os/interfaces/agui/router.py,sha256=U2w26v2oOCCOl5t5XduQChO3_eJBNH4Mk9BH77KKJag,5257
@@ -298,7 +298,7 @@ agno/os/interfaces/whatsapp/router.py,sha256=K7Ex4Xq-4hPcRKoVxBkb8KoGxH6WXC1bubd
 agno/os/interfaces/whatsapp/security.py,sha256=0GYdG28yeSpV47nMWASaFSTiGESSstcFAeL6amobvFE,1772
 agno/os/interfaces/whatsapp/whatsapp.py,sha256=tNJncEu_hm0lFOHbjaSoz5-VIKORR_pyW26cseBLfbs,989
 agno/os/middleware/__init__.py,sha256=EYsNzeixFgL3n8kepKWXT42fTTmrNyD8b8rOdXecMRI,94
-agno/os/middleware/jwt.py,sha256=xw9jQkVFMTTzUVird1k-egAYBmPd174L06YflYCvB5Q,9468
+agno/os/middleware/jwt.py,sha256=O3O_h7WeO7xN6uj4NiBMgz9tTvJpfX2c-f4w4-ykBNA,9575
 agno/os/routers/__init__.py,sha256=du4LO9aZwiY1t59VcV9M6wiAfftFFlUZc-YXsTGy9LI,97
 agno/os/routers/health.py,sha256=AO3ec6Xi1wXOwUUFJhEcM45488qu6aZRJTwF2GLz6Ak,992
 agno/os/routers/home.py,sha256=xe8DYJkRgad55qiza0lHt8pUIV5PLSyu2MkybjuPDDE,1708
@@ -346,7 +346,7 @@ agno/session/summary.py,sha256=9JnDyQyggckd3zx6L8Q5f-lglZvrFQxvPjGU8gLCgR4,10292
 agno/session/team.py,sha256=-MkB6qQCrnXLKko8L5s9fJOWPsjeK5Gx0SXEPoOwSFQ,13437
 agno/session/workflow.py,sha256=nPHnh1N0SJby5JRjysCUI-kTDCelQMFfqosEnnLzPIg,19690
 agno/team/__init__.py,sha256=toHidBOo5M3n_TIVtIKHgcDbLL9HR-_U-YQYuIt_XtE,847
-agno/team/team.py,sha256=55zfNQ7JvK2XNovOsABhgkjZ0xtHcX80p1d6ZT5gYnY,434720
+agno/team/team.py,sha256=TOc7jPk0sTVqV-3nm6By9QlX_632fsJtn-FjIvAC2Co,434423
 agno/tools/__init__.py,sha256=jNll2sELhPPbqm5nPeT4_uyzRO2_KRTW-8Or60kioS0,210
 agno/tools/agentql.py,sha256=S82Z9aTNr-E5wnA4fbFs76COljJtiQIjf2grjz3CkHU,4104
 agno/tools/airflow.py,sha256=uf2rOzZpSU64l_qRJ5Raku-R3Gky-uewmYkh6W0-oxg,2610
@@ -405,7 +405,6 @@ agno/tools/local_file_system.py,sha256=wWyhM5-lqDgDO_YNzRA8ekG-m2n89k8fWr8M1BWiQ
 agno/tools/lumalab.py,sha256=6WnZXbThKY2jL9zLswq1PVsbFm2jz81qshWqBZi59oo,6808
 agno/tools/mcp_toolbox.py,sha256=z6qZqUOmYueBoiQtTqRWV9GohhCAd3s5D4GyqtezHeY,12760
 agno/tools/mem0.py,sha256=5W5pZwJmBTt-_l4nvBdNQHavXFSKV9mVdJg5aso4JBI,7680
-agno/tools/memori.py,sha256=tubBYj0rQFbBXadhWxwTjjmb3Rnims536AVPkGdCMcw,13181
 agno/tools/memory.py,sha256=vpMoKtCqs3m6vkuqmZ4fW9IRf1OhXHQGGaq3exJK0Xo,18449
 agno/tools/mlx_transcribe.py,sha256=kuiYZAM5ZAdkiOfFdbGJsCb0gacnJRtSTFzuX8eWGLw,6379
 agno/tools/models_labs.py,sha256=E91DZDJOU2ldgI3U_ueFiZcXLLWfbh0MsZ2pYP9c0eQ,7933
@@ -457,7 +456,7 @@ agno/tools/website.py,sha256=jzcXLEsCfGIEqV88OLePiCEjfDoAbXQDCPAr9JbP8cU,1798
 agno/tools/webtools.py,sha256=5Dyp8U2ov32DkJMpj-7plzQZcfnCL7wlXbQIfRh8jmw,1305
 agno/tools/whatsapp.py,sha256=DaDyNogtbgrinJohfU7CmVQJSZsZO-00XUbRJegN_eE,11055
 agno/tools/wikipedia.py,sha256=u6xhc5wkEGcjng4leVCmo28o6Ti7OzDgRqiOqTMnmoc,2277
-agno/tools/workflow.py,sha256=4EJlPWEf1Y9sGK_cWENevh3NbdKnGZjzW1cJOJVtI30,12556
+agno/tools/workflow.py,sha256=8__QqwLQ0uWXiwk99SoLl2rC3Sr_5nJur6CeGeU0xx4,12744
 agno/tools/x.py,sha256=gaFCel9FyDPnKLL03nkjsc9NGsfQOsS92XByp3hdwMo,14495
 agno/tools/yfinance.py,sha256=1bPUMZ7BWxEyblJLxyhzV6y4O-EBmljUM_aKfr0cf0o,11540
 agno/tools/youtube.py,sha256=H6SbBfWrmkazR8QrgXvLpqBOGAy4eHbRxkgTRJEt91Q,6667
@@ -585,14 +584,14 @@ agno/workflow/__init__.py,sha256=Ze2j811jwnsS8Sjv6u0Ap4l7Pyw_ivRof1uBYNJhb1w,609
 agno/workflow/agent.py,sha256=fKLB4kop7YYFeM1e-d3IzvVPsVF5cLNX6Z0gPn81ZR4,12170
 agno/workflow/condition.py,sha256=2zKLgrvwa3Up4aDKWSACOa7vZD0lSLX_lUW-LYJa0Yc,33939
 agno/workflow/loop.py,sha256=nDBXCRMhXHaRb-jnIbtEgY78h69zCdUrHwiJ6VXl5zo,33986
-agno/workflow/parallel.py,sha256=JusRfiYesmjwM2k0d6j3Ulm1_dVD4XupeRjuQFqXX5I,37714
+agno/workflow/parallel.py,sha256=yTtc6dkNmIGU6zOI_7EZ3mATrJZFEArUUcUG47L-DwE,38007
 agno/workflow/router.py,sha256=HQDOnErjtiQdY_EskJBV_mZq_k6d1gH6JJPIM2Yfelw,32016
-agno/workflow/step.py,sha256=2BAZQI9nNs-qR0_r1I4WSuIB_sOXI7Q-OeQtb-bzSXU,74268
+agno/workflow/step.py,sha256=BbzpETSMtlDkUsm47-TwhfGAgRM6TcWnp-orIJgOTQY,74267
 agno/workflow/steps.py,sha256=rbfue2M4qYEkgHueojCY1-cB4i1MFjO-jX6uTxyoKwk,27118
 agno/workflow/types.py,sha256=LObJ0VkUtepZ-uewv3j283S4hrCXy0eCplQzIzRG1ic,19175
 agno/workflow/workflow.py,sha256=6khBK6tbO9i2aSovduwNoQUfbcyEdoKyF4F86Uyj4gM,193181
-agno-2.3.7.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-agno-2.3.7.dist-info/METADATA,sha256=RZjF1ZE-fHQ_aXUeGZjrU5GO0HZhfkB4kiQ5zYHboE0,31018
-agno-2.3.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agno-2.3.7.dist-info/top_level.txt,sha256=MKyeuVesTyOKIXUhc-d_tPa2Hrh0oTA4LM0izowpx70,5
-agno-2.3.7.dist-info/RECORD,,
+agno-2.3.8.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+agno-2.3.8.dist-info/METADATA,sha256=ANDODVqjcZLqQzSxWL-lXsxjWQUUjUl8XThT9VSy4Og,31018
+agno-2.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agno-2.3.8.dist-info/top_level.txt,sha256=MKyeuVesTyOKIXUhc-d_tPa2Hrh0oTA4LM0izowpx70,5
+agno-2.3.8.dist-info/RECORD,,

agno/tools/memori.py

@@ -1,339 +0,0 @@
-import json
-from typing import Any, Dict, List, Optional
-
-from agno.tools.toolkit import Toolkit
-from agno.utils.log import log_debug, log_error, log_info, log_warning
-
-try:
-    from memori import Memori, create_memory_tool
-except ImportError:
-    raise ImportError("`memorisdk` package not found. Please install it with `pip install memorisdk`")
-
-
-class MemoriTools(Toolkit):
-    """
-    Memori ToolKit for Agno Agents and Teams, providing persistent memory capabilities.
-
-    This toolkit integrates Memori's memory system with Agno, allowing Agents and Teams to:
-    - Store and retrieve conversation history
-    - Search through past interactions
-    - Maintain user preferences and context
-    - Build long-term memory across sessions
-
-    Requirements:
-        - pip install memorisdk
-        - Database connection string (SQLite, PostgreSQL, etc.)
-
-    Example:
-        ```python
-        from agno.tools.memori import MemoriTools
-
-        # Initialize with SQLite (default)
-        memori_tools = MemoriTools(
-            database_connect="sqlite:///agent_memory.db",
-            namespace="my_agent",
-            auto_ingest=True  # Automatically ingest conversations
-        )
-
-        # Add to agent
-        agent = Agent(
-            model=OpenAIChat(),
-            tools=[memori_tools],
-            description="An AI assistant with persistent memory"
-        )
-        ```
-    """
-
-    def __init__(
-        self,
-        database_connect: Optional[str] = None,
-        namespace: Optional[str] = None,
-        conscious_ingest: bool = True,
-        auto_ingest: bool = True,
-        verbose: bool = False,
-        config: Optional[Dict[str, Any]] = None,
-        auto_enable: bool = True,
-        enable_search_memory: bool = True,
-        enable_record_conversation: bool = True,
-        enable_get_memory_stats: bool = True,
-        all: bool = False,
-        **kwargs,
-    ):
-        """
-        Initialize Memori toolkit.
-
-        Args:
-            database_connect: Database connection string (e.g., "sqlite:///memory.db")
-            namespace: Namespace for organizing memories (e.g., "agent_v1", "user_session")
-            conscious_ingest: Whether to use conscious memory ingestion
-            auto_ingest: Whether to automatically ingest conversations into memory
-            verbose: Enable verbose logging from Memori
-            config: Additional Memori configuration
-            auto_enable: Automatically enable the memory system on initialization
-            **kwargs: Additional arguments passed to Toolkit base class
-        """
-        tools: List[Any] = []
-        if all or enable_search_memory:
-            tools.append(self.search_memory)
-        if all or enable_record_conversation:
-            tools.append(self.record_conversation)
-        if all or enable_get_memory_stats:
-            tools.append(self.get_memory_stats)
-
-        super().__init__(name="memori_tools", tools=tools, **kwargs)
-
-        # Set default database connection if not provided
-        if not database_connect:
-            sqlite_db = "sqlite:///agno_memori_memory.db"
-            log_info(f"No database connection provided, using default SQLite database at {sqlite_db}")
-            database_connect = sqlite_db
-
-        self.database_connect = database_connect
-        self.namespace = namespace or "agno_default"
-        self.conscious_ingest = conscious_ingest
-        self.auto_ingest = auto_ingest
-        self.verbose = verbose
-        self.config = config or {}
-
-        try:
-            # Initialize Memori memory system
-            log_debug(f"Initializing Memori with database: {self.database_connect}")
-            self.memory_system = Memori(
-                database_connect=self.database_connect,
-                conscious_ingest=self.conscious_ingest,
-                auto_ingest=self.auto_ingest,
-                verbose=self.verbose,
-                namespace=self.namespace,
-                **self.config,
-            )
-
-            # Enable the memory system if auto_enable is True
-            if auto_enable:
-                self.memory_system.enable()
-                log_debug("Memori memory system enabled")
-
-            # Create the memory tool for internal use
-            self._memory_tool = create_memory_tool(self.memory_system)
-
-        except Exception as e:
-            log_error(f"Failed to initialize Memori: {e}")
-            raise ConnectionError("Failed to initialize Memori memory system") from e
-
-    def search_memory(
-        self,
-        query: str,
-        limit: Optional[int] = None,
-    ) -> str:
-        """
-        Search the Agent's memory for past conversations and information.
-
-        This performs semantic search across all stored memories to find
-        relevant information based on the provided query.
-
-        Args:
-            query: What to search for in memory (e.g., "past conversations about AI", "user preferences")
-            limit: Maximum number of results to return (optional)
-
-        Returns:
-            str: JSON-encoded search results or error message
-
-        Example:
-            search_memory("user's favorite programming languages")
-            search_memory("previous discussions about machine learning")
-        """
-        try:
-            if not query.strip():
-                return json.dumps({"error": "Please provide a search query"})
-
-            log_debug(f"Searching memory for: {query}")
-
-            # Execute search using Memori's memory tool
-            result = self._memory_tool.execute(query=query.strip())
-
-            if result:
-                # If limit is specified, truncate results
-                if limit and isinstance(result, list):
-                    result = result[:limit]
-
-                return json.dumps(
-                    {
-                        "success": True,
-                        "query": query,
-                        "results": result,
-                        "count": len(result) if isinstance(result, list) else 1,
-                    }
-                )
-            else:
-                return json.dumps(
-                    {
-                        "success": True,
-                        "query": query,
-                        "results": [],
-                        "count": 0,
-                        "message": "No relevant memories found",
-                    }
-                )
-
-        except Exception as e:
-            log_error(f"Error searching memory: {e}")
-            return json.dumps({"success": False, "error": f"Memory search error: {str(e)}"})
-
-    def record_conversation(self, content: str) -> str:
-        """
-        Add important information or facts to memory.
-
-        Use this tool to store important information, user preferences, facts, or context that should be remembered
-        for future conversations.
-
-        Args:
-            content: The information/facts to store in memory
-
-        Returns:
-            str: Success message or error details
-
-        Example:
-            record_conversation("User prefers Python over JavaScript")
-            record_conversation("User is working on an e-commerce project using Django")
-            record_conversation("User's name is John and they live in NYC")
-        """
-        try:
-            if not content.strip():
-                return json.dumps({"success": False, "error": "Content cannot be empty"})
-
-            log_debug(f"Adding conversation: {content}")
-
-            # Extract the actual AI response from the agent's conversation history
-            ai_output = "I've noted this information and will remember it."
-
-            self.memory_system.record_conversation(user_input=content, ai_output=str(ai_output))
-            return json.dumps(
-                {
-                    "success": True,
-                    "message": "Memory added successfully via conversation recording",
-                    "content_length": len(content),
-                }
-            )
-
-        except Exception as e:
-            log_error(f"Error adding memory: {e}")
-            return json.dumps({"success": False, "error": f"Failed to add memory: {str(e)}"})
-
-    def get_memory_stats(
-        self,
-    ) -> str:
-        """
-        Get statistics about the memory system.
-
-        Returns information about the current state of the memory system,
-        including total memories, memory distribution by retention type
-        (short-term vs long-term), and system configuration.
-
-        Returns:
-            str: JSON-encoded memory statistics
-
-        Example:
-            Returns statistics like:
-            {
-                "success": true,
-                "total_memories": 42,
-                "memories_by_retention": {
-                    "short_term": 5,
-                    "long_term": 37
-                },
-                "namespace": "my_agent",
-                "conscious_ingest": true,
-                "auto_ingest": true,
-                "memory_system_enabled": true
-            }
-        """
-        try:
-            log_debug("Retrieving memory statistics")
-
-            # Base stats about the system configuration
-            stats = {
-                "success": True,
-                "namespace": self.namespace,
-                "database_connect": self.database_connect,
-                "conscious_ingest": self.conscious_ingest,
-                "auto_ingest": self.auto_ingest,
-                "verbose": self.verbose,
-                "memory_system_enabled": hasattr(self.memory_system, "_enabled") and self.memory_system._enabled,
-            }
-
-            # Get Memori's built-in memory statistics
-            try:
-                if hasattr(self.memory_system, "get_memory_stats"):
-                    # Use the get_memory_stats method as shown in the example
-                    memori_stats = self.memory_system.get_memory_stats()
-
-                    # Add the Memori-specific stats to our response
-                    if isinstance(memori_stats, dict):
-                        # Include total memories
-                        if "total_memories" in memori_stats:
-                            stats["total_memories"] = memori_stats["total_memories"]
-
-                        # Include memory distribution by retention type
-                        if "memories_by_retention" in memori_stats:
-                            stats["memories_by_retention"] = memori_stats["memories_by_retention"]
-
-                            # Also add individual counts for convenience
-                            retention_info = memori_stats["memories_by_retention"]
-                            stats["short_term_memories"] = retention_info.get("short_term", 0)
-                            stats["long_term_memories"] = retention_info.get("long_term", 0)
-
-                        # Include any other available stats
-                        for key, value in memori_stats.items():
-                            if key not in stats:
-                                stats[key] = value
-
-                    log_debug(
-                        f"Retrieved memory stats: total={stats.get('total_memories', 0)}, "
-                        f"short_term={stats.get('short_term_memories', 0)}, "
-                        f"long_term={stats.get('long_term_memories', 0)}"
-                    )
-
-                else:
-                    log_debug("get_memory_stats method not available, providing basic stats only")
-                    stats["total_memories"] = 0
-                    stats["memories_by_retention"] = {"short_term": 0, "long_term": 0}
-                    stats["short_term_memories"] = 0
-                    stats["long_term_memories"] = 0
-
-            except Exception as e:
-                log_debug(f"Could not retrieve detailed memory stats: {e}")
-                # Provide basic stats if detailed stats fail
-                stats["total_memories"] = 0
-                stats["memories_by_retention"] = {"short_term": 0, "long_term": 0}
-                stats["short_term_memories"] = 0
-                stats["long_term_memories"] = 0
-                stats["stats_warning"] = "Detailed memory statistics not available"
-
-            return json.dumps(stats)
-
-        except Exception as e:
-            log_error(f"Error getting memory stats: {e}")
-            return json.dumps({"success": False, "error": f"Failed to get memory statistics: {str(e)}"})
-
-    def enable_memory_system(self) -> bool:
-        """Enable the Memori memory system."""
-        try:
-            self.memory_system.enable()
-            log_debug("Memori memory system enabled")
-            return True
-        except Exception as e:
-            log_error(f"Failed to enable memory system: {e}")
-            return False
-
-    def disable_memory_system(self) -> bool:
-        """Disable the Memori memory system."""
-        try:
-            if hasattr(self.memory_system, "disable"):
-                self.memory_system.disable()
-                log_debug("Memori memory system disabled")
-                return True
-            else:
-                log_warning("Memory system disable method not available")
-                return False
-        except Exception as e:
-            log_error(f"Failed to disable memory system: {e}")
-            return False