app.py

# uvicorn app:app --host 0.0.0.0 --port 7860 --reload

from fastapi import FastAPI, HTTPException, Response, Depends
from starlette.requests import Request

from routers.webtools_v1 import router as webtools_router
from routers.users_v1 import router as users_router
from routers.openai_v1 import router as openai_router

from db import Db

from dotenv import load_dotenv
import os

# 加载.env文件
load_dotenv()

DB_PATH = os.getenv("DB_PATH")

# 依赖注入，获取 DB_PATH
def get_db_path():
    return DB_PATH

app = FastAPI()

app.include_router(webtools_router, prefix="/airs/v1", tags=["webtools"])
app.include_router(users_router, prefix="/airs/v1", tags=["users"])
app.include_router(openai_router, prefix="/airs/v1", tags=["openai"])

@app.get("/")
def greet_json(db_path: str = Depends(get_db_path)):
    print(DB_PATH)
    return {"Hello": "World!"}  
@app.get("/init_db")
def init_db():
    sql = """
CREATE TABLE api_keys (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    api_key TEXT NOT NULL,
    type TEXT NOT NULL,
    status INTEGER NOT NULL,
    idx INTEGER NOT NULL,
    dest_api_key TEXT NOT NULL
);
"""
    Db('api_keys.db').execute_query(sql)

# def create_user_api_key(user_id, api_key, type, status, idx, dest_api_key):
#     sql = f"""
# INSERT INTO api_keys (api_key, type, status, idx, dest_api_key)
# """
#     Db('api_keys.db').execute_query(sql)

# create_user_api_key('i_am_tanbushi', 'llm', '1', '0', 'dest_api_key')