wip: temp

src/blackbox/blackbox_factory.py

@@ -34,10 +34,9 @@ class BlackboxFactory:
                  audio_chat: AudioChat,
                  g2e: G2E,
                  text_and_image: TextAndImage,
-                 chroma_query: ChromaQuery,
-                 chroma_upsert: ChromaUpsert,
-                 chroma_chat: ChromaChat,
-                 melotts: MeloTTS,
+                 #chroma_query: ChromaQuery,
+                 #chroma_upsert: ChromaUpsert,
+                 #chroma_chat: ChromaChat,
                  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):

src/blackbox/g2e.py

@@ -22,8 +22,8 @@ class G2E(Blackbox):
     def processing(self, model_name, prompt, template, context: list)  -> str:
         if context == None:
             context = []
-        url = 'http://120.196.116.194:48890/v1'
-        #url = 'http://120.196.116.194:48892/v1'
+        #url = 'http://120.196.116.194:48890/v1'
+        url = 'http://120.196.116.194:48892/v1'
 
         background_prompt = '''KOMBUKIKI是一款茶饮料，目标受众 年龄：20-35岁 性别：女性 地点：一线城市、二线城市 职业：精英中产、都市白领 收入水平：中高收入，有一定消费能力 兴趣和爱好：注重健康，有运动习惯
 
@@ -65,7 +65,8 @@ 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[0].id
+        model_name = client.models.list().data[1].id
         print(model_name)
         response = client.chat.completions.create(
         model=model_name,

src/log/handler.py

@@ -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)