eaf8d718b0
<!-- .github/pull_request_template.md --> ## Description This PR introduces a new memify pipeline to save cache sessions in cognee. The QA sessions are added to the main knowledge base as separate documents. ## Type of Change <!-- Please check the relevant option --> - [ ] Bug fix (non-breaking change that fixes an issue) - [x] New feature (non-breaking change that adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [ ] Documentation update - [ ] Code refactoring - [ ] Performance improvement - [ ] Other (please specify): ## Screenshots/Videos (if applicable) None ## Pre-submission Checklist <!-- Please check all boxes that apply before submitting your PR --> - [x] **I have tested my changes thoroughly before submitting this PR** - [x] **This PR contains minimal changes necessary to address the issue/feature** - [x] My code follows the project's coding standards and style guidelines - [x] I have added tests that prove my fix is effective or that my feature works - [x] I have added necessary documentation (if applicable) - [x] All new and existing tests pass - [x] I have searched existing PRs to ensure this change hasn't been submitted already - [x] I have linked any relevant issues in the description - [x] My commits have clear and descriptive messages ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import asyncio
|
|
|
|
import cognee
|
|
from cognee import visualize_graph
|
|
from cognee.memify_pipelines.persist_sessions_in_knowledge_graph import (
|
|
persist_sessions_in_knowledge_graph_pipeline,
|
|
)
|
|
from cognee.modules.search.types import SearchType
|
|
from cognee.modules.users.methods import get_default_user
|
|
from cognee.shared.logging_utils import get_logger
|
|
|
|
logger = get_logger("conversation_session_persistence_example")
|
|
|
|
|
|
async def main():
|
|
# NOTE: CACHING has to be enabled for this example to work
|
|
await cognee.prune.prune_data()
|
|
await cognee.prune.prune_system(metadata=True)
|
|
|
|
text_1 = "Cognee is a solution that can build knowledge graph from text, creating an AI memory system"
|
|
text_2 = "Germany is a country located next to the Netherlands"
|
|
|
|
await cognee.add([text_1, text_2])
|
|
await cognee.cognify()
|
|
|
|
question = "What can I use to create a knowledge graph?"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION,
|
|
query_text=question,
|
|
)
|
|
print("\nSession ID: default_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
question = "You sure about that?"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION, query_text=question
|
|
)
|
|
print("\nSession ID: default_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
question = "This is awesome!"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION, query_text=question
|
|
)
|
|
print("\nSession ID: default_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
question = "Where is Germany?"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION,
|
|
query_text=question,
|
|
session_id="different_session",
|
|
)
|
|
print("\nSession ID: different_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
question = "Next to which country again?"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION,
|
|
query_text=question,
|
|
session_id="different_session",
|
|
)
|
|
print("\nSession ID: different_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
question = "So you remember everything I asked from you?"
|
|
search_results = await cognee.search(
|
|
query_type=SearchType.GRAPH_COMPLETION,
|
|
query_text=question,
|
|
session_id="different_session",
|
|
)
|
|
print("\nSession ID: different_session")
|
|
print(f"Question: {question}")
|
|
print(f"Answer: {search_results}\n")
|
|
|
|
session_ids_to_persist = ["default_session", "different_session"]
|
|
default_user = await get_default_user()
|
|
|
|
await persist_sessions_in_knowledge_graph_pipeline(
|
|
user=default_user,
|
|
session_ids=session_ids_to_persist,
|
|
)
|
|
|
|
await visualize_graph()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
try:
|
|
loop.run_until_complete(main())
|
|
finally:
|
|
loop.run_until_complete(loop.shutdown_asyncgens())
|