批量添加ddl到向量数据库,添加documention,重写generate_rewritten_question
添加id,sql,question到上下文
This commit is contained in:
		| @@ -2,6 +2,7 @@ from email.policy import default | ||||
| from typing import List | ||||
|  | ||||
| import orjson | ||||
| import pandas as pd | ||||
| from vanna.base import VannaBase | ||||
| from vanna.flask import MemoryCache | ||||
| from vanna.qdrant import Qdrant_VectorStore | ||||
| @@ -18,8 +19,6 @@ 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 | ||||
|         self.max_tokens = 5000 | ||||
| @@ -170,8 +169,6 @@ 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: | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 yujj128
					yujj128