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

Merge branch 'COG-adding_html_graph_render' of github.com:topoteretes/cognee into COG-adding_html_graph_render

Browse source
This commit is contained in:
Igor Ilic 2025-01-08 14:50:38 +01:00
commit 9c2c19458f
4 changed files with 631 additions and 647 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/reusable_python_example.yml vendored
View file

@ -33,7 +33,7 @@ jobs:
- name: Setup Python - name: Setup Python
uses: actions/setup-python@v5 uses: actions/setup-python@v5
with: with:
python-version: '3.11.x' python-version: '3.12.x'
- name: Install Poetry - name: Install Poetry
uses: snok/install-poetry@v1 uses: snok/install-poetry@v1

17
cognee/shared/utils.py
View file

@ -13,12 +13,7 @@ import matplotlib.pyplot as plt
import tiktoken import tiktoken
import nltk import nltk
import base64 import base64
from bokeh.plotting import figure, from_networkx
from bokeh.models import Circle, MultiLine, HoverTool, ColumnDataSource, Range1d
from bokeh.plotting import output_file, show
from bokeh.embed import file_html
from bokeh.resources import CDN
import logging import logging
import sys import sys
@ -364,6 +359,12 @@ def style_and_render_graph(p, G, layout_positions, node_attribute, node_colors,
""" """
Apply styling and render the graph into the plot. Apply styling and render the graph into the plot.
""" """
from bokeh.plotting import figure, from_networkx
from bokeh.models import Circle, MultiLine, HoverTool, ColumnDataSource, Range1d
from bokeh.plotting import output_file, show
from bokeh.embed import file_html
from bokeh.resources import CDN
graph_renderer = from_networkx(G, layout_positions) graph_renderer = from_networkx(G, layout_positions)
node_radii = [0.02 + 0.1 * centrality[node] for node in G.nodes()] node_radii = [0.02 + 0.1 * centrality[node] for node in G.nodes()]
graph_renderer.node_renderer.data_source.data["radius"] = node_radii graph_renderer.node_renderer.data_source.data["radius"] = node_radii
@ -397,6 +398,12 @@ async def create_cognee_style_network_with_logo(
""" """
Create a Cognee-inspired network visualization with an embedded logo. Create a Cognee-inspired network visualization with an embedded logo.
""" """
from bokeh.plotting import figure, from_networkx
from bokeh.models import Circle, MultiLine, HoverTool, ColumnDataSource, Range1d
from bokeh.plotting import output_file, show
from bokeh.embed import file_html
from bokeh.resources import CDN
logging.info("Converting graph to serializable format...") logging.info("Converting graph to serializable format...")
G = await convert_to_serializable_graph(G) G = await convert_to_serializable_graph(G)

1237
poetry.lock generated
View file

File diff suppressed because it is too large Load diff

22
pyproject.toml
View file

@ -18,8 +18,8 @@ classifiers = [
] ]
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.10.0,<3.12" python = ">=3.9.0,<3.13"
openai = "1.59.4" openai = "1.52.0"
pydantic = "2.8.2" pydantic = "2.8.2"
python-dotenv = "1.0.1" python-dotenv = "1.0.1"
fastapi = ">=0.109.2,<0.116.0" fastapi = ">=0.109.2,<0.116.0"
@ -34,18 +34,18 @@ falkordb = {version = "1.0.9", optional = true}
boto3 = "^1.26.125" boto3 = "^1.26.125"
botocore="^1.35.54" botocore="^1.35.54"
gunicorn = "^20.1.0" gunicorn = "^20.1.0"
sqlalchemy = "2.0.36" sqlalchemy = "2.0.35"
instructor = "1.7.2" instructor = "1.5.2"
networkx = "^3.2.1" networkx = "^3.2.1"
aiosqlite = "^0.20.0" aiosqlite = "^0.20.0"
pandas = "2.2.3" pandas = "2.0.3"
filetype = "^1.2.0" filetype = "^1.2.0"
nltk = "^3.8.1" nltk = "^3.8.1"
dlt = {extras = ["sqlalchemy"], version = "^1.4.1"} dlt = {extras = ["sqlalchemy"], version = "^1.4.1"}
aiofiles = "^23.2.1" aiofiles = "^23.2.1"
qdrant-client = {version = "^1.9.0", optional = true} qdrant-client = {version = "^1.9.0", optional = true}
graphistry = "^0.33.5" graphistry = "^0.33.5"
tenacity = "^9.0.0" tenacity = "^8.4.1"
weaviate-client = {version = "4.9.6", optional = true} weaviate-client = {version = "4.9.6", optional = true}
scikit-learn = "^1.5.0" scikit-learn = "^1.5.0"
pypdf = "^4.1.0" pypdf = "^4.1.0"
@ -58,7 +58,7 @@ langsmith = {version = "0.2.3", optional = true}
langdetect = "1.0.9" langdetect = "1.0.9"
posthog = {version = "^3.5.0", optional = true} posthog = {version = "^3.5.0", optional = true}
lancedb = "0.16.0" lancedb = "0.16.0"
litellm = "1.57.2" litellm = "1.49.1"
groq = {version = "0.8.0", optional = true} groq = {version = "0.8.0", optional = true}
langfuse = "^2.32.0" langfuse = "^2.32.0"
pydantic-settings = "^2.2.1" pydantic-settings = "^2.2.1"
@ -69,16 +69,14 @@ alembic = "^1.13.3"
asyncpg = {version = "0.30.0", optional = true} asyncpg = {version = "0.30.0", optional = true}
pgvector = {version = "^0.3.5", optional = true} pgvector = {version = "^0.3.5", optional = true}
psycopg2 = {version = "^2.9.10", optional = true} psycopg2 = {version = "^2.9.10", optional = true}
llama-index-core = {version = "^0.12.10.post1", optional = true} llama-index-core = {version = "^0.11.22", optional = true}
deepeval = {version = "^2.0.1", optional = true} deepeval = {version = "^2.0.1", optional = true}
transformers = "^4.46.3" transformers = "^4.46.3"
pymilvus = {version = "^2.5.0", optional = true} pymilvus = {version = "^2.5.0", optional = true}
unstructured = { extras = ["csv", "doc", "docx", "epub", "md", "odt", "org", "ppt", "pptx", "rst", "rtf", "tsv", "xlsx"], version = "^0.16.12", optional = true } unstructured = { extras = ["csv", "doc", "docx", "epub", "md", "odt", "org", "ppt", "pptx", "rst", "rtf", "tsv", "xlsx"], version = "^0.16.10", optional = true }
pre-commit = "^4.0.1" pre-commit = "^4.0.1"
httpx = "0.27.0" httpx = "0.27.0"
bokeh="^3.6.2"
svglib = "^1.5.1"
reportlab = "^4.2.5"