refactor: keep agent variants in subfolder

src/tulip_agent/__init__.py

@@ -29,21 +29,23 @@
 #
 import logging
 
-from .base_agent import BaseAgent
-from .function_analyzer import FunctionAnalyzer
-from .tool_agent import CotToolAgent, NaiveToolAgent
-from .tool_library import ToolLibrary
-from .tulip_agent import (
+from tulip_agent.agents import (
     AutoTulipAgent,
+    BaseAgent,
+    CotToolAgent,
     CotTulipAgent,
     DfsTulipAgent,
     InformedCotTulipAgent,
     MinimalTulipAgent,
+    NaiveToolAgent,
     NaiveTulipAgent,
     OneShotCotTulipAgent,
     PrimedCotTulipAgent,
 )
 
+from .function_analyzer import FunctionAnalyzer
+from .tool_library import ToolLibrary
+
 
 __all__ = [
     AutoTulipAgent,

src/tulip_agent/agents/__init__.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+#
+# Copyright (c) 2024, Honda Research Institute Europe GmbH
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+#
+# 3. Neither the name of the copyright holder nor the names of its
+#    contributors may be used to endorse or promote products derived from
+#    this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+from .auto_tulip_agent import AutoTulipAgent as AutoTulipAgent
+from .base_agent import BaseAgent as BaseAgent
+from .cot_tool_agent import CotToolAgent as CotToolAgent
+from .cot_tulip_agent import CotTulipAgent as CotTulipAgent
+from .dfs_tulip_agent import DfsTulipAgent as DfsTulipAgent
+from .informed_cot_tulip_agent import InformedCotTulipAgent as InformedCotTulipAgent
+from .minimal_tulip_agent import MinimalTulipAgent as MinimalTulipAgent
+from .naive_tool_agent import NaiveToolAgent as NaiveToolAgent
+from .naive_tulip_agent import NaiveTulipAgent as NaiveTulipAgent
+from .one_shot_cot_tulip_agent import OneShotCotTulipAgent as OneShotCotTulipAgent
+from .primed_cot_tulip_agent import PrimedCotTulipAgent as PrimedCotTulipAgent