gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: renames conversation history save method

Browse source
This commit is contained in:
hajdul88 2025-10-20 10:28:03 +02:00
parent 4d3ba3f313
commit 49e9d7dc27
8 changed files with 19 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/retrieval/EntityCompletionRetriever.py
View file

@ -7,7 +7,7 @@ from cognee.infrastructure.context.BaseContextProvider import BaseContextProvide
from cognee.modules.retrieval.base_retriever import BaseRetriever
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.context_global_variables import session_user
@ -144,7 +144,7 @@ class EntityCompletionRetriever(BaseRetriever):
)
if session_save:
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

4
cognee/modules/retrieval/completion_retriever.py
View file

@ -5,7 +5,7 @@ from cognee.shared.logging_utils import get_logger
from cognee.infrastructure.databases.vector import get_vector_engine
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.modules.retrieval.base_retriever import BaseRetriever
@ -130,7 +130,7 @@ class CompletionRetriever(BaseRetriever):
)
if session_save:
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

4
cognee/modules/retrieval/graph_completion_context_extension_retriever.py
View file

@ -5,7 +5,7 @@ from cognee.shared.logging_utils import get_logger
from cognee.modules.retrieval.graph_completion_retriever import GraphCompletionRetriever
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.context_global_variables import session_user
@ -160,7 +160,7 @@ class GraphCompletionContextExtensionRetriever(GraphCompletionRetriever):
)
if session_save:
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

4
cognee/modules/retrieval/graph_completion_cot_retriever.py
View file

@ -6,7 +6,7 @@ from cognee.shared.logging_utils import get_logger
from cognee.modules.retrieval.graph_completion_retriever import GraphCompletionRetriever
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.infrastructure.llm.LLMGateway import LLMGateway
@ -163,7 +163,7 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
# Save to session cache
if session_save:
context_summary = await summarize_text(context_text)
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

4
cognee/modules/retrieval/graph_completion_retriever.py
View file

@ -11,7 +11,7 @@ from cognee.modules.retrieval.base_graph_retriever import BaseGraphRetriever
from cognee.modules.retrieval.utils.brute_force_triplet_search import brute_force_triplet_search
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.shared.logging_utils import get_logger
@ -198,7 +198,7 @@ class GraphCompletionRetriever(BaseGraphRetriever):
)
if session_save:
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

4
cognee/modules/retrieval/temporal_retriever.py
View file

@ -7,7 +7,7 @@ from operator import itemgetter
from cognee.infrastructure.databases.vector import get_vector_engine
from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
get_conversation_history,
)
from cognee.infrastructure.databases.graph import get_graph_engine
@ -195,7 +195,7 @@ class TemporalRetriever(GraphCompletionRetriever):
)
if session_save:
await save_to_session_cache(
await save_conversation_history(
query=query,
context_summary=context_summary,
answer=completion,

2
cognee/modules/retrieval/utils/session_cache.py
View file

@ -7,7 +7,7 @@ from cognee.shared.logging_utils import get_logger
logger = get_logger("session_cache")
async def save_to_session_cache(
async def save_conversation_history(
query: str,
context_summary: str,
answer: str,

12
cognee/tests/unit/modules/retrieval/conversation_history_test.py
View file

@ -80,7 +80,7 @@ class TestConversationHistoryUtils:
@pytest.mark.asyncio
async def test_save_to_session_cache_saves_correctly(self):
"""Test save_to_session_cache calls add_qa with correct parameters."""
"""Test save_conversation_history calls add_qa with correct parameters."""
user = create_mock_user()
session_user.set(user)
@ -99,10 +99,10 @@ class TestConversationHistoryUtils:
MockCacheConfig.return_value = mock_config
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
)
result = await save_to_session_cache(
result = await save_conversation_history(
query="What is Python?",
context_summary="Python is a programming language",
answer="Python is a high-level programming language",
@ -120,7 +120,7 @@ class TestConversationHistoryUtils:
@pytest.mark.asyncio
async def test_save_to_session_cache_uses_default_session_when_none(self):
"""Test save_to_session_cache uses 'default_session' when session_id is None."""
"""Test save_conversation_history uses 'default_session' when session_id is None."""
user = create_mock_user()
session_user.set(user)
@ -139,10 +139,10 @@ class TestConversationHistoryUtils:
MockCacheConfig.return_value = mock_config
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
save_conversation_history,
)
result = await save_to_session_cache(
result = await save_conversation_history(
query="Test question",
context_summary="Test context",
answer="Test answer",