1

app.py

@@ -332,11 +332,11 @@ def main(args: argparse.Namespace):
     )
 
 if __name__ == "__main__":
-    print("Start serving vllm...")
-    script_path = os.path.join(os.path.dirname(__file__), "serve_vllm.sh")
-    subprocess.Popen(["bash", script_path])
-    time.sleep(450)
-    print("Served vllm!")
+    # print("Start serving vllm...")
+    # script_path = os.path.join(os.path.dirname(__file__), "serve_vllm.sh")
+    # subprocess.Popen(["bash", script_path])
+    # time.sleep(450)
+    # print("Served vllm!")
 
 
     args = parse_args()

kimi_dev/serve/templates.py

@@ -119,15 +119,15 @@ def create_structure(directory_path):
     """
     structure = {}
     for root, _, files in os.walk(directory_path):
+        repo_name = os.path.basename(directory_path)
         relative_root = os.path.relpath(root, directory_path)
         if relative_root == ".":
-            curr_struct = structure
-        else:
-            curr_struct = structure
-            for part in relative_root.split(os.sep):
-                if part not in curr_struct:
-                    curr_struct[part] = {}
-                curr_struct = curr_struct[part]
+            relative_root = repo_name
+        curr_struct = structure
+        for part in relative_root.split(os.sep):
+            if part not in curr_struct:
+                curr_struct[part] = {}
+            curr_struct = curr_struct[part]
         for file_name in files:
             if file_name.endswith(".py"):
                 file_path = os.path.join(root, file_name)
@@ -138,18 +138,47 @@ def create_structure(directory_path):
                     "text": file_lines,
                 }
             else:
-                file_path = os.path.join(root, file_name)
-                try:
-                    with open(file_path, 'r', encoding='utf-8') as f:
-                        file_content = f.read()
-                    curr_struct[file_name] = {
-                        "text": file_content.splitlines()
-                    }
-                except Exception as e:
-                    print(f"fail to read {file_path}: {e}")
-                    curr_struct[file_name] = {}
+                curr_struct[file_name] = {}
     return structure
 
+# def create_structure(directory_path):
+#     """Create the structure of the repository directory by parsing Python files.
+#     :param directory_path: Path to the repository directory.
+#     :return: A dictionary representing the structure.
+#     """
+#     structure = {}
+#     for root, _, files in os.walk(directory_path):
+#         relative_root = os.path.relpath(root, directory_path)
+#         if relative_root == ".":
+#             curr_struct = structure
+#         else:
+#             curr_struct = structure
+#             for part in relative_root.split(os.sep):
+#                 if part not in curr_struct:
+#                     curr_struct[part] = {}
+#                 curr_struct = curr_struct[part]
+#         for file_name in files:
+#             if file_name.endswith(".py"):
+#                 file_path = os.path.join(root, file_name)
+#                 class_info, function_names, file_lines = parse_python_file(file_path)
+#                 curr_struct[file_name] = {
+#                     "classes": class_info,
+#                     "functions": function_names,
+#                     "text": file_lines,
+#                 }
+#             else:
+#                 file_path = os.path.join(root, file_name)
+#                 try:
+#                     with open(file_path, 'r', encoding='utf-8') as f:
+#                         file_content = f.read()
+#                     curr_struct[file_name] = {
+#                         "text": file_content.splitlines()
+#                     }
+#                 except Exception as e:
+#                     print(f"fail to read {file_path}: {e}")
+#                     curr_struct[file_name] = {}
+#     return structure
+
 def build_repo_structure(root_path):
     """Build repository structure using improved parsing method"""
     return create_structure(root_path)