chore: desc

pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "srctag"
 version = "0.1.1"
-description = "Tag source files with real-world requirements."
+description = "Tag source files with real-world stories."
 authors = ["williamfzc <[email protected]>"]
 license = "Apache-2.0"
 readme = "README.md"

srctag/tagger.py

@@ -1,5 +1,6 @@
 import csv
 import typing
+from collections import OrderedDict
 
 from chromadb import QueryResult, Metadata
 from pydantic import BaseModel
@@ -12,6 +13,7 @@
 
 
 class TagResult(BaseModel):
+    # todo: need some other ways for querying flexibly
     files: typing.Dict[str, SingleTagResult] = dict()
 
     def export_csv(self, path: str = "srctag-output.csv") -> None:
@@ -73,7 +75,9 @@ def tag(self, storage: Storage) -> TagResult:
             for each_file, each_score in zip(files, normalized_scores):
                 each_file_name = each_file["source"]
                 if each_file_name not in ret:
-                    ret[each_file_name] = dict()
+                    ret[each_file_name] = OrderedDict()
                 ret[each_file_name][each_tag] = each_score
+            # END file loop
+        # END tag loop
 
         return TagResult(files=ret)