hf adaption

freeplay2api.py

@@ -7,6 +7,7 @@ import logging
 import asyncio
 import concurrent.futures
 from typing import Any, List, Optional, Dict, Generator, Union
+from contextlib import asynccontextmanager
 
 from proxy_pool import ProxyPool
 from fastapi import FastAPI, HTTPException, Depends, Response, Request
@@ -274,10 +275,9 @@ class AccountManager:
             logging.info(f"Loaded {len(self.accounts)} accounts from {self.filepath}")
 
     def save_accounts(self):
-        with self.lock:
-            with open(self.filepath, "w", encoding="utf-8") as f:
-                for account in self.accounts:
-                    f.write(json.dumps(account) + "\n")
+        # This operation is disabled to ensure the application is stateless.
+        # Account data is now only managed in memory for the duration of the process.
+        pass
 
     def add_account(self, account: Dict):
         with self.lock:
@@ -367,7 +367,12 @@ class KeyMaintainer(threading.Thread):
 
 
 # --- FastAPI应用 ---
-app = FastAPI(title="Freeplay.ai to OpenAI API Adapter")
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    initialize_app()
+    yield
+
+app = FastAPI(title="Freeplay.ai to OpenAI API Adapter", lifespan=lifespan)
 security = HTTPBearer()
 
 
@@ -379,30 +384,28 @@ def initialize_app():
             return
 
         # 1. 加载配置
-        if not os.path.exists("config.json"):
-            default_config = {
-                "HOST": "0.0.0.0",
-                "PORT": 8057,
-                "ACCOUNTS_FILE": "accounts.json",
-                "LOW_BALANCE_THRESHOLD": 2.0,
-                "ACTIVE_KEY_THRESHOLD": 5,
-                "CHECK_INTERVAL_SECONDS": 300,
-                "REGISTRATION_CONCURRENCY": 2,
-                "USE_PROXY_POOL": True,
-                "PROXY_POOL_CONFIG": {
-                    "target_count": 20,
-                    "min_threshold": 5,
-                    "check_interval": 180,
-                }
+        default_config = {
+            "HOST": "0.0.0.0",
+            "PORT": 8057,
+            "ACCOUNTS_FILE": "accounts.json",
+            "LOW_BALANCE_THRESHOLD": 2.0,
+            "ACTIVE_KEY_THRESHOLD": 5,
+            "CHECK_INTERVAL_SECONDS": 300,
+            "REGISTRATION_CONCURRENCY": 2,
+            "USE_PROXY_POOL": True,
+            "PROXY_POOL_CONFIG": {
+                "target_count": 20,
+                "min_threshold": 5,
+                "check_interval": 180,
             }
-            with open("config.json", "w") as f:
-                json.dump(default_config, f, indent=4)
-            config = default_config
-            logging.info("Created default config.json")
-        else:
+        }
+        try:
             with open("config.json", "r") as f:
                 config = json.load(f)
             logging.info("Loaded config from config.json")
+        except (FileNotFoundError, json.JSONDecodeError):
+            config = default_config
+            logging.info("Using default config as config.json was not found or invalid.")
 
         # 2. 加载客户端密钥
         api_key = os.environ.get("API_KEY", "sk-123456")
@@ -434,9 +437,6 @@ async def authenticate_client(auth: HTTPAuthorizationCredentials = Depends(secur
         raise HTTPException(status_code=403, detail="Invalid client API key.")
 
 
-@app.on_event("startup")
-async def startup_event():
-    initialize_app()
 
 
 @app.get("/v1/models", response_model=ModelList)