Delete api

api/__pycache__/dummy.txt

@@ -1 +0,0 @@
-

api/app.py

@@ -1,40 +0,0 @@
-from fastapi import FastAPI, Request
-from starlette.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-from api.logger import setup_logger
-from api.routes import router
-
-logger = setup_logger(__name__)
-
-def create_app():
-    app = FastAPI(
-        title="NiansuhAI API Gateway",
-        docs_url=None,          # Disable Swagger UI
-        redoc_url=None,         # Disable ReDoc
-        openapi_url=None,       # Disable OpenAPI schema
-    )
-
-    # CORS settings
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=["*"],                   # Adjust as needed for security
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    # Include routes
-    app.include_router(router)
-
-    # Global exception handler for better error reporting
-    @app.exception_handler(Exception)
-    async def global_exception_handler(request: Request, exc: Exception):
-        logger.error(f"An error occurred: {str(exc)}")
-        return JSONResponse(
-            status_code=500,
-            content={"message": "An internal server error occurred."},
-        )
-
-    return app
-
-app = create_app()

api/auth.py

@@ -1,10 +0,0 @@
-from fastapi import Depends, HTTPException
-from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
-from api.config import APP_SECRET
-
-security = HTTPBearer()
-
-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

api/config.py

@@ -1,58 +0,0 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-
-# DDG API Configurations
-DDG_API_ENDPOINT = "https://duckduckgo.com/duckchat/v1/chat"
-DDG_STATUS_URL = "https://duckduckgo.com/duckchat/v1/status"
-
-APP_SECRET = os.getenv("APP_SECRET")
-
-# Allowed Models for DDG
-ALLOWED_MODELS = [
-    {"id": "gpt-4o-mini", "name": "GPT-4o Mini"},
-    {"id": "claude-3-haiku-20240307", "name": "Claude 3 Haiku"},
-    {"id": "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo", "name": "Meta LLaMA 3.1-70B Instruct Turbo"},
-    {"id": "mistralai/Mixtral-8x7B-Instruct-v0.1", "name": "Mistral Mixtral 8x7B Instruct v0.1"},
-    # Add more DDG models as needed
-]
-
-MODEL_MAPPING = {
-    "gpt-4o-mini": "gpt-4o-mini",
-    "claude-3-haiku": "claude-3-haiku-20240307",
-    "llama-3.1-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
-    "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
-    # Add more mappings as needed
-}
-
-# AGENT_MODE, TRENDING_AGENT_MODE, MODEL_PREFIXES, MODEL_REFERERS for DDG models
-AGENT_MODE = {
-    'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
-    'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "Niansuh"},
-    # Add DDG-specific agent modes if required
-}
-
-TRENDING_AGENT_MODE = {
-    "gpt-4o-mini": {},
-    "claude-3-haiku-20240307": {},
-    "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo": {},
-    "mistralai/Mixtral-8x7B-Instruct-v0.1": {},
-    # Add more DDG-specific trending agent modes if required
-}
-
-MODEL_PREFIXES = {
-    'gpt-4o-mini': '@GPT-4o Mini',
-    'claude-3-haiku-20240307': '@Claude-3-Haiku',
-    'meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo': '@Meta-LLaMA-3.1-70B',
-    'mistralai/Mixtral-8x7B-Instruct-v0.1': '@Mistral-Mixtral-8x7B',
-    # Add more DDG-specific prefixes as needed
-}
-
-MODEL_REFERERS = {
-    "gpt-4o-mini": "/?model=gpt-4o-mini",
-    "claude-3-haiku-20240307": "/?model=claude-3-haiku-20240307",
-    "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo": "/?model=meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
-    "mistralai/Mixtral-8x7B-Instruct-v0.1": "/?model=mistralai/Mixtral-8x7B-Instruct-v0.1",
-    # Add more DDG-specific referers as needed
-}

api/logger.py

@@ -1,20 +0,0 @@
-import logging
-
-def setup_logger(name):
-    logger = logging.getLogger(name)
-    if not logger.handlers:
-        logger.setLevel(logging.INFO)
-        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-        
-        # Console handler
-        console_handler = logging.StreamHandler()
-        console_handler.setFormatter(formatter)
-        logger.addHandler(console_handler)
-        
-        # File Handler - Error Level
-        # error_file_handler = logging.FileHandler('error.log')
-        # error_file_handler.setFormatter(formatter)
-        # error_file_handler.setLevel(logging.ERROR)
-        # logger.addHandler(error_file_handler)
-    
-    return logger

