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

添加id,sql,question到上下文
This commit is contained in:
yujj128
2025-09-23 20:06:32 +08:00
parent 1393b021eb
commit 23916fc328
3 changed files with 16 additions and 16 deletions

View File

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