From ab7c6f55a207a39d13d193b99dd80561d1f84e32 Mon Sep 17 00:00:00 2001 From: Zain Hoda <7146154+zainhoda@users.noreply.github.com> Date: Thu, 14 Sep 2023 14:41:23 -0400 Subject: [PATCH] delete mock --- src/vanna/mock.py | 21 --------------------- 1 file changed, 21 deletions(-) delete mode 100644 src/vanna/mock.py diff --git a/src/vanna/mock.py b/src/vanna/mock.py deleted file mode 100644 index 27690c46..00000000 --- a/src/vanna/mock.py +++ /dev/null @@ -1,21 +0,0 @@ -from .chromadb_vector import ChromaDB_VectorStore -from .openai_embeddings import OpenAI_Embeddings - - -class MockModel(ChromaDB_VectorStore): - def __init__(self, config=None): - # OpenAI_Embeddings.__init__(self, config=config) - ChromaDB_VectorStore.__init__(self, config=config) - - def get_prompt( - self, - question: str, - question_sql_list: list, - ddl_list: list, - doc_list: list, - **kwargs, - ) -> str: - pass - - def submit_prompt(self, prompt: str, **kwargs) -> str: - pass