api/models.py

@@ -1,14 +0,0 @@
-from typing import List, Optional
-from pydantic import BaseModel
-
-class Message(BaseModel):
-    role: str
-    content: str | list
-
-class ChatRequest(BaseModel):
-    model: str
-    messages: List[Message]
-    stream: Optional[bool] = False
-    temperature: Optional[float] = 0.7
-    top_p: Optional[float] = 0.9
-    max_tokens: Optional[int] = 99999999

api/routes.py

@@ -1,59 +0,0 @@
-import json
-from fastapi import APIRouter, Depends, HTTPException, Request, Response
-from fastapi.responses import StreamingResponse
-from api.auth import verify_app_secret
-from api.config import ALLOWED_MODELS
-from api.models import ChatRequest
-from api.utils import process_ddg_non_streaming_response, process_ddg_streaming_response
-from api.logger import setup_logger
-
-logger = setup_logger(__name__)
-
-router = APIRouter()
-
-@router.options("/v1/chat/completions")
-@router.options("/api/v1/chat/completions")
-async def chat_completions_options():
-    return Response(
-        status_code=200,
-        headers={
-            "Access-Control-Allow-Origin": "*",
-            "Access-Control-Allow-Methods": "POST, OPTIONS",
-            "Access-Control-Allow-Headers": "Content-Type, Authorization",
-        },
-    )
-
-@router.get("/v1/models")
-@router.get("/api/v1/models")
-async def list_models():
-    return {"object": "list", "data": ALLOWED_MODELS}
-
-@router.post("/v1/chat/completions")
-@router.post("/api/v1/chat/completions")
-async def chat_completions(
-    request: ChatRequest, app_secret: str = Depends(verify_app_secret)
-):
-    logger.info("Entering chat_completions route")
-    logger.info(f"Processing chat completion request for model: {request.model}")
-
-    if request.model not in [model["id"] for model in ALLOWED_MODELS]:
-        raise HTTPException(
-            status_code=400,
-            detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(model['id'] for model in ALLOWED_MODELS)}",
-        )
-
-    if request.stream:
-        logger.info("Streaming response")
-        return StreamingResponse(process_ddg_streaming_response(request), media_type="text/event-stream")
-    else:
-        logger.info("Non-streaming response")
-        return await process_ddg_non_streaming_response(request)
-
-@router.route('/')
-@router.route('/healthz')
-@router.route('/ready')
-@router.route('/alive')
-@router.route('/status')
-@router.get("/health")
-def health_check(request: Request):
-    return Response(content=json.dumps({"status": "ok"}), media_type="application/json")

api/utils.py

