cognee 0.3.6__py3-none-any.whl → 0.3.7.dev1__py3-none-any.whl

cognee/tests/unit/infrastructure/databases/cache/test_cache_config.py

@@ -0,0 +1,89 @@
+"""Tests for cache configuration."""
+
+import pytest
+from cognee.infrastructure.databases.cache.config import CacheConfig, get_cache_config
+
+
+def test_cache_config_defaults():
+    """Test that CacheConfig has the correct default values."""
+    config = CacheConfig()
+
+    assert config.caching is False
+    assert config.shared_kuzu_lock is False
+    assert config.cache_host == "localhost"
+    assert config.cache_port == 6379
+    assert config.agentic_lock_expire == 240
+    assert config.agentic_lock_timeout == 300
+
+
+def test_cache_config_custom_values():
+    """Test that CacheConfig accepts custom values."""
+    config = CacheConfig(
+        caching=True,
+        shared_kuzu_lock=True,
+        cache_host="redis.example.com",
+        cache_port=6380,
+        agentic_lock_expire=120,
+        agentic_lock_timeout=180,
+    )
+
+    assert config.caching is True
+    assert config.shared_kuzu_lock is True
+    assert config.cache_host == "redis.example.com"
+    assert config.cache_port == 6380
+    assert config.agentic_lock_expire == 120
+    assert config.agentic_lock_timeout == 180
+
+
+def test_cache_config_to_dict():
+    """Test the to_dict method returns all configuration values."""
+    config = CacheConfig(
+        caching=True,
+        shared_kuzu_lock=True,
+        cache_host="test-host",
+        cache_port=7000,
+        agentic_lock_expire=100,
+        agentic_lock_timeout=200,
+    )
+
+    config_dict = config.to_dict()
+
+    assert config_dict == {
+        "caching": True,
+        "shared_kuzu_lock": True,
+        "cache_host": "test-host",
+        "cache_port": 7000,
+        "cache_username": None,
+        "cache_password": None,
+        "agentic_lock_expire": 100,
+        "agentic_lock_timeout": 200,
+    }
+
+
+def test_get_cache_config_singleton():
+    """Test that get_cache_config returns the same instance."""
+    config1 = get_cache_config()
+    config2 = get_cache_config()
+
+    assert config1 is config2
+
+
+def test_cache_config_extra_fields_allowed():
+    """Test that CacheConfig allows extra fields due to extra='allow'."""
+    config = CacheConfig(extra_field="extra_value", another_field=123)
+
+    assert hasattr(config, "extra_field")
+    assert config.extra_field == "extra_value"
+    assert hasattr(config, "another_field")
+    assert config.another_field == 123
+
+
+def test_cache_config_boolean_type_validation():
+    """Test that boolean fields accept various truthy/falsy values."""
+    config1 = CacheConfig(caching="true", shared_kuzu_lock="yes")
+    assert config1.caching is True
+    assert config1.shared_kuzu_lock is True
+
+    config2 = CacheConfig(caching="false", shared_kuzu_lock="no")
+    assert config2.caching is False
+    assert config2.shared_kuzu_lock is False

cognee/tests/unit/modules/retrieval/conversation_history_test.py

