Merge pull request #5 from AnswerDotAI/fix_arg

fix: remove if statement too eagerly

Author: bclavie

msglm/core.py

@@ -161,6 +161,6 @@ def mk_msg_anthropic(*args, cache=False, **kwargs):
 def mk_msgs_anthropic(*args, cache=False, cache_last_ckpt_only=False, **kwargs):
     "Create a list of Anthropic compatible messages."
     msgs = partial(mk_msgs, api="anthropic")(*args, **kwargs)
-    msgs = [_remove_cache_ckpts(m) for m in msgs]
+    if cache_last_ckpt_only: msgs = [_remove_cache_ckpts(m) for m in msgs]
     msgs[-1] = _add_cache_control(msgs[-1], cache=cache)
     return msgs

nbs/00_core.ipynb

@@ -1713,7 +1713,7 @@
     "def mk_msgs_anthropic(*args, cache=False, cache_last_ckpt_only=False, **kwargs):\n",
     "    \"Create a list of Anthropic compatible messages.\"\n",
     "    msgs = partial(mk_msgs, api=\"anthropic\")(*args, **kwargs)\n",
-    "    msgs = [_remove_cache_ckpts(m) for m in msgs]\n",
+    "    if cache_last_ckpt_only: msgs = [_remove_cache_ckpts(m) for m in msgs]\n",
     "    msgs[-1] = _add_cache_control(msgs[-1], cache=cache)\n",
     "    return msgs"
    ]