fix: inject fail

This commit is contained in:
superobk
2024-04-29 09:31:37 +08:00
parent f76231485c
commit 4ddc33e75b
3 changed files with 7 additions and 6 deletions

View File

@ -9,6 +9,7 @@ from src.dotchain.runtime.runtime import Runtime
from src.blackbox.blackbox_factory import BlackboxFactory from src.blackbox.blackbox_factory import BlackboxFactory
import uvicorn import uvicorn
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from injector import Injector
app = FastAPI() app = FastAPI()
@ -20,8 +21,8 @@ app.add_middleware(
allow_headers=["*"], allow_headers=["*"],
) )
injector = Injector()
blackbox_factory = BlackboxFactory() blackbox_factory = injector.get(BlackboxFactory)
@app.post("/") @app.post("/")
async def blackbox(blackbox_name: Union[str, None] = None, request: Request = None): async def blackbox(blackbox_name: Union[str, None] = None, request: Request = None):

View File

@ -2,9 +2,9 @@ from fastapi import Request, Response,status
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from injector import inject from injector import inject
from blackbox.asr import ASR from .asr import ASR
from blackbox.tesou import Tesou from .tesou import Tesou
from blackbox.tts import TTS from .tts import TTS
from .blackbox import Blackbox from .blackbox import Blackbox

View File

@ -9,7 +9,7 @@ from .tesou import Tesou
from .fastchat import Fastchat from .fastchat import Fastchat
from .g2e import G2E from .g2e import G2E
from .text_and_image import TextAndImage from .text_and_image import TextAndImage
from injector import Injector, inject from injector import inject
class BlackboxFactory: class BlackboxFactory:
models = {} models = {}