[GENAI] Add tests and fix error

Author: Subfly

.github/workflows/genai-ci-cd.yml

@@ -39,7 +39,6 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
-          pip install pytest pytest-asyncio httpx
 
       - name: Run Python tests
         run: |
@@ -61,6 +60,9 @@ jobs:
           assert response.status_code == 200
           print('Health check test passed')
           "
+          
+          # Run pytest tests
+          pytest test/ -v --tb=short
 
   build:
     name: Build and Push Docker Images

genai/rag.py

@@ -701,11 +701,11 @@ def get_collection_stats(self) -> Dict[str, Any]:
 
             result = {
                 "collection_name": "recipes",
-                "total_objects": stats.total,
+                "total_objects": len(stats),
                 "status": "healthy"
             }
 
-            logger.info(f"Retrieved collection statistics in {total_duration}ms: {stats.total} objects")
+            logger.info(f"Retrieved collection statistics in {total_duration}ms: {len(stats)} objects")
             structured_logger.info(
                 f"Collection statistics retrieval completed",
                 extra={
@@ -715,7 +715,7 @@ def get_collection_stats(self) -> Dict[str, Any]:
                         'operation': 'get_collection_stats',
                         'collection_name': 'recipes',
                         'status': 'success',
-                        'total_objects': stats.total,
+                        'total_objects': len(stats),
                         'stats_duration_ms': stats_duration
                     }
                 }

genai/requirements.txt

@@ -12,5 +12,8 @@ langchain-huggingface
 sentence-transformers
 pytest==7.4.3
 pytest-asyncio==0.23.2
+pytest-cov==4.1.0
+pytest-xdist==3.3.1
+pytest-mock==3.12.0
 httpx==0.27.0
 prometheus-fastapi-instrumentator