Update api/utils.py

api/utils.py

@@ -12,7 +12,6 @@ from fastapi.security import HTTPAuthorizationCredentials
 
 from api.config import APP_SECRET, BASE_URL
 from api.models import ChatRequest
-
 from api.logger import setup_logger
 
 logger = setup_logger(__name__)
@@ -60,7 +59,6 @@ async def process_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
 
-    # Log reduced information
     logger.info(
         f"Streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
@@ -101,17 +99,7 @@ async def process_streaming_response(request: ChatRequest):
                 response.raise_for_status()
                 async for line in response.aiter_lines():
                     timestamp = int(datetime.now().timestamp())
-                    if line:
-                        content = line if line.endswith("\n") else line + "\n"
-                        if "https://www.blackbox.ai" in content:
-                            validate.getHid(True)
-                            content = "Hid has been refreshed; feel free to restart the conversation.\n"
-                            yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
-                            break
-                        if content.startswith("$@$v=undefined-rv1$@$"):
-                            yield f"data: {json.dumps(create_chat_completion_data(content[21:], request.model, timestamp))}\n\n"
-                        else:
-                            yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
+                    yield f"data: {json.dumps(create_chat_completion_data(line, request.model, timestamp))}\n\n"
 
                 yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
                 yield "data: [DONE]\n\n"
@@ -126,7 +114,6 @@ async def process_non_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
 
-    # Log reduced information
     logger.info(
         f"Non-streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
@@ -161,12 +148,8 @@ async def process_non_streaming_response(request: ChatRequest):
             method="POST", url=f"{BASE_URL}/api/chat", headers=headers, json=json_data
         ) as response:
             async for chunk in response.aiter_text():
-                full_response += chunk if chunk.endswith("\n") else chunk + "\n"
-    if "https://www.blackbox.ai" in full_response:
-        validate.getHid(True)
-        full_response = "Hid has been refreshed; feel free to restart the conversation."
-    if full_response.startswith("$@$v=undefined-rv1$@$"):
-        full_response = full_response[21:]
+                full_response += chunk
+
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion",