Merge pull request #321 from andreped/ollama-fix

Added smart import shortcuts for some submodules

src/vanna/ZhipuAI/ZhipuAI_Chat.py

@@ -1,11 +1,11 @@
 import re
 from typing import List
+
 import pandas as pd
 from zhipuai import ZhipuAI
+
 from ..base import VannaBase
-import re
-from typing import List
-import pandas as pd
+
 
 class ZhipuAI_Chat(VannaBase):
     def __init__(self, config=None):

src/vanna/ZhipuAI/__init__.py

@@ -0,0 +1,2 @@
+from .ZhipuAI_Chat import ZhipuAI_Chat
+from .ZhipuAI_embeddings import ZhipuAI_Embeddings, ZhipuAIEmbeddingFunction

src/vanna/anthropic/__init__.py

@@ -0,0 +1 @@
+from .anthropic_chat import Anthropic_Chat

src/vanna/chromadb/__init__.py

@@ -0,0 +1 @@
+from .chromadb_vector import ChromaDB_VectorStore

src/vanna/marqo/__init__.py

@@ -0,0 +1 @@
+from .marqo import Marqo_VectorStore

src/vanna/marqo/marqo.py

@@ -1,6 +1,4 @@
-import json
 import uuid
-from abc import abstractmethod
 
 import marqo
 import pandas as pd

src/vanna/mistral/__init__.py

@@ -0,0 +1 @@
+from .mistral import Mistral

src/vanna/ollama/__init__.py

@@ -0,0 +1 @@
+from .ollama import Ollama

src/vanna/ollama/ollama.py

@@ -1,4 +1,3 @@
-import json
 import re
 
 import requests

src/vanna/openai/__init__.py

@@ -0,0 +1,2 @@
+from .openai_chat import OpenAI_Chat
+from .openai_embeddings import OpenAI_Embeddings

src/vanna/vannadb/__init__.py

@@ -0,0 +1 @@
+from .vannadb_vector import VannaDB_VectorStore