@@ -0,0 +1,154 @@
+import pytest
+from unittest.mock import AsyncMock, patch, MagicMock
+from cognee.context_global_variables import session_user
+import importlib
+
+
+def create_mock_cache_engine(qa_history=None):
+    mock_cache = AsyncMock()
+    if qa_history is None:
+        qa_history = []
+    mock_cache.get_latest_qa = AsyncMock(return_value=qa_history)
+    mock_cache.add_qa = AsyncMock(return_value=None)
+    return mock_cache
+
+
+def create_mock_user():
+    mock_user = MagicMock()
+    mock_user.id = "test-user-id-123"
+    return mock_user
+
+
+class TestConversationHistoryUtils:
+    @pytest.mark.asyncio
+    async def test_get_conversation_history_returns_empty_when_no_history(self):
+        user = create_mock_user()
+        session_user.set(user)
+        mock_cache = create_mock_cache_engine([])
+
+        cache_module = importlib.import_module(
+            "cognee.infrastructure.databases.cache.get_cache_engine"
+        )
+
+        with patch.object(cache_module, "get_cache_engine", return_value=mock_cache):
+            from cognee.modules.retrieval.utils.session_cache import get_conversation_history
+
+            result = await get_conversation_history(session_id="test_session")
+
+        assert result == ""
+
+    @pytest.mark.asyncio
+    async def test_get_conversation_history_formats_history_correctly(self):
+        """Test get_conversation_history formats Q&A history with correct structure."""
+        user = create_mock_user()
+        session_user.set(user)
+
+        mock_history = [
+            {
+                "time": "2024-01-15 10:30:45",
+                "question": "What is AI?",
+                "context": "AI is artificial intelligence",
+                "answer": "AI stands for Artificial Intelligence",
+            }
+        ]
+        mock_cache = create_mock_cache_engine(mock_history)
+
+        # Import the real module to patch safely
+        cache_module = importlib.import_module(
+            "cognee.infrastructure.databases.cache.get_cache_engine"
+        )
+
+        with patch.object(cache_module, "get_cache_engine", return_value=mock_cache):
+            with patch(
+                "cognee.modules.retrieval.utils.session_cache.CacheConfig"
+            ) as MockCacheConfig:
+                mock_config = MagicMock()
+                mock_config.caching = True
+                MockCacheConfig.return_value = mock_config
+
+                from cognee.modules.retrieval.utils.session_cache import (
+                    get_conversation_history,
+                )
+
+                result = await get_conversation_history(session_id="test_session")
+
+                assert "Previous conversation:" in result
+                assert "[2024-01-15 10:30:45]" in result
+                assert "QUESTION: What is AI?" in result
+                assert "CONTEXT: AI is artificial intelligence" in result
+                assert "ANSWER: AI stands for Artificial Intelligence" in result
+
+    @pytest.mark.asyncio
+    async def test_save_to_session_cache_saves_correctly(self):
+        """Test save_conversation_history calls add_qa with correct parameters."""
+        user = create_mock_user()
+        session_user.set(user)
+
+        mock_cache = create_mock_cache_engine([])
+
+        cache_module = importlib.import_module(
+            "cognee.infrastructure.databases.cache.get_cache_engine"
+        )
+
+        with patch.object(cache_module, "get_cache_engine", return_value=mock_cache):
+            with patch(
+                "cognee.modules.retrieval.utils.session_cache.CacheConfig"
+            ) as MockCacheConfig:
+                mock_config = MagicMock()
+                mock_config.caching = True
+                MockCacheConfig.return_value = mock_config
+
+                from cognee.modules.retrieval.utils.session_cache import (
+                    save_conversation_history,
+                )
+
+                result = await save_conversation_history(
+                    query="What is Python?",
+                    context_summary="Python is a programming language",
+                    answer="Python is a high-level programming language",
+                    session_id="my_session",
+                )
+
+                assert result is True
+                mock_cache.add_qa.assert_called_once()
+
+                call_kwargs = mock_cache.add_qa.call_args.kwargs
+                assert call_kwargs["question"] == "What is Python?"
+                assert call_kwargs["context"] == "Python is a programming language"
+                assert call_kwargs["answer"] == "Python is a high-level programming language"
+                assert call_kwargs["session_id"] == "my_session"
+
+    @pytest.mark.asyncio
+    async def test_save_to_session_cache_uses_default_session_when_none(self):
+        """Test save_conversation_history uses 'default_session' when session_id is None."""
+        user = create_mock_user()
+        session_user.set(user)
+
+        mock_cache = create_mock_cache_engine([])
+
+        cache_module = importlib.import_module(
+            "cognee.infrastructure.databases.cache.get_cache_engine"
+        )
+
+        with patch.object(cache_module, "get_cache_engine", return_value=mock_cache):
+            with patch(
+                "cognee.modules.retrieval.utils.session_cache.CacheConfig"
+            ) as MockCacheConfig:
+                mock_config = MagicMock()
+                mock_config.caching = True
+                MockCacheConfig.return_value = mock_config
+
+                from cognee.modules.retrieval.utils.session_cache import (
+                    save_conversation_history,
+                )
+
+                result = await save_conversation_history(
+                    query="Test question",
+                    context_summary="Test context",
+                    answer="Test answer",
+                    session_id=None,
+                )
+
+                assert result is True
+                call_kwargs = mock_cache.add_qa.call_args.kwargs
+                assert call_kwargs["session_id"] == "default_session"

cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py

@@ -2,6 +2,7 @@ import os
 import pytest
 import pathlib
 from typing import Optional, Union
+from pydantic import BaseModel
 
 import cognee
 from cognee.low_level import setup, DataPoint
