mohit-sarvam commited on
Commit
1eb7a52
·
verified ·
1 Parent(s): 16e4c05

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -104,13 +104,13 @@ response = client.chat.completions.create(model=model, messages=messages)
104
  output_text = response.choices[0].message.content
105
 
106
  if "</think>" in output_text:
107
- thinking_content = output_text.split("</think>")[0].rstrip("\n")
108
  content = output_text.split("</think>")[-1].lstrip("\n").rstrip("</s>")
109
  else:
110
- thinking_content = ""
111
  content = output_text.rstrip("</s>")
112
 
113
- print("reasoning_content:", thinking_content)
114
  print("content:", content)
115
 
116
  # For the next round, add the assistant's response and reasoning to the messages.
 
104
  output_text = response.choices[0].message.content
105
 
106
  if "</think>" in output_text:
107
+ reasoning_content = output_text.split("</think>")[0].rstrip("\n")
108
  content = output_text.split("</think>")[-1].lstrip("\n").rstrip("</s>")
109
  else:
110
+ reasoning_content = ""
111
  content = output_text.rstrip("</s>")
112
 
113
+ print("reasoning_content:", reasoning_content)
114
  print("content:", content)
115
 
116
  # For the next round, add the assistant's response and reasoning to the messages.