Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix: fix graphiti test
  • Loading branch information
borisarzentar committed Feb 19, 2025
commit 398b7aa04c0d6252ddaaeeb615a1cb4ca80dc17b
2 changes: 1 addition & 1 deletion evals/eval_swe_bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from cognee.modules.retrieval.description_to_codepart_search import (
code_description_to_code_part_search,
)
from evals.eval_utils import download_github_repo, retrieved_edges_to_string
from evals.eval_utils import download_github_repo


def check_install_package(package_name):
Expand Down
24 changes: 17 additions & 7 deletions notebooks/cognee_graphiti_demo.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,9 @@
{
"cell_type": "markdown",
"metadata": {},
"source": "# Cognee Graphiti integration demo"
"source": [
"# Cognee Graphiti integration demo"
]
},
{
"cell_type": "markdown",
Expand All @@ -12,7 +14,9 @@
"languageId": "plaintext"
}
},
"source": "First we import the necessary libaries"
"source": [
"First we import the necessary libaries"
]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -90,7 +94,9 @@
{
"cell_type": "markdown",
"metadata": {},
"source": "## Input texts with temporal information"
"source": [
"## Input texts with temporal information"
]
},
{
"cell_type": "code",
Expand All @@ -113,7 +119,9 @@
{
"cell_type": "markdown",
"metadata": {},
"source": "## Running graphiti + transforming its graph into cognee's core system (graph transformation + vector embeddings)"
"source": [
"## Running graphiti + transforming its graph into cognee's core system (graph transformation + vector embeddings)"
]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -181,11 +189,13 @@
{
"cell_type": "markdown",
"metadata": {},
"source": "## Retrieving and generating answer from graphiti graph with cognee retriever"
"source": [
"## Retrieving and generating answer from graphiti graph with cognee retriever"
]
},
{
"cell_type": "code",
"execution_count": 5,
"execution_count": null,
"metadata": {
"ExecuteTime": {
"end_time": "2025-01-15T10:44:27.844438Z",
Expand Down Expand Up @@ -213,7 +223,7 @@
")\n",
"\n",
"# Step 3: Preparing the Context for the LLM\n",
"context = retrieved_edges_to_string(triplets)\n",
"context = await retrieved_edges_to_string(triplets)\n",
"\n",
"args = {\"question\": query, \"context\": context}\n",
"\n",
Expand Down
Loading