Delete app

app/api/auth.py

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

app/api/config.py

@@ -1,73 +0,0 @@
-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 = "http://216.24.57.252"
-    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': 'http://216.24.57.252',
-        'referer': 'http://216.24.57.252/',
-        'host': '216.24.57.252',
-        'connection': 'keep-alive',
-        'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "Google Chrome";v="120"',
-        '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/120.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

@@ -1,24 +0,0 @@
-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

@@ -1,14 +0,0 @@
-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

@@ -1,24 +0,0 @@
-from fastapi import APIRouter, Depends
-from fastapi.responses import StreamingResponse
-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 StreamingResponse(
-            process_streaming_response(request),
-            media_type="text/event-stream"
-        )
-    else:
-        return await process_non_streaming_response(request) 

app/api/utils.py

@@ -1,187 +0,0 @@
-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):
-    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": settings.MODEL_MAPPING.get(request.model),
-        "validated": validate.getVid(),
-        "webSearchModePrompt": True if request.model.endswith("-search") else False
-    }
-
-    try:
-        async with httpx.AsyncClient(verify=False) as client:
-            async with client.stream(
-                method="POST",
-                url=f"{settings.BASE_URL}/api/chat",
-                headers=settings.HEADERS,
-                json=json_data
-            ) as response:
-                response.raise_for_status()
-                timestamp = int(datetime.now().timestamp())
-                search_results = None
-
-                async for line in response.aiter_lines():
-                    if not line:
-                        continue
-                    
-                    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 "http://216.24.57.252" 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"
-    except httpx.HTTPStatusError as e:
-        logger.error(f"HTTP error occurred: {str(e)}")
-        raise HTTPException(status_code=e.response.status_code, detail=str(e))
-    except Exception as e:
-        logger.error(f"Error occurred: {str(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 "http://216.24.57.252" 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

@@ -1,26 +0,0 @@
-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 
-
-base_url = "http://216.24.57.252" 

app/main.py

@@ -1,17 +0,0 @@
-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")