fix: small change to accelerag __init__ module file, | fix: fixed import error in cota_engine thought_actions file | fix: fixed imports across tests

Author: Kernel-Dirichlet

cotarag/accelerag/__init__.py

@@ -70,7 +70,6 @@
 
 __all__ = [
         "Embedder",
-        "ClaudeEmbedder",
         "Indexer",
         "Retriever",
         "QueryEngine",

cotarag/cota_engine/thought_actions.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from accelerag.query_engines.query_engines import AnthropicEngine
+from cotarag.accelerag.query_engines.query_engines import AnthropicEngine
 
 
 # Update ThoughtAction to include __str__ and __repr__ methods

pyproject.toml

@@ -52,6 +52,3 @@ Issues = "https://github.com/Kernel-Dirichlet/CoTARAG/issues"
 
 [tool.setuptools]
 packages = ["cotarag", "cotarag.cota_engine", "cotarag.accelerag"]
-
-[tool.setuptools.package-data]
-cotarag = ["py.typed"]

tests/test_cota_engine.py

@@ -7,8 +7,8 @@
 # Add the project root to Python path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from cota_engine.cota_engines import CoTAEngine
-from cota_engine.thought_actions import LLMThoughtAction
+from cotarag.cota_engine.cota_engines import CoTAEngine
+from cotarag.cota_engine.thought_actions import LLMThoughtAction
 
 class TestCoTAEngine(unittest.TestCase):
     @classmethod

tests/test_rag_image.py

@@ -12,10 +12,10 @@
 
 # Add the project root to Python path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from accelerag.managers import RAGManager
-from accelerag.embedders import ImageEmbedder
-from accelerag.indexers import ImageIndexer
-from accelerag.retrievers import ImageRetriever
+from cotarag.accelerag.managers import RAGManager
+from cotarag.accelerag.embedders import ImageEmbedder
+from cotarag.accelerag.indexers import ImageIndexer
+from cotarag.accelerag.retrievers import ImageRetriever
 
 
 def generate_digits_dataset(output_dir):
@@ -78,13 +78,13 @@ def setUpClass(cls):
         cls.rag = RAGManager(
             api_key=cls.api_key,
             dir_to_idx=cls.digits_dir,
-            grounding='soft',
-            enable_cache=False,  # Disable caching
-            use_cache=False,     # Disable caching
-            logging_enabled=True,
-            force_reindex=True,
-            hard_grounding_prompt='prompts/hard_grounding_prompt.txt',
-            soft_grounding_prompt='prompts/soft_grounding_prompt.txt'
+            grounding = 'soft',
+            enable_cache = False,  # Disable caching
+            use_cache = False,     # Disable caching
+            logging_enabled = True,
+            force_reindex = True,
+            hard_grounding_prompt = 'prompts/hard_grounding_prompt.txt',
+            soft_grounding_prompt = 'prompts/soft_grounding_prompt.txt'
         )
 
     @classmethod

tests/test_rag_text.py

@@ -8,9 +8,9 @@
 # Add the project root to Python path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from accelerag.managers import RAGManager
-from accelerag.query_utils import create_tag_hierarchy
-from accelerag.query_engines.query_engines import OpenAIEngine, AnthropicEngine
+from cotarag.accelerag.managers import RAGManager
+from cotarag.accelerag.query_utils import create_tag_hierarchy
+from cotarag.accelerag.query_engines.query_engines import OpenAIEngine, AnthropicEngine
 
 class BaseRAGTest(unittest.TestCase):
     """Base class for RAG testing with common setup and test methods."""
@@ -22,7 +22,9 @@ def setUpClass(cls):
 
         # Copy test data to temp directory
         cls.data_dir = os.path.join(cls.test_dir, 'test_data')
-        test_data_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'accelerag/arxiv_mini')
+        test_data_path = os.path.join(os.path.dirname(os.path.dirname(__file__)),
+                                      'cotarag/accelerag/arxiv_mini')
+
         shutil.copytree(test_data_path, cls.data_dir)
 
         cls.db_path = os.path.join(cls.test_dir, 'test_embeddings.db.sqlite')
@@ -32,7 +34,10 @@ def setUpClass(cls):
 
         # Get project root for constructing prompt file paths
         cls.project_root = os.path.dirname(os.path.dirname(__file__))
-        cls.prompts_dir = os.path.join(cls.project_root,'accelerag','prompts')
+        cls.prompts_dir = os.path.join(cls.project_root,
+                                       'cotarag',
+                                       'accelerag',
+                                       'prompts')
 
 
     @classmethod
@@ -103,7 +108,7 @@ def _run_rag_flow_test(self, rag):
         print(f"Query: {similar_query}")
 
         # Get chunks for similar query
-        similar_chunks = rag.retrieve(similar_query, top_k=5)
+        similar_chunks = rag.retrieve(similar_query, top_k = 5)
         similar_response = rag.generate_response(similar_query)
         print("\nResponse:")
         print("-"*50)
@@ -162,16 +167,19 @@ def setUpClass(cls):
             force_reindex = True,
             query_engine = OpenAIEngine(api_key = cls.api_key),
             hard_grounding_prompt = os.path.join(cls.project_root,
+                                                 'cotarag',
                                                  'accelerag',
                                                  'prompts', 
                                                  'hard_grounding_prompt.txt'),
 
             soft_grounding_prompt=os.path.join(cls.project_root,
+                                               'cotarag',
                                                'accelerag',
                                                'prompts',
                                                'soft_grounding_prompt.txt'),
 
             template_path=os.path.join(cls.project_root,
+                                       'cotarag',
                                        'accelerag',
                                        'web_rag_template.txt')
         )
@@ -216,16 +224,19 @@ def setUpClass(cls):
             force_reindex = True,
             query_engine = AnthropicEngine(api_key=cls.api_key),
             hard_grounding_prompt = os.path.join(cls.project_root,
+                                                 'cotarag',
                                                  'accelerag',
                                                  'prompts',
                                                  'hard_grounding_prompt.txt'),
 
             soft_grounding_prompt = os.path.join(cls.project_root,
+                                                 'cotarag',
                                                  'accelerag',
                                                  'prompts', 
                                                  'soft_grounding_prompt.txt'),
 
             template_path = os.path.join(cls.project_root,
+                                         'cotarag',
                                          'accelerag',
                                          'web_rag_template.txt')
         )