Niansuh commited on
Commit
40c87d6
·
verified ·
1 Parent(s): b0d426a

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +3 -4
api/utils.py CHANGED
@@ -4,9 +4,8 @@ import uuid
4
  import asyncio
5
  import random
6
  import string
7
- from typing import Any, Dict, Optional
8
  import re
9
-
10
  import httpx
11
  from fastapi import HTTPException
12
  from api.config import (
@@ -23,6 +22,7 @@ from api.models import ChatRequest
23
  from api.logger import setup_logger
24
  from api.validate import getHid # Import the asynchronous getHid function
25
 
 
26
  logger = setup_logger(__name__)
27
 
28
  # Define the blocked message
@@ -62,7 +62,6 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
62
  "imageBase64": image_base64,
63
  "fileText": "",
64
  "title": "snapshot",
65
- # Added imagesData field here
66
  "imagesData": [
67
  {
68
  "filePath": f"MultipleFiles/{uuid.uuid4().hex}.jpg",
@@ -87,7 +86,7 @@ def remove_message_between_special_tags(content: str) -> str:
87
  content = re.sub(r'\$~~~\$.*?\$~~~\$', '', content, flags=re.DOTALL)
88
  return content
89
 
90
- # Function to remove search results but keep the links
91
  def remove_search_results(content: str) -> str:
92
  """Remove search result sections, keeping the URLs intact."""
93
  # Regex to match search result JSON (structured with `title`, `snippet`, `link`)
 
4
  import asyncio
5
  import random
6
  import string
 
7
  import re
8
+ from typing import Any, Dict, Optional
9
  import httpx
10
  from fastapi import HTTPException
11
  from api.config import (
 
22
  from api.logger import setup_logger
23
  from api.validate import getHid # Import the asynchronous getHid function
24
 
25
+ # Setup logger
26
  logger = setup_logger(__name__)
27
 
28
  # Define the blocked message
 
62
  "imageBase64": image_base64,
63
  "fileText": "",
64
  "title": "snapshot",
 
65
  "imagesData": [
66
  {
67
  "filePath": f"MultipleFiles/{uuid.uuid4().hex}.jpg",
 
86
  content = re.sub(r'\$~~~\$.*?\$~~~\$', '', content, flags=re.DOTALL)
87
  return content
88
 
89
+ # Function to remove search results but keep the links intact
90
  def remove_search_results(content: str) -> str:
91
  """Remove search result sections, keeping the URLs intact."""
92
  # Regex to match search result JSON (structured with `title`, `snippet`, `link`)