multimodal cache modify

Author: peng3307165

modelcache/manager/scalar_data/sql_storage.py

@@ -40,6 +40,7 @@ def _insert(self, data: List):
         embedding_data = data[2]
         model = data[3]
         answer_type = 0
+        print('embedding_data: {}'.format(embedding_data))
         embedding_data = embedding_data.tobytes()
 
         table_name = "cache_codegpt_answer"
@@ -61,6 +62,7 @@ def _insert(self, data: List):
     def batch_insert(self, all_data: List[CacheData]):
         ids = []
         for data in all_data:
+            print('data: {}'.format(data))
             ids.append(self._insert(data))
         return ids
 

modelcache/manager_mm/__init__.py

@@ -0,0 +1,12 @@
+# -*- coding: utf-8 -*-
+"""
+   Alipay.com Inc.
+   Copyright (c) 2004-2023 All Rights Reserved.
+   ------------------------------------------------------
+   File Name : __init__.py.py
+   Author : fuhui.phe
+   Create Time : 2024/4/17 10:53
+   Description : description what the main function of this file
+   Change Activity: 
+        version0 : 2024/4/17 10:53 by fuhui.phe  init
+"""

modelcache_mm/__init__.py

@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
 import logging
 
-from modelcache.adapter_mm.adapter_query import adapt_query
-from modelcache.adapter_mm.adapter_insert import adapt_insert
-from modelcache.adapter_mm.adapter_remove import adapt_remove
-from modelcache.adapter_mm.adapter_register import adapt_register
+from modelcache_mm.adapter_mm.adapter_query import adapt_query
+from modelcache_mm.adapter_mm.adapter_insert import adapt_insert
+from modelcache_mm.adapter_mm.adapter_remove import adapt_remove
+from modelcache_mm.adapter_mm.adapter_register import adapt_register
 
 
 class ChatCompletion(object):

modelcache/adapter_mm/__init__.py renamed to modelcache_mm/adapter_mm/__init__.py

@@ -66,6 +66,9 @@ def adapt_insert(*args, **kwargs):
         image_embeddings = embedding_data_resp['image_embedding']
         text_embeddings = embedding_data_resp['text_embeddings']
 
+        print('image_embeddings: {}'.format(image_embeddings))
+        print('text_embeddings: {}'.format(text_embeddings))
+
         if len(image_embeddings) > 0 and len(image_embeddings) > 0:
             image_embedding = np.array(image_embeddings[0])
             text_embedding = text_embeddings[0]
@@ -82,6 +85,7 @@ def adapt_insert(*args, **kwargs):
         else:
             raise ValueError('maya embedding service return both empty list, please check!')
 
+    print('embedding_data: {}'.format(embedding_data))
     chat_cache.data_manager.save(
         pre_embedding_text,
         pre_embedding_image_url,

modelcache/adapter_mm/adapter.py renamed to modelcache_mm/adapter_mm/adapter.py

@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-