fix code style

Author: Yucheng-Jiang

examples/storm_examples/run_storm_wiki_mistral.py

@@ -15,6 +15,7 @@
         storm_gen_article.txt           # Final article generated
         storm_gen_article_polished.txt  # Polished final article (if args.do_polish_article is True)
 """
+
 import os
 from argparse import ArgumentParser
 

examples/storm_examples/run_storm_wiki_ollama.py

@@ -15,6 +15,7 @@
         storm_gen_article.txt           # Final article generated
         storm_gen_article_polished.txt  # Polished final article (if args.do_polish_article is True)
 """
+
 import os
 import sys
 from argparse import ArgumentParser

frontend/demo_light/pages_util/CreateNewArticle.py

@@ -34,9 +34,9 @@ def handle_not_started():
                 ]:
                     if not st.session_state["page3_topic"].strip():
                         pass_appropriateness_check = False
-                        st.session_state[
-                            "page3_warning_message"
-                        ] = "topic could not be empty"
+                        st.session_state["page3_warning_message"] = (
+                            "topic could not be empty"
+                        )
 
                     st.session_state["page3_topic_name_cleaned"] = (
                         st.session_state["page3_topic"]

frontend/demo_light/pages_util/MyArticles.py

@@ -23,9 +23,9 @@ def my_articles_page():
     if "page2_user_articles_file_path_dict" not in st.session_state:
         local_dir = os.path.join(demo_util.get_demo_dir(), "DEMO_WORKING_DIR")
         os.makedirs(local_dir, exist_ok=True)
-        st.session_state[
-            "page2_user_articles_file_path_dict"
-        ] = DemoFileIOHelper.read_structure_to_dict(local_dir)
+        st.session_state["page2_user_articles_file_path_dict"] = (
+            DemoFileIOHelper.read_structure_to_dict(local_dir)
+        )
 
     # if no feature demo selected, display all featured articles as info cards
     def article_card_setup(column_to_add, card_title, article_name):

knowledge_storm/dataclass.py

@@ -773,9 +773,9 @@ def merge_node(node):
     def update_all_info_path(self):
         def _helper(node):
             for citation_idx in node.content:
-                self.info_uuid_to_info_dict[citation_idx].meta[
-                    "placement"
-                ] = " -> ".join(node.get_path_from_root())
+                self.info_uuid_to_info_dict[citation_idx].meta["placement"] = (
+                    " -> ".join(node.get_path_from_root())
+                )
             for child in node.children:
                 _helper(child)
 

knowledge_storm/encoder.py

@@ -53,6 +53,7 @@ class Encoder:
         Refer to the LiteLLM documentation for details on supported embedding models:
         https://docs.litellm.ai/docs/embedding/supported_embedding
     """
+
     def __init__(
         self,
         encoder_type: Optional[str] = None,

knowledge_storm/interface.py

@@ -502,9 +502,9 @@ def wrapper(*args, **kwargs):
             logger.info(f"{func.__name__} executed in {execution_time:.4f} seconds")
             self.lm_cost[func.__name__] = self.lm_configs.collect_and_reset_lm_usage()
             if hasattr(self, "retriever"):
-                self.rm_cost[
-                    func.__name__
-                ] = self.retriever.collect_and_reset_rm_usage()
+                self.rm_cost[func.__name__] = (
+                    self.retriever.collect_and_reset_rm_usage()
+                )
             return result
 
         return wrapper

knowledge_storm/lm.py

@@ -269,6 +269,7 @@ def __call__(self, prompt=None, messages=None, **kwargs):
 # The following language model classes were deprecated after v1.0.2.
 # They remain in this file for backward compatibility but will no longer be maintained.
 
+
 class OpenAIModel(dspy.OpenAI):
     """A wrapper class for dspy.OpenAI."""
 

knowledge_storm/storm_wiki/modules/storm_dataclass.py

@@ -58,9 +58,9 @@ class StormInformationTable(InformationTable):
     def __init__(self, conversations=List[Tuple[str, List[DialogueTurn]]]):
         super().__init__()
         self.conversations = conversations
-        self.url_to_info: Dict[
-            str, Information
-        ] = StormInformationTable.construct_url_to_info(self.conversations)
+        self.url_to_info: Dict[str, Information] = (
+            StormInformationTable.construct_url_to_info(self.conversations)
+        )
 
     @staticmethod
     def construct_url_to_info(