@@ -10,6 +11,11 @@ from cognee.tasks.storage import add_data_points
 from cognee.modules.retrieval.graph_completion_cot_retriever import GraphCompletionCotRetriever
 
 
+class TestAnswer(BaseModel):
+    answer: str
+    explanation: str
+
+
 class TestGraphCompletionCoTRetriever:
     @pytest.mark.asyncio
     async def test_graph_completion_cot_context_simple(self):
@@ -168,3 +174,48 @@ class TestGraphCompletionCoTRetriever:
         assert all(isinstance(item, str) and item.strip() for item in answer), (
             "Answer must contain only non-empty strings"
         )
+
+    @pytest.mark.asyncio
+    async def test_get_structured_completion(self):
+        system_directory_path = os.path.join(
+            pathlib.Path(__file__).parent, ".cognee_system/test_get_structured_completion"
+        )
+        cognee.config.system_root_directory(system_directory_path)
+        data_directory_path = os.path.join(
+            pathlib.Path(__file__).parent, ".data_storage/test_get_structured_completion"
+        )
+        cognee.config.data_root_directory(data_directory_path)
+
+        await cognee.prune.prune_data()
+        await cognee.prune.prune_system(metadata=True)
+        await setup()
+
+        class Company(DataPoint):
+            name: str
+
+        class Person(DataPoint):
+            name: str
+            works_for: Company
+
+        company1 = Company(name="Figma")
+        person1 = Person(name="Steve Rodger", works_for=company1)
+
+        entities = [company1, person1]
+        await add_data_points(entities)
+
+        retriever = GraphCompletionCotRetriever()
+
+        # Test with string response model (default)
+        string_answer = await retriever.get_structured_completion("Who works at Figma?")
+        assert isinstance(string_answer, str), f"Expected str, got {type(string_answer).__name__}"
+        assert string_answer.strip(), "Answer should not be empty"
+
+        # Test with structured response model
+        structured_answer = await retriever.get_structured_completion(
+            "Who works at Figma?", response_model=TestAnswer
+        )
+        assert isinstance(structured_answer, TestAnswer), (
+            f"Expected TestAnswer, got {type(structured_answer).__name__}"
+        )
+        assert structured_answer.answer.strip(), "Answer field should not be empty"
+        assert structured_answer.explanation.strip(), "Explanation field should not be empty"

{cognee-0.3.6.dist-info → cognee-0.3.7.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.3.6
+Version: 0.3.7.dev1
 Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
 Project-URL: Homepage, https://www.cognee.ai
 Project-URL: Repository, https://github.com/topoteretes/cognee
@@ -15,13 +15,13 @@ Classifier: Operating System :: MacOS :: MacOS X
 Classifier: Operating System :: Microsoft :: Windows
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: <=3.13,>=3.10
+Requires-Python: <3.14,>=3.10
 Requires-Dist: aiofiles<24.0.0,>=23.2.1
 Requires-Dist: aiohttp<4.0.0,>=3.11.14
 Requires-Dist: aiosqlite<1.0.0,>=0.20.0
 Requires-Dist: alembic<2,>=1.13.3
 Requires-Dist: fastapi-users[sqlalchemy]<15.0.0,>=14.0.1
-Requires-Dist: fastapi<1.0.0,>=0.115.7
+Requires-Dist: fastapi<1.0.0,>=0.116.2
 Requires-Dist: fastembed<=0.6.0
 Requires-Dist: filetype<2.0.0,>=1.2.0
 Requires-Dist: gunicorn<24,>=20.1.0
@@ -31,6 +31,7 @@ Requires-Dist: kuzu==0.11.3
 Requires-Dist: lancedb<1.0.0,>=0.24.0
 Requires-Dist: limits<5,>=4.4.1
 Requires-Dist: litellm>=1.76.0
+Requires-Dist: mistralai>=1.9.10
 Requires-Dist: nbformat<6.0.0,>=5.7.0
 Requires-Dist: networkx<4,>=3.4.2
 Requires-Dist: numpy<=4.0.0,>=1.26.4
@@ -47,6 +48,7 @@ Requires-Dist: python-multipart<1.0.0,>=0.0.20
 Requires-Dist: rdflib<7.2.0,>=7.1.4
 Requires-Dist: sqlalchemy<3.0.0,>=2.0.39
 Requires-Dist: structlog<26,>=25.2.0
+Requires-Dist: tenacity>=9.0.0
 Requires-Dist: tiktoken<1.0.0,>=0.8.0
 Requires-Dist: typing-extensions<5.0.0,>=4.12.2
 Requires-Dist: uvicorn<1.0.0,>=0.34.0
@@ -90,7 +92,11 @@ Provides-Extra: distributed
 Requires-Dist: modal<2.0.0,>=1.0.5; extra == 'distributed'
 Provides-Extra: dlt
 Requires-Dist: dlt[sqlalchemy]<2,>=1.9.0; extra == 'dlt'
+Provides-Extra: docling
+Requires-Dist: docling>=2.54; extra == 'docling'
+Requires-Dist: transformers>=4.55; extra == 'docling'
 Provides-Extra: docs
+Requires-Dist: lxml<6.0.0; extra == 'docs'
 Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,pdf,ppt,pptx,rst,rtf,tsv,xlsx]<19,>=0.18.1; extra == 'docs'
 Provides-Extra: evals
 Requires-Dist: gdown<6,>=5.2.0; extra == 'evals'
