Format Templates (#12396)

templates/elastic-query-generator/elastic_query_generator/chain.py

@@ -1,11 +1,12 @@
 import os
-from langchain.chat_models import ChatOpenAI
-from langchain.output_parsers.json import SimpleJsonOutputParser
-from elasticsearch import Elasticsearch
 from pathlib import Path
 
-from .prompts import DSL_PROMPT
+from elasticsearch import Elasticsearch
+from langchain.chat_models import ChatOpenAI
+from langchain.output_parsers.json import SimpleJsonOutputParser
+
 from .elastic_index_info import get_indices_infos
+from .prompts import DSL_PROMPT
 
 es_host = os.environ["ELASTIC_SEARCH_SERVER"]
 es_password = os.environ["ELASTIC_PASSWORD"]

templates/elastic-query-generator/elastic_query_generator/elastic_index_info.py

@@ -1,5 +1,6 @@
 from typing import List
 
+
 def _list_indices(database, include_indices=None, ignore_indices=None) -> List[str]:
     all_indices = [
         index["index"] for index in database.cat.indices(format="json")

templates/elastic-query-generator/elastic_query_generator/prompts.py

@@ -16,6 +16,6 @@ Use the following format:
 
 Question: Question here
 ESQuery: Elasticsearch Query formatted as json
-"""
+"""  # noqa: E501
 
 DSL_PROMPT = PromptTemplate.from_template(DEFAULT_DSL_TEMPLATE + PROMPT_SUFFIX)

templates/elastic-query-generator/ingest.py

@@ -1,4 +1,5 @@
 import os
+
 from elasticsearch import Elasticsearch
 
 es_host = os.environ["ELASTIC_SEARCH_SERVER"]

templates/elastic-query-generator/main.py

@@ -1,5 +1,4 @@
 from elastic_query_generator.chain import chain
 
-
 if __name__ == "__main__":
     print(chain.invoke({"input": "how many customers named Carol"}))