zrb 1.5.14__py3-none-any.whl → 1.5.15__py3-none-any.whl

zrb/task/llm/context_enrichment.py

@@ -13,11 +13,9 @@ from zrb.task.llm.typing import ListOfDict
 from zrb.util.attr import get_bool_attr, get_int_attr
 
 if TYPE_CHECKING:
-    from pydantic_ai import Agent
     from pydantic_ai.models import Model
     from pydantic_ai.settings import ModelSettings
 else:
-    Agent = Any
     Model = Any
     ModelSettings = Any
 
@@ -41,6 +39,8 @@ async def enrich_context(
     history_list: ListOfDict,
 ) -> dict[str, Any]:
     """Runs an LLM call to extract key info and merge it into the context."""
+    from pydantic_ai import Agent
+
     ctx.log_info("Attempting to enrich conversation context...")
     # Prepare context and history for the enrichment prompt
     try:
@@ -166,7 +166,6 @@ async def maybe_enrich_context(
         context_enrichment_threshold_attr,
         render_context_enrichment_threshold,
     ):
-        # Use the enrich_context function now defined in this file
         return await enrich_context(
             ctx=ctx,
             config=EnrichmentConfig(

{zrb-1.5.14.dist-info → zrb-1.5.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.5.14
+Version: 1.5.15
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.5.14.dist-info → zrb-1.5.15.dist-info}/RECORD

@@ -314,7 +314,7 @@ zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=pZAbn0vQOXFkJqnHBo8rfFq3OZD3yePr39jzcRqn43A,5248
 zrb/task/llm/config.py,sha256=R6mkbm4d5ecN4KjjZaXbqNq9-8bXfdUGl_BML8hUWqY,3205
 zrb/task/llm/context.py,sha256=s313jXooOp8ht7j2sc3-d8xyW7oWs9bFv4KEyjicwys,3893
-zrb/task/llm/context_enrichment.py,sha256=um-uI16HLUQz5GxPwHZefp9dw9x1VcFimv2Fqa9lsB8,6045
+zrb/task/llm/context_enrichment.py,sha256=lLPCtSSdB0jyReB3rk5taNvFPpXMmjDZ_Ro_Fz5GX68,5963
 zrb/task/llm/error.py,sha256=27DQXSG8SH1-XuvXFdZQKzP39wZDWmd_YnSTz6DJKKI,3690
 zrb/task/llm/history.py,sha256=LnrJdXLyo2qz-bNCwLorhoqGmgSiPTUU0bzY63w67-E,9257
 zrb/task/llm/history_summarization.py,sha256=0cWChp4OE_OiaNDhHRWi4rwHKTHsqWLYy3pS5IAIHpQ,6293
@@ -363,7 +363,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.5.14.dist-info/METADATA,sha256=pYeLSb24lcrPCcnBfinqz3kM0aEuGFxWv03Ad6Ci1V8,8347
-zrb-1.5.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.5.14.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.5.14.dist-info/RECORD,,
+zrb-1.5.15.dist-info/METADATA,sha256=G1B30COaisd9TjKv4-r0FpOwQnU-N5UBk5UEexfN2cQ,8347
+zrb-1.5.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.5.15.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.5.15.dist-info/RECORD,,