Problem occurs when using Ollama with PostgresVectorstore

[jais001]

Hi team,

I am migrating cloud genai to local ollama. When using Ollama together with Postgres vector store I am getting a run time error. Please find my code below:

from vanna.ollama import Ollama
from vanna.pgvector import PG_VectorStore


class MyVanna(PG_VectorStore, Ollama):
    """Vanna AI DB Query model"""

    def __init__(self, config=None):
        PG_VectorStore.__init__(self, config=config)
        Ollama.__init__(self, config=config)

    def add_training_data_from_documentation(
        self, documentation: str, **kwargs
    ) -> str:
        """adds training data from documentation"""
        id = self.add_documentation(documentation=documentation, **kwargs)
        return id

    def get_related_documentation(self, question: str, **kwargs) -> list:
        documents = self.documentation_collection.similarity_search(
            query=question, k=10  # Optionally adjust the number of results
        )
        return [document.page_content for document in documents]
        
        
 
my_config = {
    "ollama_host": "http://localhost:11434", "model": "mistral",
    "connection_string" : 'postgresql://testadmin:abcd123@localhost:6002/testdb'
}

vn = MyVanna(config=my_config)

ModuleNotFoundError                       Traceback (most recent call last)
File ~/Jais/vanna_ollama/venv/lib/python3.12/site-packages/transformers/utils/import_utils.py:1967, in _LazyModule._get_module(self, module_name)
   1966 try:
-> 1967     return importlib.import_module("." + module_name, self.__name__)
   1968 except Exception as e:

File ~/Jais/vanna_ollama/venv/lib/python3.12/importlib/__init__.py:90, in import_module(name, package)
     89         level += 1
---> 90 return _bootstrap._gcd_import(name[level:], package, level)

File <frozen importlib._bootstrap>:1381, in _gcd_import(name, package, level)

File <frozen importlib._bootstrap>:1354, in _find_and_load(name, import_)

File <frozen importlib._bootstrap>:1325, in _find_and_load_unlocked(name, import_)

File <frozen importlib._bootstrap>:929, in _load_unlocked(spec)

File <frozen importlib._bootstrap_external>:994, in exec_module(self, module)

File <frozen importlib._bootstrap>:488, in _call_with_frames_removed(f, *args, **kwds)

File ~/Jais/vanna_ollama/venv/lib/python3.12/site-packages/transformers/generation/utils.py:30
     28 from torch.nn import functional as F
...
   1972     ) from e

RuntimeError: Failed to import transformers.models.auto.modeling_auto because of the following error (look up to see its traceback):
Failed to import transformers.generation.utils because of the following error (look up to see its traceback):
No module named 'numpy.char'

[jais001]

issue got fixed when added requirements for langchain packages