chore: lint fix (#4)

* chore: lint fix

* chore: update pyproject.toml instead or changing isort command on github workflow

Author: gRedHeadphone

docs/property_graph_store.ipynb

@@ -318,8 +318,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# This allows running nested event loop \n",
+    "# This allows running nested event loop\n",
     "import nest_asyncio\n",
+    "\n",
     "nest_asyncio.apply()"
    ]
   },

pyproject.toml

@@ -25,3 +25,9 @@ ipykernel = "^6.29.5"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.black]
+target-version = ['py39']
+
+[tool.isort]
+profile = "black"

src/llama_index_spanner/__init__.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .graph_retriever import (SpannerGraphCustomRetriever,
-                              SpannerGraphTextToGQLRetriever)
+from .graph_retriever import SpannerGraphCustomRetriever, SpannerGraphTextToGQLRetriever
 from .property_graph_store import SpannerPropertyGraphStore
 from .version import __version__
 

src/llama_index_spanner/graph_retriever.py

@@ -21,15 +21,18 @@
 from llama_index.core.output_parsers import PydanticOutputParser
 from llama_index.core.postprocessor.llm_rerank import LLMRerank
 from llama_index.core.prompts import PromptType
-from llama_index.core.retrievers import (CustomPGRetriever,
-                                         VectorContextRetriever)
+from llama_index.core.retrievers import CustomPGRetriever, VectorContextRetriever
 from llama_index.core.schema import NodeWithScore, QueryBundle, TextNode
 from llama_index.core.vector_stores.types import VectorStore
 from pydantic import BaseModel
 
 from .graph_utils import extract_gql, fix_gql_syntax
-from .prompts import (DEFAULT_GQL_FIX_TEMPLATE, DEFAULT_GQL_VERIFY_TEMPLATE,
-                      DEFAULT_SPANNER_GQL_TEMPLATE, DEFAULT_SUMMARY_TEMPLATE)
+from .prompts import (
+    DEFAULT_GQL_FIX_TEMPLATE,
+    DEFAULT_GQL_VERIFY_TEMPLATE,
+    DEFAULT_SPANNER_GQL_TEMPLATE,
+    DEFAULT_SUMMARY_TEMPLATE,
+)
 from .property_graph_store import SpannerPropertyGraphStore
 
 GQL_GENERATION_PROMPT = PromptTemplate(

src/llama_index_spanner/property_graph_store.py

@@ -17,16 +17,25 @@
 from typing import Any, Dict, List, Optional, Tuple
 
 from google.cloud import spanner
-from llama_index.core.graph_stores.types import (ChunkNode, EntityNode,
-                                                 LabelledNode,
-                                                 PropertyGraphStore, Relation,
-                                                 Triplet)
+from llama_index.core.graph_stores.types import (
+    ChunkNode,
+    EntityNode,
+    LabelledNode,
+    PropertyGraphStore,
+    Relation,
+    Triplet,
+)
 from llama_index.core.prompts import PromptTemplate, PromptType
 from llama_index.core.vector_stores.types import VectorStoreQuery
 
 from .prompts import DEFAULT_SPANNER_GQL_TEMPLATE
-from .schema import (ElementSchema, GraphDocumentUtility, SpannerGraphSchema,
-                     group_edges, group_nodes)
+from .schema import (
+    ElementSchema,
+    GraphDocumentUtility,
+    SpannerGraphSchema,
+    group_edges,
+    group_nodes,
+)
 from .spanner import SpannerImpl, SpannerInterface, client_with_user_agent
 from .version import __version__
 

src/llama_index_spanner/schema.py

@@ -19,8 +19,12 @@
 from typing import Any, Dict, Generator, Iterable, List, Optional, Tuple
 
 from google.cloud.spanner_v1 import JsonObject, param_types
-from llama_index.core.graph_stores.types import (ChunkNode, EntityNode,
-                                                 LabelledNode, Relation)
+from llama_index.core.graph_stores.types import (
+    ChunkNode,
+    EntityNode,
+    LabelledNode,
+    Relation,
+)
 from requests.structures import CaseInsensitiveDict
 
 from .type_utils import TypeUtility

tests/test_graph_retriever.py

@@ -20,7 +20,9 @@
 from llama_index.readers.wikipedia import WikipediaReader
 
 from llama_index_spanner.graph_retriever import (
-    SpannerGraphCustomRetriever, SpannerGraphTextToGQLRetriever)
+    SpannerGraphCustomRetriever,
+    SpannerGraphTextToGQLRetriever,
+)
 from tests.utils import get_resources
 
 

tests/test_spanner_property_graph_store.py

@@ -16,18 +16,24 @@
 
 import pytest
 from llama_index.core.graph_stores.types import ChunkNode, EntityNode, Relation
-from llama_index.core.vector_stores.types import (ExactMatchFilter,
-                                                  FilterCondition,
-                                                  FilterOperator,
-                                                  MetadataFilter,
-                                                  MetadataFilters,
-                                                  VectorStoreQuery)
+from llama_index.core.vector_stores.types import (
+    ExactMatchFilter,
+    FilterCondition,
+    FilterOperator,
+    MetadataFilter,
+    MetadataFilters,
+    VectorStoreQuery,
+)
 
 from llama_index_spanner.property_graph_store import SpannerPropertyGraphStore
 from llama_index_spanner.schema import ElementSchema
-from tests.utils import (get_random_suffix, get_spanner_property_graph_store,
-                         spanner_database_id, spanner_graph_name,
-                         spanner_instance_id)
+from tests.utils import (
+    get_random_suffix,
+    get_spanner_property_graph_store,
+    spanner_database_id,
+    spanner_graph_name,
+    spanner_instance_id,
+)
 
 pytestmark = pytest.mark.skipif(
     (not spanner_instance_id or not spanner_database_id or not spanner_graph_name),