Update api/utils.py

api/utils.py

@@ -18,8 +18,7 @@ from api.config import (
     BASE_URL,
     AGENT_MODE,
     TRENDING_AGENT_MODE,
-    MODEL_PREFIXES,
-    MODEL_REFERERS
+    MODEL_PREFIXES
 )
 from api.models import ChatRequest
 from api.logger import setup_logger
@@ -109,9 +108,14 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
         return content[len(model_prefix):].strip()
     return content
 
+# Simplified function to get the base referer URL
+def get_referer_url() -> str:
+    """Return the base URL for the referer without model-specific logic."""
+    return BASE_URL
+
 # Process streaming response with headers from config.py
 async def process_streaming_response(request: ChatRequest, request_obj: Request):
-    referer_url = get_referer_url(request.model)
+    referer_url = get_referer_url()  # Get the base URL without model-specific logic
     logger.info(f"Processing streaming response - Model: {request.model} - URL: {referer_url}")
 
     # Get the IP address and check rate limit
@@ -184,7 +188,7 @@ async def process_streaming_response(request: ChatRequest, request_obj: Request)
 
 # Process non-streaming response with headers from config.py
 async def process_non_streaming_response(request: ChatRequest, request_obj: Request):
-    referer_url = get_referer_url(request.model)
+    referer_url = get_referer_url()  # Get the base URL without model-specific logic
     logger.info(f"Processing non-streaming response - Model: {request.model} - URL: {referer_url}")
 
     # Get the IP address and check rate limit