File size: 396 Bytes
3ca8396 e5c9448 3ca8396 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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('THEO_API_KEY')
class MyVanna(MyCustomVectorDB, OpenAI_Chat):
def __init__(self, config=None):
MyCustomVectorDB.__init__(self, config=config)
OpenAI_Chat.__init__(self, config=config) |