Skip to content

Commit

Permalink
add workflow for pgvector
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Jan 5, 2024
1 parent 339c6d8 commit 73254d8
Show file tree
Hide file tree
Showing 2 changed files with 63 additions and 0 deletions.
49 changes: 49 additions & 0 deletions .github/workflows/pgvector.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
# This workflow comes from https://github.com/ofek/hatch-mypyc
# https://github.com/ofek/hatch-mypyc/blob/5a198c0ba8660494d02716cfc9d79ce4adfb1442/.github/workflows/test.yml
name: Test / pgvector

on:
schedule:
- cron: "0 0 * * *"
pull_request:
paths:
- "integrations/pgvector/**"
- ".github/workflows/pgvector.yml"

concurrency:
group: pgvector-${{ github.head_ref }}
cancel-in-progress: true

env:
PYTHONUNBUFFERED: "1"
FORCE_COLOR: "1"

jobs:
run:
name: Python ${{ matrix.python-version }} on ${{ startsWith(matrix.os, 'macos-') && 'macOS' || startsWith(matrix.os, 'windows-') && 'Windows' || 'Linux' }}
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest]
python-version: ["3.9","3.10","3.11"]

steps:
- uses: actions/checkout@v4

- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}

- name: Install Hatch
run: pip install --upgrade hatch

- name: Lint
working-directory: integrations/pgvector
if: matrix.python-version == '3.9'
run: hatch run lint:all

- name: Run tests
working-directory: integrations/pgvector
run: hatch run cov
14 changes: 14 additions & 0 deletions integrations/pgvector/tests/test_document_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,17 @@
import pytest
from haystack import Document
from haystack.testing.document_store import CountDocumentsTest, DeleteDocumentsTest, WriteDocumentsTest


from pgvector_haystack.document_store import PGvectorDocumentStore


class TestDocumentStore(CountDocumentsTest, DeleteDocumentsTest, WriteDocumentsTest):

@pytest.fixture
def docstore(self) -> PGvectorDocumentStore:
"""
This is the most basic requirement for the child class: provide
an instance of this document store so the base class can use it.
"""
return PGvectorDocumentStore() # FIXME

0 comments on commit 73254d8

Please sign in to comment.