mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-14 00:53:25 +00:00
wip: temp
This commit is contained in:
@ -37,7 +37,6 @@ class BlackboxFactory:
|
||||
#chroma_query: ChromaQuery,
|
||||
#chroma_upsert: ChromaUpsert,
|
||||
#chroma_chat: ChromaChat,
|
||||
melotts: MeloTTS,
|
||||
vlms: VLMS) -> None:
|
||||
self.models["audio_to_text"] = audio_to_text
|
||||
self.models["text_to_audio"] = text_to_audio
|
||||
@ -50,10 +49,9 @@ class BlackboxFactory:
|
||||
self.models["audio_chat"] = audio_chat
|
||||
self.models["g2e"] = g2e
|
||||
self.models["text_and_image"] = text_and_image
|
||||
# self.models["chroma_query"] = chroma_query
|
||||
# self.models["chroma_upsert"] = chroma_upsert
|
||||
# self.models["chroma_chat"] = chroma_chat
|
||||
self.models["melotts"] = melotts
|
||||
#self.models["chroma_query"] = chroma_query
|
||||
#self.models["chroma_upsert"] = chroma_upsert
|
||||
#self.models["chroma_chat"] = chroma_chat
|
||||
self.models["vlms"] = vlms
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
|
||||
@ -65,6 +65,7 @@ class G2E(Blackbox):
|
||||
api_key='YOUR_API_KEY',
|
||||
base_url=url
|
||||
)
|
||||
#model_name = client.models.list().data[0].id
|
||||
model_name = client.models.list().data[1].id
|
||||
print(model_name)
|
||||
response = client.chat.completions.create(
|
||||
|
||||
@ -1,13 +1,13 @@
|
||||
import logging
|
||||
|
||||
class LogHandler(logging.Handler):
|
||||
class LogHandler(logging.StreamHandler):
|
||||
|
||||
def __init__(self)-> None:
|
||||
logging.Handler.__init__(self=self)
|
||||
|
||||
|
||||
def emit(self, record) -> None:
|
||||
print("emit", record)
|
||||
|
||||
|
||||
def handle(self, record) -> None:
|
||||
print("@ handle",record)
|
||||
|
||||
Reference in New Issue
Block a user