Skip to content

Commit de8ed61

Browse files
fix repr md
1 parent c76e071 commit de8ed61

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lisette/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
def _repr_markdown_(self: litellm.ModelResponse):
2323
message = self.choices[0].message
2424
content = ''
25-
if message.content: content += message.content
25+
if mc:=message.content: content += mc[0]['text'] if isinstance(mc,list) else mc
2626
if message.tool_calls:
2727
tool_calls = [f"\n\n🔧 {nested_idx(tc,'function','name')}({nested_idx(tc,'function','arguments')})\n" for tc in message.tool_calls]
2828
content += "\n".join(tool_calls)

nbs/00_core.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@
107107
"def _repr_markdown_(self: litellm.ModelResponse):\n",
108108
" message = self.choices[0].message\n",
109109
" content = ''\n",
110-
" if message.content: content += message.content\n",
110+
" if mc:=message.content: content += mc[0]['text'] if isinstance(mc,list) else mc\n",
111111
" if message.tool_calls:\n",
112112
" tool_calls = [f\"\\n\\n🔧 {nested_idx(tc,'function','name')}({nested_idx(tc,'function','arguments')})\\n\" for tc in message.tool_calls]\n",
113113
" content += \"\\n\".join(tool_calls)\n",

0 commit comments

Comments
 (0)