Fix extra_body to disable thinking

#141
Files changed (1) hide show
  1. unit1/dummy_agent_library.ipynb +4 -4
unit1/dummy_agent_library.ipynb CHANGED
@@ -74,7 +74,7 @@
74
  " ],\n",
75
  " stream=False,\n",
76
  " max_tokens=1024,\n",
77
- " extra_body={'thinking': {'type': 'disabled'}},\n",
78
  ")\n",
79
  "print(output.choices[0].message.content)"
80
  ]
@@ -183,7 +183,7 @@
183
  " messages=messages,\n",
184
  " stream=False,\n",
185
  " max_tokens=200,\n",
186
- " extra_body={'thinking': {'type': 'disabled'}},\n",
187
  ")\n",
188
  "print(output.choices[0].message.content)"
189
  ]
@@ -210,7 +210,7 @@
210
  " messages=messages,\n",
211
  " max_tokens=150,\n",
212
  " stop=[\"Observation:\"], # Let's stop before any actual function is called\n",
213
- " extra_body={'thinking': {'type': 'disabled'}},\n",
214
  ")\n",
215
  "\n",
216
  "print(output.choices[0].message.content)"
@@ -261,7 +261,7 @@
261
  " messages=messages,\n",
262
  " stream=False,\n",
263
  " max_tokens=200,\n",
264
- " extra_body={'thinking': {'type': 'disabled'}},\n",
265
  ")\n",
266
  "\n",
267
  "print(output.choices[0].message.content)"
 
74
  " ],\n",
75
  " stream=False,\n",
76
  " max_tokens=1024,\n",
77
+ " extra_body={'chat_template_kwargs': {'enable_thinking': False}},\n",
78
  ")\n",
79
  "print(output.choices[0].message.content)"
80
  ]
 
183
  " messages=messages,\n",
184
  " stream=False,\n",
185
  " max_tokens=200,\n",
186
+ " extra_body={'chat_template_kwargs': {'enable_thinking': False}},\n",
187
  ")\n",
188
  "print(output.choices[0].message.content)"
189
  ]
 
210
  " messages=messages,\n",
211
  " max_tokens=150,\n",
212
  " stop=[\"Observation:\"], # Let's stop before any actual function is called\n",
213
+ " extra_body={'chat_template_kwargs': {'enable_thinking': False}},\n",
214
  ")\n",
215
  "\n",
216
  "print(output.choices[0].message.content)"
 
261
  " messages=messages,\n",
262
  " stream=False,\n",
263
  " max_tokens=200,\n",
264
+ " extra_body={'chat_template_kwargs': {'enable_thinking': False}},\n",
265
  ")\n",
266
  "\n",
267
  "print(output.choices[0].message.content)"