mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 09:32:22 +03:00
380cf82706
# Description # Testing backend ## Docker setup 1. Copy `.env.example` to `.env`. Some env variables were added : EMBEDDING_DIM 2. Apply supabase migratrions : ```sh supabase stop supabase db reset supabase start ``` 3. Start backend containers ``` make dev ``` ## Local setup You can also run backend without docker. 1. Install [`rye`](https://rye.astral.sh/guide/installation/). Choose the managed python version and set the version to 3.11 2. Run the following: ``` cd quivr/backend rye sync ``` 3. Source `.venv` virtual env : `source .venv/bin/activate` 4. Run the backend, make sure you are running redis and supabase API: ``` LOG_LEVEL=debug uvicorn quivr_api.main:app --log-level debug --reload --host 0.0.0.0 --port 5050 --workers 1 ``` Worker: ``` LOG_LEVEL=debug celery -A quivr_worker.celery_worker worker -l info -E --concurrency 1 ``` Notifier: ``` LOG_LEVEL=debug python worker/quivr_worker/celery_monitor.py ``` --------- Co-authored-by: chloedia <chloedaems0@gmail.com> Co-authored-by: aminediro <aminedirhoussi1@gmail.com> Co-authored-by: Antoine Dewez <44063631+Zewed@users.noreply.github.com> Co-authored-by: Chloé Daems <73901882+chloedia@users.noreply.github.com> Co-authored-by: Zewed <dewez.antoine2@gmail.com>
39 lines
898 B
Python
39 lines
898 B
Python
from pathlib import Path
|
|
from uuid import uuid4
|
|
|
|
from quivr_core.files.file import FileExtension, QuivrFile
|
|
|
|
|
|
def test_create_file():
|
|
id = uuid4()
|
|
brain_id = uuid4()
|
|
qfile = QuivrFile(
|
|
id=id,
|
|
brain_id=brain_id,
|
|
original_filename="name",
|
|
path=Path("/tmp/name"),
|
|
file_extension=FileExtension.txt,
|
|
file_sha1="123",
|
|
)
|
|
|
|
assert qfile.id == id
|
|
assert qfile.brain_id == brain_id
|
|
assert qfile.original_filename == "name"
|
|
assert qfile.path == Path("/tmp/name")
|
|
|
|
|
|
def test_create_file_add_metadata():
|
|
id = uuid4()
|
|
brain_id = uuid4()
|
|
qfile = QuivrFile(
|
|
id=id,
|
|
brain_id=brain_id,
|
|
original_filename="name",
|
|
path=Path("/tmp/name"),
|
|
file_extension=FileExtension.txt,
|
|
file_sha1="123",
|
|
metadata={"other_id": "id"},
|
|
)
|
|
|
|
assert qfile.metadata["other_id"] == "id"
|