Update proxy_handler.py

proxy_handler.py

@@ -26,7 +26,7 @@ class ProxyHandler:
         if not self.client.is_closed:
             await self.client.aclose()
     
-def _clean_thinking_content(self, text: str) -> str:
+    def _clean_thinking_content(self, text: str) -> str:
         """
         Aggressively cleans raw thinking content strings.
         Removes tool calls, HTML-like tags, and other metadata.
@@ -70,6 +70,27 @@ def _clean_thinking_content(self, text: str) -> str:
         # Final trim
         return cleaned_text.strip()
 
+    def _clean_answer_content(self, text: str) -> str:
+        """
+        Cleans only <glm_block> tags from answer content.
+        Does NOT strip whitespace to preserve markdown in streams.
+        """
+        if not text:
+            return ""
+        # Remove only tool call blocks
+        cleaned_text = re.sub(r'<glm_block.*?</glm_block>', '', text, flags=re.DOTALL)
+        return cleaned_text
+
+    def _serialize_msgs(self, msgs) -> list:
+        """Converts message objects to a list of dictionaries."""
+        out = []
+        for m in msgs:
+            if hasattr(m, "dict"): out.append(m.dict())
+            elif hasattr(m, "model_dump"): out.append(m.model_dump())
+            elif isinstance(m, dict): out.append(m)
+            else: out.append({"role": getattr(m, "role", "user"), "content": getattr(m, "content", str(m))})
+        return out
+
     async def _prep_upstream(self, req: ChatCompletionRequest) -> Tuple[Dict[str, Any], Dict[str, str], str]:
         """Prepares the request body, headers, and cookie for the upstream API."""
         ck = await cookie_manager.get_next_cookie()