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

wip: temp

Browse Source
This commit is contained in:
superobk
2024-05-16 16:02:13 +08:00
parent e4136ee4e3
commit 0612c065c0
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -7,18 +7,19 @@ from src.configuration import EnvConf, LogConf, singleton
@singleton @singleton
class Main(): class Main():
env_conf: EnvConf env_conf: EnvConf
@inject @inject
def __init__(self, log_config: LogConf, env_conf: EnvConf, log_handler: LogHandler) -> None: def __init__(self, log_config: LogConf, env_conf: EnvConf, log_handler: LogHandler) -> None:
self.env_conf=env_conf self.env_conf=env_conf
logging.basicConfig( logging.basicConfig(
handlers=[log_handler], handlers=[logging.StreamHandler()],
level=log_config.level, level=log_config.level,
datefmt=log_config.time_format, datefmt=log_config.time_format,
format='%(asctime)s %(message)s') format='%(asctime)s %(message)s')
def run(self): def run(self):
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.info("jarvis-models start", "version", "0.0.1") logger.info("jarvis-models start", extra={"version": "0.0.1"})
uvicorn.run("server:app", host="0.0.0.0", port=8000, log_level="info") uvicorn.run("server:app", host="0.0.0.0", port=8000, log_level="info")
if __name__ == "__main__": if __name__ == "__main__":

8
src/blackbox/blackbox_factory.py
View File

@ -37,7 +37,6 @@ class BlackboxFactory:
#chroma_query: ChromaQuery, #chroma_query: ChromaQuery,
#chroma_upsert: ChromaUpsert, #chroma_upsert: ChromaUpsert,
#chroma_chat: ChromaChat, #chroma_chat: ChromaChat,
melotts: MeloTTS,
vlms: VLMS) -> None: vlms: VLMS) -> None:
self.models["audio_to_text"] = audio_to_text self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio self.models["text_to_audio"] = text_to_audio
@ -50,10 +49,9 @@ class BlackboxFactory:
self.models["audio_chat"] = audio_chat self.models["audio_chat"] = audio_chat
self.models["g2e"] = g2e self.models["g2e"] = g2e
self.models["text_and_image"] = text_and_image self.models["text_and_image"] = text_and_image
# self.models["chroma_query"] = chroma_query #self.models["chroma_query"] = chroma_query
# self.models["chroma_upsert"] = chroma_upsert #self.models["chroma_upsert"] = chroma_upsert
# self.models["chroma_chat"] = chroma_chat #self.models["chroma_chat"] = chroma_chat
self.models["melotts"] = melotts
self.models["vlms"] = vlms self.models["vlms"] = vlms
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):

1
src/blackbox/g2e.py
View File

@ -65,6 +65,7 @@ class G2E(Blackbox):
api_key='YOUR_API_KEY', api_key='YOUR_API_KEY',
base_url=url base_url=url
) )
#model_name = client.models.list().data[0].id
model_name = client.models.list().data[1].id model_name = client.models.list().data[1].id
print(model_name) print(model_name)
response = client.chat.completions.create( response = client.chat.completions.create(

6
src/log/handler.py
View File

@ -1,13 +1,13 @@
import logging import logging
class LogHandler(logging.Handler): class LogHandler(logging.StreamHandler):
def __init__(self)-> None: def __init__(self)-> None:
logging.Handler.__init__(self=self) logging.Handler.__init__(self=self)
def emit(self, record) -> None: def emit(self, record) -> None:
print("emit", record) print("emit", record)
def handle(self, record) -> None: def handle(self, record) -> None:
print("@ handle",record) print("@ handle",record)