reorganizing

agent.py

@@ -4,7 +4,7 @@ from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.table import Table
 from rich.box import SIMPLE
-from wiki_run_engine import WikiRunEnvironment
+from local.engine import WikiRunEnvironment
 from langsmith import traceable
 import os
 from openai import OpenAI

cli.py

@@ -3,11 +3,10 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.prompt import Prompt
-from rich.text import Text
 from rich.table import Table
 
-from wiki_run_engine import WikiRunEnvironment
-from wiki_solver import WikiSolver
+from engine import WikiRunEnvironment
+from BFS_solver import WikiSolver
 
 console = Console()
 

wiki_run_engine.py → engine.py

@@ -4,7 +4,7 @@
 # 3. Navigation between articles
 
 import random
-from wiki_db_json import WikiDBJson
+from db.wiki_db_json import WikiDBJson
 
 class WikiRunEnvironment:
     def __init__(self, wiki_data_path):