from dotenv import load_dotenv
from climateqa.engine.talk_to_data.vanna_class import MyCustomVectorDB
from vanna.openai import OpenAI_Chat
import os

load_dotenv()

OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')

class MyVanna(MyCustomVectorDB, OpenAI_Chat):
    def __init__(self, config=None):
        MyCustomVectorDB.__init__(self, config=config)
        OpenAI_Chat.__init__(self, config=config)