Fix a few bugs

AnswerDotAI · Sep 30, 2024 · 238d982 · 238d982
1 parent b107de6
commit 238d982
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/00_core.ipynb b/00_core.ipynb
@@ -1863,7 +1863,7 @@
     "             **kwargs):\n",
     "    \"Make a call to Claude.\"\n",
     "    if tools: kwargs['tools'] = [get_schema(o) for o in listify(tools)]\n",
-    "    if tool_choice and pr: kwargs['tool_choice'] = mk_tool_choice(tool_choice)\n",
+    "    if tool_choice: kwargs['tool_choice'] = mk_tool_choice(tool_choice)\n",
     "    msgs = self._precall(msgs, prefill, stop, kwargs)\n",
     "    if stream: return self._stream(msgs, prefill=prefill, max_tokens=maxtok, system=sp, temperature=temp, **kwargs)\n",
     "    res = self.c.messages.create(model=self.model, messages=msgs, max_tokens=maxtok, system=sp, temperature=temp, **kwargs)\n",
@@ -1953,7 +1953,7 @@
     "    \"Return the value of all tool calls (generally used for structured outputs)\"\n",
     "    res = self(msgs, **kwargs)\n",
     "    if ns is None: ns=globals()\n",
-    "    cts = getattr(r, 'content', [])\n",
+    "    cts = getattr(res, 'content', [])\n",
     "    tcs = [call_func(o, ns=ns, obj=obj) for o in cts if isinstance(o,ToolUseBlock)]\n",
     "    return tcs"
    ]

diff --git a/claudette/core.py b/claudette/core.py
@@ -211,7 +211,7 @@ def __call__(self:Client,
              **kwargs):
     "Make a call to Claude."
     if tools: kwargs['tools'] = [get_schema(o) for o in listify(tools)]
-    if tool_choice and pr: kwargs['tool_choice'] = mk_tool_choice(tool_choice)
+    if tool_choice: kwargs['tool_choice'] = mk_tool_choice(tool_choice)
     msgs = self._precall(msgs, prefill, stop, kwargs)
     if stream: return self._stream(msgs, prefill=prefill, max_tokens=maxtok, system=sp, temperature=temp, **kwargs)
     res = self.c.messages.create(model=self.model, messages=msgs, max_tokens=maxtok, system=sp, temperature=temp, **kwargs)
@@ -228,7 +228,7 @@ def structured(self:Client,
     "Return the value of all tool calls (generally used for structured outputs)"
     res = self(msgs, **kwargs)
     if ns is None: ns=globals()
-    cts = getattr(r, 'content', [])
+    cts = getattr(res, 'content', [])
     tcs = [call_func(o, ns=ns, obj=obj) for o in cts if isinstance(o,ToolUseBlock)]
     return tcs