Skip to content

Commit d916c06

Browse files
authored
bump: networkx, langchain (#85)
* bump: dependencies * fix: langchain * ignore: `F824`
1 parent a7776d5 commit d916c06

File tree

2 files changed

+6
-11
lines changed

2 files changed

+6
-11
lines changed

nx_arangodb/classes/graph.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,7 @@
3737
__all__ = ["Graph"]
3838

3939
try:
40-
from langchain_community.chains.graph_qa.arangodb import ArangoGraphQAChain
41-
from langchain_community.graphs import ArangoGraph
40+
from langchain_arangodb import ArangoGraph, ArangoGraphQAChain
4241
from langchain_core.language_models import BaseLanguageModel
4342
from langchain_openai import ChatOpenAI
4443

@@ -616,10 +615,7 @@ def chat(
616615
if llm is None:
617616
llm = ChatOpenAI(temperature=0, model_name="gpt-4")
618617

619-
graph = ArangoGraph(
620-
self.db,
621-
# graph_name=self.name # not yet supported
622-
)
618+
graph = ArangoGraph(self.db, schema_graph_name=self.name)
623619

624620
chain = ArangoGraphQAChain.from_llm(
625621
llm=llm,

pyproject.toml

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ classifiers = [
2828
"Topic :: Software Development :: Libraries :: Python Modules",
2929
]
3030
dependencies = [
31-
"networkx>=3.0,<=3.4",
31+
"networkx>=3.0,<=3.5",
3232
"phenolrs~=0.5",
3333
"python-arango~=8.1",
3434
"adbnx-adapter~=5.0.5"
@@ -54,9 +54,8 @@ dev = [
5454
"sphinx_rtd_theme",
5555
]
5656
llm = [
57-
"langchain~=0.2.14",
58-
"langchain-openai~=0.1.22",
59-
"langchain-community~=0.2.12"
57+
"langchain-arangodb",
58+
"langchain_openai"
6059
]
6160

6261
[project.urls]
@@ -109,7 +108,7 @@ extend_skip_glob = [
109108

110109
[tool.flake8]
111110
max-line-length = 88
112-
extend-ignore = ["E203", "W503", "E251", "F401", "F403"]
111+
extend-ignore = ["E203", "W503", "E251", "F401", "F403", "F824"]
113112
exclude = [".git", ".idea", ".*_cache", "dist", "venv"]
114113

115114
[tool.mypy]

0 commit comments

Comments
 (0)