Upload 10 files

Dockerfile

@@ -1,8 +1,13 @@
-FROM hpyp/bbapi:latest
+FROM python:3.10-slim
 
-# 设置环境变量
-ENV NAMESERVER_1=8.8.8.8
-ENV NAMESERVER_2=8.8.4.4
+WORKDIR /app
+
+# 安装依赖
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 复制应用代码
+COPY app/ .
 
 EXPOSE 8001
 

app/api/auth.py

@@ -0,0 +1,5 @@
+import os
+from fastapi.security import HTTPBearer
+
+security = HTTPBearer()
+APP_SECRET = os.getenv("APP_SECRET", "sk-zhoudan20241029") 

app/api/config.py

@@ -0,0 +1,71 @@
+import os
+from pydantic_settings import BaseSettings
+from dotenv import load_dotenv
+from typing import Dict, List
+
+load_dotenv()
+
+class Settings(BaseSettings):
+    # Server settings
+    HOST: str = os.getenv("HOST", "0.0.0.0")
+    PORT: int = int(os.getenv("PORT", "8001"))
+    DEBUG: bool = os.getenv("DEBUG", "False").lower() == "true"
+    WORKERS: int = int(os.getenv("WORKERS", "1"))
+    LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
+
+    # API settings
+    BASE_URL: str = "https://www.blackbox.ai"
+    APP_SECRET: str = os.getenv("APP_SECRET", "")
+    REQUEST_TIMEOUT: int = int(os.getenv("REQUEST_TIMEOUT", "30"))
+
+    # Headers
+    HEADERS: Dict[str, str] = {
+        'accept': '*/*',
+        'accept-language': 'zh-CN,zh;q=0.9',
+        'content-type': 'application/json',
+        'origin': 'https://www.blackbox.ai',
+        'priority': 'u=1, i',
+        'sec-ch-ua': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
+        'sec-ch-ua-mobile': '?0',
+        'sec-ch-ua-platform': '"Windows"',
+        'sec-fetch-dest': 'empty',
+        'sec-fetch-mode': 'cors',
+        'sec-fetch-site': 'same-origin',
+        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36',
+    }
+
+    ALLOWED_MODELS: List[Dict[str, str]] = [
+        {"id": "gpt-4o", "name": "gpt-4o"},
+        {"id": "gemini-1.5-pro", "name": "gemini-pro"},
+        {"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
+        {"id": "blackboxai", "name": "blackboxai"},
+        {"id": "blackboxai-pro", "name": "blackboxai-pro"},
+        {"id": "blackboxai-search", "name": "blackboxai-search"},
+        {"id": "meta-llama/Llama-3.3-70B-Instruct-Turbo", "name": "meta-llama/Llama-3.3-70B-Instruct-Turbo"},
+        {"id": "meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro", "name": "meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro"},
+        {"id": "Qwen/QwQ-32B-Preview", "name": "Qwen/QwQ-32B-Preview"},
+    ]
+
+    MODEL_MAPPING: Dict[str, str] = {
+        "gpt-4o": "gpt-4o",
+        "gemini-1.5-pro": "gemini-pro",
+        "claude-3-5-sonnet": "claude-sonnet-3.5",
+        "blackboxai": "blackboxai",
+        "blackboxai-pro": "blackboxai-pro",
+        "blackboxai-search": "blackboxai-search",
+        "meta-llama/Llama-3.3-70B-Instruct-Turbo": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
+        "meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro": "meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro",
+        "Qwen/QwQ-32B-Preview": "Qwen/QwQ-32B-Preview",
+    }
+
+    class Config:
+        env_file = ".env"
+        case_sensitive = True
+
+_settings = None
+
+def get_settings() -> Settings:
+    global _settings
+    if _settings is None:
+        _settings = Settings()
+    return _settings 

app/api/logger.py

@@ -0,0 +1,24 @@
+import logging
+from logging.handlers import RotatingFileHandler
+import os
+
+def setup_logger(name):
+    logger = logging.getLogger(name)
+    
+    if not logger.handlers:
+        logger.setLevel(logging.INFO)
+        
+        # 创建控制台处理器
+        console_handler = logging.StreamHandler()
+        console_handler.setLevel(logging.INFO)
+        
+        # 设置日志格式
+        formatter = logging.Formatter(
+            '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+        )
+        console_handler.setFormatter(formatter)
+        
+        # 添加处理器到logger
+        logger.addHandler(console_handler)
+        
+    return logger 

app/api/models.py

@@ -0,0 +1,14 @@
+from typing import List, Optional, Union
+from pydantic import BaseModel, Field
+
+class Message(BaseModel):
+    role: str
+    content: Union[str, List]
+
+class ChatRequest(BaseModel):
+    model: str
+    messages: List[Message]
+    stream: bool = False
+    temperature: float = Field(default=0.7, ge=0, le=2)
+    top_p: float = Field(default=0.95, ge=0, le=1)
+    max_tokens: int = Field(default=8192, ge=1) 

app/api/routes.py

@@ -0,0 +1,20 @@
+from fastapi import APIRouter, Depends
+from api.models import ChatRequest
+from api.utils import process_streaming_response, process_non_streaming_response
+from api.logger import setup_logger
+
+logger = setup_logger(__name__)
+router = APIRouter()
+
+@router.post("/chat/completions")
+async def chat_completions(request: ChatRequest):
+    """
+    处理聊天完成请求
+    """
+    logger.info("Entering chat_completions route")
+    logger.info(f"Received request: {request}")
+    
+    if request.stream:
+        return process_streaming_response(request)
+    else:
+        return await process_non_streaming_response(request) 

app/api/utils.py

@@ -0,0 +1,200 @@
+from datetime import datetime
+from http.client import HTTPException
+import json
+from typing import Any, Dict, Optional
+import uuid
+
+import httpx
+from api import validate
+from api.auth import APP_SECRET
+from api.config import get_settings
+from fastapi import Depends, security
+from fastapi.security import HTTPAuthorizationCredentials
+
+from api.models import ChatRequest
+
+from api.logger import setup_logger
+
+logger = setup_logger(__name__)
+
+settings = get_settings()
+BASE_URL = settings.BASE_URL
+MODEL_MAPPING = settings.MODEL_MAPPING
+
+def create_chat_completion_data(
+    content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
+) -> Dict[str, Any]:
+    return {
+        "id": f"chatcmpl-{uuid.uuid4()}",
+        "object": "chat.completion.chunk",
+        "created": timestamp,
+        "model": model,
+        "choices": [
+            {
+                "index": 0,
+                "delta": {"content": content, "role": "assistant"},
+                "finish_reason": finish_reason,
+            }
+        ],
+        "usage": None,
+    }
+
+def verify_app_secret(credentials: HTTPAuthorizationCredentials = Depends(security)):
+    if credentials.credentials != APP_SECRET:
+        raise HTTPException(status_code=403, detail="Invalid APP_SECRET")
+    return credentials.credentials
+
+def message_to_dict(message):
+    if isinstance(message.content, str):
+        return {"role": message.role, "content": message.content}
+    elif isinstance(message.content, list) and len(message.content) == 2:
+        return {
+            "role": message.role,
+            "content": message.content[0]["text"],
+            "data": {
+                "imageBase64": message.content[1]["image_url"]["url"],
+                "fileText": "",
+                "title": "snapshoot",
+            },
+        }
+    else:
+        return {"role": message.role, "content": message.content}
+
+async def process_streaming_response(request: ChatRequest):
+    search_results = []  # 初始化搜索结果数组
+
+    json_data = {
+        "messages": [message_to_dict(msg) for msg in request.messages],
+        "previewToken": None,
+        "userId": None,
+        "codeModelMode": True,
+        "agentMode": {},
+        "trendingAgentMode": {},
+        "isMicMode": False,
+        "userSystemPrompt": None,
+        "maxTokens": request.max_tokens,
+        "playgroundTopP": request.top_p,
+        "playgroundTemperature": request.temperature,
+        "isChromeExt": False,
+        "githubToken": None,
+        "clickedAnswer2": False,
+        "clickedAnswer3": False,
+        "clickedForceWebSearch": False,
+        "visitFromDelta": False,
+        "mobileClient": False,
+        "userSelectedModel": MODEL_MAPPING.get(request.model),
+        "validated": validate.getVid(),
+        "webSearchModePrompt": True if request.model.endswith("-search") else False
+    }
+
+    async with httpx.AsyncClient(verify=False) as client:
+        try:
+            async with client.stream(
+                "POST",
+                f"{BASE_URL}/api/chat",
+                headers=settings.HEADERS,
+                json=json_data,
+                timeout=100,
+            ) as response:
+                response.raise_for_status()
+                timestamp = int(datetime.now().timestamp())
+                async for line in response.aiter_lines():
+                    if line:
+                        if line.startswith("$~~~$"):  # 处理搜索结果
+                            try:
+                                json_str = line[5:-5]  # 提取中间的JSON字符串
+                                results = json.loads(json_str)
+                                search_results = results  # 保存搜索结果
+                                continue
+                            except json.JSONDecodeError:
+                                logger.error("Failed to parse search results")
+                                continue
+
+                        if line == "**":
+                            continue
+                        content = line + "\n"
+                        print(content)
+                        if "https://www.blackbox.ai" in content:
+                            validate.getVid(True)
+                            content = "vid已刷新,重新对话即可\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"
+                    elif line == "":
+                        content = line + "\n\n"
+                        print(content)
+                        if "https://www.blackbox.ai" in content:
+                            validate.getVid(True)
+                            content = "vid已刷新,重新对话即可\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"
+                if search_results:  # 在结束前输出引用来源
+                    sources = "\n\n**引用来源**\n" + "\n".join([f"- [{r['title']}]({r['link']})" for r in search_results])
+                    yield f"data: {json.dumps(create_chat_completion_data(sources, 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"
+        except httpx.HTTPStatusError as e:
+            logger.error(f"HTTP error occurred: {e}")
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except httpx.RequestError as e:
+            logger.error(f"Error occurred during request: {e}")
+            raise HTTPException(status_code=500, detail=str(e))
+
+async def process_non_streaming_response(request: ChatRequest):
+    json_data = {
+        "messages": [message_to_dict(msg) for msg in request.messages],
+        "previewToken": None,
+        "userId": None,
+        "codeModelMode": True,
+        "agentMode": {},
+        "trendingAgentMode": {},
+        "isMicMode": False,
+        "userSystemPrompt": None,
+        "maxTokens": request.max_tokens,
+        "playgroundTopP": request.top_p,
+        "playgroundTemperature": request.temperature,
+        "isChromeExt": False,
+        "githubToken": None,
+        "clickedAnswer2": False,
+        "clickedAnswer3": False,
+        "clickedForceWebSearch": False,
+        "visitFromDelta": False,
+        "mobileClient": False,
+        "userSelectedModel": MODEL_MAPPING.get(request.model),
+        "validated": validate.getVid(),
+        "webSearchModePrompt": True if request.model.endswith("-search") else False
+    }
+    full_response = ""
+    async with httpx.AsyncClient(verify=False) as client:
+        async with client.stream(
+            method="POST", url=f"{BASE_URL}/api/chat", headers=settings.HEADERS, json=json_data
+        ) as response:
+            async for chunk in response.aiter_text():
+                full_response += chunk
+    if "https://www.blackbox.ai" in full_response:
+        validate.getVid(True)
+        full_response = "vid已刷新，重新对话即可"
+    if full_response.startswith("$@$v=undefined-rv1$@$"):
+        full_response = full_response[21:]
+    return {
+        "id": f"chatcmpl-{uuid.uuid4()}",
+        "object": "chat.completion",
+        "created": int(datetime.now().timestamp()),
+        "model": request.model,
+        "choices": [
+            {
+                "index": 0,
+                "message": {"role": "assistant", "content": full_response},
+                "finish_reason": "stop",
+            }
+        ],
+        "usage": None,
+    } 

app/api/validate.py

@@ -0,0 +1,24 @@
+import time
+import random
+import string
+
+_vid = None
+_last_update = 0
+
+def generate_vid():
+    """生成随机vid"""
+    length = 32
+    characters = string.ascii_letters + string.digits
+    return ''.join(random.choice(characters) for _ in range(length))
+
+def getVid(force_refresh=False):
+    """获取或刷新vid"""
+    global _vid, _last_update
+    current_time = time.time()
+    
+    # 如果强制刷新或者vid不存在或者已过期(1小时)
+    if force_refresh or _vid is None or (current_time - _last_update) > 3600:
+        _vid = generate_vid()
+        _last_update = current_time
+    
+    return _vid 

app/main.py

@@ -0,0 +1,17 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from api.routes import router
+
+app = FastAPI()
+
+# 配置CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 添加路由
+app.include_router(router, prefix="/api/v1") 

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.109.0
+uvicorn==0.27.0
+httpx==0.26.0
+python-dotenv==1.0.0
+pydantic-settings==2.1.0
+python-multipart==0.0.6