批量添加ddl到向量数据库，添加documention,重写generate_rewritten_question

service/cus_vanna_srevice.py

@@ -3,6 +3,7 @@ from typing import List
 
 import orjson
 from vanna.base import VannaBase
+from vanna.flask import MemoryCache
 from vanna.qdrant import Qdrant_VectorStore
 from qdrant_client import QdrantClient
 from openai import OpenAI
@@ -17,6 +18,7 @@ from datetime import datetime
 class OpenAICompatibleLLM(VannaBase):
     def __init__(self, client=None, config_file=None):
         VannaBase.__init__(self, config=config_file)
+        self.cache = MemoryCache()
 
         # default parameters - can be overrided using config
         self.temperature = 0.5
@@ -168,6 +170,8 @@ class OpenAICompatibleLLM(VannaBase):
                 [{'role': 'system', 'content': sys_char_temp}, {'role': 'user', 'content': user_char_temp}], **kwargs)
             print(llm_response2)
             result['chart'] = orjson.loads(extract_nested_json(llm_response2))
+        result['id'] = self.cache.generate_id(question=question)
+        print("result----------------------{0}".format(result))
         return result
 
     def generate_rewritten_question(self, last_question: str, new_question: str, **kwargs) -> str: