diff --git a/llm-service/app/tests/conftest.py b/llm-service/app/tests/conftest.py index 092df5048..eec513e26 100644 --- a/llm-service/app/tests/conftest.py +++ b/llm-service/app/tests/conftest.py @@ -44,6 +44,7 @@ from typing import Any, Dict, Sequence import boto3 +import lipsum import pytest from boto3.resources.base import ServiceResource from fastapi.testclient import TestClient @@ -67,6 +68,7 @@ from app.services.rag_qdrant_vector_store import RagQdrantVectorStore from app.services.utils import get_last_segment + @dataclass class BotoObject: bucket_name: str @@ -249,12 +251,14 @@ def s3_object( bucket_name = "test_bucket" key = "test/" + document_id + body = lipsum.generate_words(1000) + bucket = s3_client.Bucket(bucket_name) bucket.create(CreateBucketConfiguration={"LocationConstraint": aws_region}) bucket.put_object( Key=key, # TODO: fixturize file - Body=b"Some text to be summarized and indexed", + Body=body.encode("utf-8"), Metadata={"originalfilename": "test.txt"}, ) return BotoObject(bucket_name=bucket_name, key=key) diff --git a/llm-service/app/tests/routers/index/test_data_source.py b/llm-service/app/tests/routers/index/test_data_source.py index 5498d6237..ec57c6e94 100644 --- a/llm-service/app/tests/routers/index/test_data_source.py +++ b/llm-service/app/tests/routers/index/test_data_source.py @@ -153,4 +153,4 @@ def test_get_size( response = client.get(f"/data_sources/{data_source_id}/size") assert response.status_code == 200 - assert response.json() == 1 + assert response.json() == 6 diff --git a/llm-service/pdm.lock b/llm-service/pdm.lock index f7cc56681..923a95c0a 100644 --- a/llm-service/pdm.lock +++ b/llm-service/pdm.lock @@ -5,7 +5,7 @@ groups = ["default", "dev"] strategy = ["inherit_metadata"] lock_version = "4.5.0" -content_hash = "sha256:37f4e048b91c6850dc6133d0ab73273fa296454ed6776ad177856e619ff421d3" +content_hash = "sha256:f384682ce1a4ad75526a66be2d7658e41d8cea4f0a75e16676334a51ca735365" [[metadata.targets]] requires_python = "==3.10.*" @@ -795,6 +795,15 @@ files = [ {file = "jsonpath_python-1.0.6-py3-none-any.whl", hash = "sha256:1e3b78df579f5efc23565293612decee04214609208a2335884b3ee3f786b575"}, ] +[[package]] +name = "lipsum" +version = "0.1.2" +summary = "A randomised Lorem Ipsum generator library for Python" +groups = ["dev"] +files = [ + {file = "lipsum-0.1.2.tar.gz", hash = "sha256:ba5f46cef19104c07f889b14486a3772845fc25afa1eb5e2eee1f2d9badcb8ab"}, +] + [[package]] name = "llama-index-core" version = "0.10.68" diff --git a/llm-service/pyproject.toml b/llm-service/pyproject.toml index 8e796a9c6..4769d513e 100644 --- a/llm-service/pyproject.toml +++ b/llm-service/pyproject.toml @@ -31,6 +31,7 @@ dev = [ "pytest>=8.3.3", "ruff>=0.7.4", "mypy>=1.13.0", + "lipsum>=0.1.2", ] [tool.pytest.ini_options]