wip

Author: tschellenbach

examples/03_phone_and_rag_example/conftest.py

@@ -0,0 +1,5 @@
+"""Pytest configuration for phone example tests."""
+
+import pytest
+
+# Override root conftest - we don't need blockbuster for these tests

examples/03_phone_and_rag_example/pyproject.toml

@@ -20,6 +20,9 @@ dependencies = [
   "turbopuffer>=0.1",
   "langchain-google-genai>=2.0",
   "langchain-text-splitters>=0.3",
+  # Testing
+  "pytest>=8.0",
+  "pytest-asyncio>=0.24",
 ]
 
 [tool.uv.sources]

examples/03_phone_and_rag_example/pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+asyncio_mode = auto
+markers =
+    integration: marks tests as integration tests (require API keys)

examples/03_phone_and_rag_example/rag_turbopuffer.py

@@ -104,7 +104,7 @@ def __init__(
         self,
         namespace: str,
         embedding_model: str = "models/gemini-embedding-001",
-        chunk_size: int = 1000,
+        chunk_size: int = 10000,
         chunk_overlap: int = 200,
         region: str = "gcp-us-central1",
     ):

examples/03_phone_and_rag_example/rag_utils.py

@@ -0,0 +1,5 @@
+
+"""
+1. find files with markdown
+
+"""

examples/03_phone_and_rag_example/test_rag_turbopuffer.py

@@ -0,0 +1,36 @@
+"""Tests for TurboPuffer Hybrid RAG."""
+
+from pathlib import Path
+
+import pytest
+from dotenv import load_dotenv
+
+from rag_turbopuffer import TurboPufferRAG
+
+load_dotenv()
+import logging
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+@pytest.fixture
+def knowledge_dir(tmp_path: Path) -> Path:
+    """Create temp directory with a test document."""
+    doc = tmp_path / "test.md"
+    doc.write_text("# Stream Chat API\n\nThe Chat API supports real-time messaging and moderation.")
+    return tmp_path
+
+
+@pytest.mark.integration
+async def test_index_and_search(knowledge_dir: Path):
+    """Index a document and find it via search."""
+    rag = TurboPufferRAG(namespace="test-rag")
+
+    await rag.index_directory(knowledge_dir)
+
+    result = await rag.search("chat messaging")
+    assert "Chat" in result
+    logger.info("result %s", result)
+    import pdb; pdb.set_trace()
+
+    await rag.close()