import pandas as pd
from pandasai import Agent
from pandasai.ee.vectorstores import Qdrant

qdrant_vector_store = Qdrant()
# Sample DataFrame
sales_by_country = pd.DataFrame({
    "country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
    "revenue": [5000, 3200, 2900, 4100, 2300, 2100, 2500, 2600, 4500, 7000]
})

# By default, unless you choose a different LLM, it will use BambooLLM.
# You can get your free API key signing up at https://pandabi.ai (you can also configure it in your .env file)
os.environ["PANDASAI_API_KEY"] = "YOUR_API_KEY"

agent = Agent(sales_by_country, vectorstore=qdrant_vector_store)
agent.chat('Which are the top 5 countries by sales?')

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[9], [line 4](vscode-notebook-cell:?execution_count=9&line=4)
      [2](vscode-notebook-cell:?execution_count=9&line=2) import pandas as pd
      [3](vscode-notebook-cell:?execution_count=9&line=3) from pandasai import Agent
----> [4](vscode-notebook-cell:?execution_count=9&line=4) from pandasai.ee.vectorstores import Qdrant
      [6](vscode-notebook-cell:?execution_count=9&line=6) qdrant_vector_store = Qdrant()
      [7](vscode-notebook-cell:?execution_count=9&line=7) # Sample DataFrame

File ~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:6
      [1](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:1) """
      [2](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:2) Vector stores to store data for training purpose
      [3](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:3) """
      [5](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:5) from .chroma import ChromaDB
----> [6](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:6) from .lanceDB import LanceDB
      [7](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:7) from .milvus import Milvus
      [8](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/__init__.py:8) from .qdrant import Qdrant

File ~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:10
      [8](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:8) from lancedb.embeddings.registry import register
      [9](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:9) from lancedb.pydantic import LanceModel, Vector
---> [10](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:10) from sentence_transformers import SentenceTransformer
     [12](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:12) from pandasai.helpers.logger import Logger
     [13](https://file+.vscode-resource.vscode-cdn.net/Users/eureka/Desktop/Zilliz/~/miniconda3/envs/compute/lib/python3.10/site-packages/pandasai/ee/vectorstores/lanceDB.py:13) from pandasai.vectorstores.vectorstore import VectorStore

ModuleNotFoundError: No module named 'sentence_transformers'