ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-14 00:53:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: singleton

Browse Source
This commit is contained in:
superobk
2024-04-30 15:41:08 +08:00
parent 37f1c588cf
commit ddd00aeabb
14 changed files with 62 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/blackbox/blackbox_factory.py
View File

@ -9,8 +9,9 @@ from .tesou import Tesou
from .fastchat import Fastchat
from .g2e import G2E
from .text_and_image import TextAndImage
from injector import inject
from injector import inject, singleton
@singleton
class BlackboxFactory:
models = {}
@ -40,7 +41,7 @@ class BlackboxFactory:
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)
def create_blackbox(self, blackbox_name: str) -> Blackbox:
def call_blackbox(self, blackbox_name: str) -> Blackbox:
model = self.models.get(blackbox_name)
if model is None:
raise ValueError("Invalid blockbox type")