Update sql_storage.py

table name update to "modelcache_llm_answer"

Author: peng3307165

modelcache/manager/scalar_data/sql_storage.py

@@ -43,7 +43,7 @@ def _insert(self, data: List):
         embedding_data = embedding_data.tobytes()
         is_deleted = 0
 
-        table_name = "cache_codegpt_answer"
+        table_name = "modelcache_llm_answer"
         insert_sql = "INSERT INTO {} (question, answer, answer_type, model, embedding_data, is_deleted) VALUES (%s, %s, %s, %s, _binary%s, %s)".format(table_name)
         conn = self.pool.connection()
         try:
@@ -91,7 +91,7 @@ def insert_query_resp(self, query_resp, **kwargs):
             conn.close()
 
     def get_data_by_id(self, key: int):
-        table_name = "cache_codegpt_answer"
+        table_name = "modelcache_llm_answer"
         query_sql = "select question, answer, embedding_data, model from {} where id={}".format(table_name, key)
         conn_start = time.time()
         conn = self.pool.connection()
@@ -112,7 +112,7 @@ def get_data_by_id(self, key: int):
             return None
 
     def update_hit_count_by_id(self, primary_id: int):
-        table_name = "cache_codegpt_answer"
+        table_name = "modelcache_llm_answer"
         update_sql = "UPDATE {} SET hit_count = hit_count+1 WHERE id={}".format(table_name, primary_id)
         conn = self.pool.connection()