NathanFradet commited on
Commit
acaf70e
·
verified ·
1 Parent(s): a905382

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -602,11 +602,9 @@ response = client.chat.completions.create(
602
  result = response.choices[0].message.content
603
 
604
  if "</think>" in result:
605
- reasoning = result.split("<think>")[1].split("</think>")[0]
606
- answer = result.split("</think>")[-1].strip()
607
  else:
608
- reasoning = None
609
- answer = result
610
 
611
  print(answer)
612
  ```
 
602
  result = response.choices[0].message.content
603
 
604
  if "</think>" in result:
605
+ reasoning, answer = [part.strip() for part in result.split("</think>")]
 
606
  else:
607
+ reasoning, answer = None, result
 
608
 
609
  print(answer)
610
  ```