Sort all imports (#77)

Author: jonbesga

.github/workflows/pr.yaml

@@ -35,8 +35,9 @@ jobs:
         run: poetry install --no-interaction
       - name: Check format and linting
         run: |
-          poetry run ruff format --check .
+          poetry run ruff check --select I .
           poetry run ruff check .
+          poetry run ruff format --check .
       - name: Run strict mypy check
         run: poetry run mypy --strict --ignore-missing-imports --allow-subclassing-any --allow-untyped-calls .
       - name: Run unit tests and check coverage

.pre-commit-config.yaml

@@ -20,9 +20,14 @@ repos:
           - neo4j
           - langchain_community
           - langchain_openai
-
   - repo: local
     hooks:
+      - id: ruff-lint-isort
+        name: Ruff Lint Sort Imports
+        entry: poetry run ruff check --select I .
+        language: system
+        types: [ python ]
+        stages: [ commit, push ]
       - id: ruff-lint
         name: Ruff Lint Check
         entry: poetry run ruff format --check

examples/graphrag.py

@@ -8,13 +8,13 @@
 """
 
 import logging
-import neo4j
 
+import neo4j
 from neo4j_genai.embeddings.openai import OpenAIEmbeddings
-from neo4j_genai.types import RetrieverResultItem
-from neo4j_genai.retrievers import VectorCypherRetriever
-from neo4j_genai.llm import OpenAILLM
 from neo4j_genai.generation import GraphRAG
+from neo4j_genai.llm import OpenAILLM
+from neo4j_genai.retrievers import VectorCypherRetriever
+from neo4j_genai.types import RetrieverResultItem
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/graphrag_custom_prompt.py

@@ -9,13 +9,13 @@
 """
 
 import logging
-import neo4j
 
-from neo4j_genai.types import RetrieverResultItem
+import neo4j
 from neo4j_genai.embeddings.openai import OpenAIEmbeddings
-from neo4j_genai.retrievers import VectorCypherRetriever
 from neo4j_genai.generation import GraphRAG, RagTemplate
 from neo4j_genai.llm import OpenAILLM
+from neo4j_genai.retrievers import VectorCypherRetriever
+from neo4j_genai.types import RetrieverResultItem
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/graphrag_with_langchain_llm.py

@@ -8,13 +8,13 @@
 """
 
 import logging
-import neo4j
 
+import neo4j
 from langchain.chat_models import ChatOpenAI
 from langchain.embeddings import OpenAIEmbeddings
-from neo4j_genai.types import RetrieverResultItem
-from neo4j_genai.retrievers import VectorCypherRetriever
 from neo4j_genai.generation import GraphRAG
+from neo4j_genai.retrievers import VectorCypherRetriever
+from neo4j_genai.types import RetrieverResultItem
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/hybrid_cypher_search.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
-from neo4j import GraphDatabase
 
 from random import random
-from neo4j_genai.retrievers import HybridCypherRetriever
+
+from neo4j import GraphDatabase
 from neo4j_genai.embedder import Embedder
-from neo4j_genai.indexes import create_vector_index, create_fulltext_index
+from neo4j_genai.indexes import create_fulltext_index, create_vector_index
+from neo4j_genai.retrievers import HybridCypherRetriever
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/hybrid_search.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
-from neo4j import GraphDatabase
 
 from random import random
-from neo4j_genai.retrievers import HybridRetriever
+
+from neo4j import GraphDatabase
 from neo4j_genai.embedder import Embedder
-from neo4j_genai.indexes import create_vector_index, create_fulltext_index
+from neo4j_genai.indexes import create_fulltext_index, create_vector_index
+from neo4j_genai.retrievers import HybridRetriever
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/openai_search.py

@@ -1,10 +1,9 @@
-from neo4j import GraphDatabase
-from neo4j_genai.retrievers import VectorRetriever
-
 from random import random
-from neo4j_genai.indexes import create_vector_index
 
 from langchain_openai import OpenAIEmbeddings
+from neo4j import GraphDatabase
+from neo4j_genai.indexes import create_vector_index
+from neo4j_genai.retrievers import VectorRetriever
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/similarity_search_for_text.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
-from neo4j import GraphDatabase
-from neo4j_genai.retrievers import VectorRetriever
 
 from random import random
+
+from neo4j import GraphDatabase
 from neo4j_genai.embedder import Embedder
 from neo4j_genai.indexes import create_vector_index
+from neo4j_genai.retrievers import VectorRetriever
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")

examples/similarity_search_for_vector.py

@@ -1,9 +1,8 @@
-from neo4j import GraphDatabase
-from neo4j_genai.retrievers import VectorRetriever
-
 from random import random
 
+from neo4j import GraphDatabase
 from neo4j_genai.indexes import create_vector_index
+from neo4j_genai.retrievers import VectorRetriever
 
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")