mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-13 16:53:24 +00:00
fix: inject fail
This commit is contained in:
5
main.py
5
main.py
@ -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):
|
||||||
|
|||||||
@ -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
|
||||||
|
|
||||||
|
|||||||
@ -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 = {}
|
||||||
|
|||||||
Reference in New Issue
Block a user