From 4398c539764028bea0f5fbbf35e9dec39e2211af Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Fri, 6 Jun 2025 14:37:52 -0400 Subject: [PATCH 1/3] bump: dependencies --- pyproject.toml | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 35aa776..b5e876b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,7 @@ classifiers = [ "Topic :: Software Development :: Libraries :: Python Modules", ] dependencies = [ - "networkx>=3.0,<=3.4", + "networkx>=3.0,<=3.5", "phenolrs~=0.5", "python-arango~=8.1", "adbnx-adapter~=5.0.5" @@ -54,9 +54,8 @@ dev = [ "sphinx_rtd_theme", ] llm = [ - "langchain~=0.2.14", - "langchain-openai~=0.1.22", - "langchain-community~=0.2.12" + "langchain-arangodb", + "langchain_openai" ] [project.urls] From d0749f187ba29cef86b1edcfe034643786f0834a Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Fri, 6 Jun 2025 14:38:02 -0400 Subject: [PATCH 2/3] fix: langchain --- nx_arangodb/classes/graph.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/nx_arangodb/classes/graph.py b/nx_arangodb/classes/graph.py index 8c9f335..ab43bac 100644 --- a/nx_arangodb/classes/graph.py +++ b/nx_arangodb/classes/graph.py @@ -37,8 +37,7 @@ __all__ = ["Graph"] try: - from langchain_community.chains.graph_qa.arangodb import ArangoGraphQAChain - from langchain_community.graphs import ArangoGraph + from langchain_arangodb import ArangoGraph, ArangoGraphQAChain from langchain_core.language_models import BaseLanguageModel from langchain_openai import ChatOpenAI @@ -616,10 +615,7 @@ def chat( if llm is None: llm = ChatOpenAI(temperature=0, model_name="gpt-4") - graph = ArangoGraph( - self.db, - # graph_name=self.name # not yet supported - ) + graph = ArangoGraph(self.db, schema_graph_name=self.name) chain = ArangoGraphQAChain.from_llm( llm=llm, From 5f32bfefbe95f28bdfe467bfb02216289cfde33b Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Fri, 6 Jun 2025 14:39:50 -0400 Subject: [PATCH 3/3] ignore: `F824` --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index b5e876b..0fe6462 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -108,7 +108,7 @@ extend_skip_glob = [ [tool.flake8] max-line-length = 88 -extend-ignore = ["E203", "W503", "E251", "F401", "F403"] +extend-ignore = ["E203", "W503", "E251", "F401", "F403", "F824"] exclude = [".git", ".idea", ".*_cache", "dist", "venv"] [tool.mypy]