Skip to content

Commit

Permalink
remove pgvector tests
Browse files Browse the repository at this point in the history
  • Loading branch information
zainhoda committed Oct 23, 2024
1 parent dbb76ab commit 932131f
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions tests/test_pgvector.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,23 +2,23 @@

from dotenv import load_dotenv

from vanna.pgvector import PG_VectorStore
# from vanna.pgvector import PG_VectorStore

load_dotenv()

# Removing thiese tests for now until the dependencies are sorted out
# def get_vanna_connection_string():
# server = os.environ.get("PG_SERVER")
# driver = "psycopg"
# port = 5434
# database = os.environ.get("PG_DATABASE")
# username = os.environ.get("PG_USERNAME")
# password = os.environ.get("PG_PASSWORD")

def get_vanna_connection_string():
server = os.environ.get("PG_SERVER")
driver = "psycopg"
port = 5434
database = os.environ.get("PG_DATABASE")
username = os.environ.get("PG_USERNAME")
password = os.environ.get("PG_PASSWORD")
# return f"postgresql+psycopg://{username}:{password}@{server}:{port}/{database}"

return f"postgresql+psycopg://{username}:{password}@{server}:{port}/{database}"


def test_pgvector():
connection_string = get_vanna_connection_string()
pgclient = PG_VectorStore(config={"connection_string": connection_string})
assert pgclient is not None
# def test_pgvector():
# connection_string = get_vanna_connection_string()
# pgclient = PG_VectorStore(config={"connection_string": connection_string})
# assert pgclient is not None

0 comments on commit 932131f

Please sign in to comment.