update location

app.py

@@ -138,7 +138,6 @@ def predict(
     raw_answer=post_process(response)
     model_found_files = raw_answer.strip().split("\n")
 
-
     files, _, _ = get_full_file_paths_and_classes_and_functions(structure)
     model_found_files = [correct_file_path_in_structure(file, structure) for file in model_found_files]
     found_files = correct_file_paths(model_found_files, files)
@@ -146,7 +145,6 @@ def predict(
     highlight_response = highlight_thinking(response)
     yield [[prompt,highlight_response]], [["null test","null test2"]], "Generate: Success"
 
-
     contents = ""
     for file_path in found_files:
         file_name = file_path.replace("```","")
@@ -337,7 +335,7 @@ 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(480)
+    time.sleep(450)
     print("Served vllm!")
 
 

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 == ".":
-            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]
+            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)
@@ -138,7 +138,16 @@ def create_structure(directory_path):
                     "text": file_lines,
                 }
             else:
-                curr_struct[file_name] = {}
+                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):