@@ -1,250 +0,0 @@
-from datetime import datetime
-import json
-from typing import Any, Dict, Optional
-
-import httpx
-import aiohttp
-from api.config import (
-    DDG_API_ENDPOINT,
-    DDG_STATUS_URL,
-    MODEL_MAPPING,
-    ALLOWED_MODELS,
-    AGENT_MODE,
-    TRENDING_AGENT_MODE,
-    MODEL_PREFIXES,
-    MODEL_REFERERS
-)
-from fastapi import HTTPException
-from api.models import ChatRequest
-
-from api.logger import setup_logger
-
-import uuid
-
-logger = setup_logger(__name__)
-
-async def get_ddg_vqd():
-    status_url = DDG_STATUS_URL
-
-    headers = {
-        'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-        'Accept': 'text/event-stream',
-        'x-vqd-accept': '1'
-    }
-
-    async with aiohttp.ClientSession() as session:
-        try:
-            async with session.get(status_url, headers=headers) as response:
-                if response.status == 200:
-                    vqd = response.headers.get("x-vqd-4")
-                    if not vqd:
-                        logger.error("VQD token not found in response headers.")
-                    else:
-                        logger.debug(f"VQD token retrieved: {vqd}")
-                    return vqd
-                else:
-                    logger.error(f"Error: Status code {response.status} when fetching VQD")
-                    return None
-        except Exception as e:
-            logger.error(f"Error getting VQD: {e}")
-            return None
-
-def message_to_dict_ddg(message):
-    if isinstance(message.content, str):
-        return {"role": message.role, "content": message.content}
-    else:
-        # If DDG API does not support non-string content, raise an error
-        raise ValueError("Message content must be a string.")
-
-def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
-    """Remove the model prefix from the response content if present."""
-    if model_prefix and content.startswith(model_prefix):
-        logger.debug(f"Stripping prefix '{model_prefix}' from content.")
-        return content[len(model_prefix):].strip()
-    logger.debug("No prefix to strip from content.")
-    return content
-
-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,
-    }
-
-async def process_ddg_streaming_response(request: ChatRequest):
-    agent_mode = AGENT_MODE.get(request.model, {})
-    trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
-    model_prefix = MODEL_PREFIXES.get(request.model, "")
-    # referer_path and referer_url are not required unless specified by DDG API
-
-    # Obtain VQD token
-    vqd = await get_ddg_vqd()
-    if not vqd:
-        raise HTTPException(status_code=500, detail="Failed to obtain VQD token")
-
-    # Build headers
-    dynamic_headers = {
-        'accept': 'text/event-stream',
-        'content-type': 'application/json',
-        'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-        'x-vqd-4': vqd
-    }
-
-    # Construct message history
-    try:
-        message_history = [message_to_dict_ddg(msg) for msg in request.messages]
-    except ValueError as ve:
-        logger.error(f"Invalid message format: {ve}")
-        raise HTTPException(status_code=400, detail=str(ve))
-
-    json_data = {
-        "model": MODEL_MAPPING.get(request.model, request.model),
-        "messages": message_history,
-        "previewToken": None,
-        "userId": None,
-        "codeModelMode": True,
-        "agentMode": agent_mode,
-        "trendingAgentMode": trending_agent_mode,
-        "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, request.model),
-    }
-
-    logger.debug(f"Sending JSON payload to DDG API: {json.dumps(json_data)}")
-
-    async with httpx.AsyncClient() as client:
-        try:
-            async with client.stream(
-                "POST",
-                DDG_API_ENDPOINT,
-                headers=dynamic_headers,
-                json=json_data,
-                timeout=100,
-            ) as response:
-                response.raise_for_status()
-                async for line in response.aiter_lines():
-                    timestamp = int(datetime.now().timestamp())
-                    if line:
-                        content = line
-                        if content.startswith("$@$v=undefined-rv1$@$"):
-                            content = content[21:]
-                        # Strip the model prefix from the response content
-                        cleaned_content = strip_model_prefix(content, model_prefix)
-                        yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, 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_ddg_non_streaming_response(request: ChatRequest):
-    agent_mode = AGENT_MODE.get(request.model, {})
-    trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
-    model_prefix = MODEL_PREFIXES.get(request.model, "")
-
-    # Obtain VQD token
-    vqd = await get_ddg_vqd()
-    if not vqd:
-        raise HTTPException(status_code=500, detail="Failed to obtain VQD token")
-
-    # Build headers
-    dynamic_headers = {
-        'accept': 'application/json',
-        'content-type': 'application/json',
-        'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-        'x-vqd-4': vqd
-    }
-
-    # Construct message history
-    try:
-        message_history = [message_to_dict_ddg(msg) for msg in request.messages]
-    except ValueError as ve:
-        logger.error(f"Invalid message format: {ve}")
-        raise HTTPException(status_code=400, detail=str(ve))
-
-    json_data = {
-        "model": MODEL_MAPPING.get(request.model, request.model),
-        "messages": message_history,
-        "previewToken": None,
-        "userId": None,
-        "codeModelMode": True,
-        "agentMode": agent_mode,
-        "trendingAgentMode": trending_agent_mode,
-        "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, request.model),
-    }
-
-    logger.debug(f"Sending JSON payload to DDG API: {json.dumps(json_data)}")
-
-    async with httpx.AsyncClient() as client:
-        try:
-            response = await client.post(
-                DDG_API_ENDPOINT,
-                headers=dynamic_headers,
-                json=json_data,
-                timeout=100
-            )
-            response.raise_for_status()
-            full_response = response.text
-        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))
-
-    if full_response.startswith("$@$v=undefined-rv1$@$"):
-        full_response = full_response[21:]
-
-    cleaned_full_response = strip_model_prefix(full_response, model_prefix)
-
-    return {
-        "id": f"chatcmpl-{uuid.uuid4()}",
-        "object": "chat.completion",
-        "created": int(datetime.now().timestamp()),
-        "model": request.model,
-        "choices": [
-            {
-                "index": 0,
-                "message": {"role": "assistant", "content": cleaned_full_response},
-                "finish_reason": "stop",
-            }
-        ],
-        "usage": None,
-    }