@@ -132,6 +138,15 @@ Requires-Dist: pgvector<0.4,>=0.3.5; extra == 'postgres-binary'
 Requires-Dist: psycopg2-binary<3.0.0,>=2.9.10; extra == 'postgres-binary'
 Provides-Extra: posthog
 Requires-Dist: posthog<4,>=3.5.0; extra == 'posthog'
+Provides-Extra: redis
+Requires-Dist: redis<6.0.0,>=5.0.3; extra == 'redis'
+Provides-Extra: scraping
+Requires-Dist: apscheduler<=3.11.0,>=3.10.0; extra == 'scraping'
+Requires-Dist: beautifulsoup4>=4.13.1; extra == 'scraping'
+Requires-Dist: lxml>=4.9.3; extra == 'scraping'
+Requires-Dist: playwright>=1.9.0; extra == 'scraping'
+Requires-Dist: protego>=0.1; extra == 'scraping'
+Requires-Dist: tavily-python>=0.7.12; extra == 'scraping'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -207,7 +222,7 @@ Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Ext
 
 ## Get Started
 
-Get started quickly with a Google Colab  <a href="https://colab.research.google.com/drive/1jHbWVypDgCLwjE71GSXhRL3YxYhCZzG1?usp=sharing">notebook</a> , <a href="https://deepnote.com/workspace/cognee-382213d0-0444-4c89-8265-13770e333c02/project/cognee-demo-78ffacb9-5832-4611-bb1a-560386068b30/notebook/Notebook-1-75b24cda566d4c24ab348f7150792601?utm_source=share-modal&utm_medium=product-shared-content&utm_campaign=notebook&utm_content=78ffacb9-5832-4611-bb1a-560386068b30">Deepnote notebook</a> or  <a href="https://github.com/topoteretes/cognee/tree/main/cognee-starter-kit">starter repo</a>
+Get started quickly with a Google Colab  <a href="https://colab.research.google.com/drive/12Vi9zID-M3fpKpKiaqDBvkk98ElkRPWy?usp=sharing">notebook</a> , <a href="https://deepnote.com/workspace/cognee-382213d0-0444-4c89-8265-13770e333c02/project/cognee-demo-78ffacb9-5832-4611-bb1a-560386068b30/notebook/Notebook-1-75b24cda566d4c24ab348f7150792601?utm_source=share-modal&utm_medium=product-shared-content&utm_campaign=notebook&utm_content=78ffacb9-5832-4611-bb1a-560386068b30">Deepnote notebook</a> or  <a href="https://github.com/topoteretes/cognee/tree/main/cognee-starter-kit">starter repo</a>
 
 
 ## About cognee
@@ -360,12 +375,12 @@ We now have a paper you can cite:
 
 ```bibtex
 @misc{markovic2025optimizinginterfaceknowledgegraphs,
-      title={Optimizing the Interface Between Knowledge Graphs and LLMs for Complex Reasoning}, 
+      title={Optimizing the Interface Between Knowledge Graphs and LLMs for Complex Reasoning},
       author={Vasilije Markovic and Lazar Obradovic and Laszlo Hajdu and Jovan Pavlovic},
       year={2025},
       eprint={2505.24478},
       archivePrefix={arXiv},
       primaryClass={cs.AI},
-      url={https://arxiv.org/abs/2505.24478}, 
+      url={https://arxiv.org/abs/2505.24478},
 }
 ```