Fixing a bug with Reasoning in the (g4f/Provider/Blackbox.py) provider

g4f/Provider/Blackbox.py

@@ -297,6 +297,7 @@ async def create_async_generator(
                 await raise_for_status(response)
                 response_text = await response.text()
                 parts = response_text.split('$~~~$')
+                print(parts)
                 text_to_yield = parts[2] if len(parts) >= 3 else response_text
 
                 if not text_to_yield or text_to_yield.isspace():
@@ -311,9 +312,17 @@ async def create_async_generator(
                         prompt = messages[-1]["content"]
                         yield ImageResponse(images=[image_url], alt=prompt)
                 else:
-                    if "<think>" in text_to_yield and "</think>" in text_to_yield:
-                        pre_think, rest = text_to_yield.split('<think>', 1)
-                        think_content, post_think = rest.split('</think>', 1)
+                    if "\n\n\n" in text_to_yield:
+                        think_split = text_to_yield.split("\n\n\n", 1)
+                        if len(think_split) > 1:
+                            think_content, answer = think_split[0].strip(), think_split[1].strip()
+                            yield Reasoning(status=think_content)
+                            yield answer
+                            full_response = f"{think_content}\n\n{answer}"
+                        else:
+                            yield text_to_yield
+                            full_response = text_to_yield
+                    elif "<think>" in text_to_yield:
 
                         pre_think = pre_think.strip()
                         think_content = think_content.strip()