fix: packaging

MANIFEST.in

@@ -1,2 +1,3 @@
 include README.md
-include LICENSE
+include LICENSE
+recursive-include src/openai_batch_sdk *.py

src/__init__.py

@@ -1,5 +1,5 @@
 from .advanced_batch_processor import init_monitoring
-from .event_handler import EventHandler
+from .openai_batch_sdk.event_handler import EventHandler
 from .notifier import send_notification
 from .result_handler import retrieve_batch_results
 

src/advanced_batch_processor.py

@@ -14,14 +14,14 @@
                                           retrieve_batch_result,
                                           retrieve_batches_results,
                                           submit_batch_job)
-from event_handler import EventHandler
+from src.openai_batch_sdk.event_handler import EventHandler
 from gpt_conversation_handler import Role, create_message
 from jsonl_handler import create_json_line, write_jsonl_file
 from utils.env import load_environment
 from utils.logging import logger, setup_logging
 from utils.project import get_project_root
 
-from .core_batch_processor import (check_batches_results, gen_submit_batch_job,
+from .openai_batch_sdk.core import (check_batches_results, gen_submit_batch_job,
                                    init_monitoring,
                                    retrieve_batches_results_handler)
 

src/main_l1.py

@@ -16,15 +16,15 @@
 import time
 from datetime import datetime
 
-from event_handler import EventHandler
+from src.openai_batch_sdk.event_handler import EventHandler
 from utils.env import load_environment
 from utils.logging import logger, setup_logging
 from utils.project import get_project_root
 
 from jsonl_handler import create_json_line, write_jsonl_file
 from gpt_conversation_handler import Role, create_message
 
-from core_batch_processor import (init_monitoring, retrieve_batches_results_handler, retrieve_batches_results, graceful_shutdown)
+from src.openai_batch_sdk.core import (init_monitoring, retrieve_batches_results_handler, retrieve_batches_results, graceful_shutdown)
 
 async def shutdown(signal, loop, executor):
     print(f"\nReceived exit signal {signal.name}...")

src/main_l2.py

@@ -19,7 +19,7 @@
 from deps.oai.batch_api.batch_api import (check_batch_status,
                                              retrieve_batch_result,
                                              submit_batch_job)
-from event_handler import EventHandler
+from src.openai_batch_sdk.event_handler import EventHandler
 from utils.env import load_environment
 from utils.logging import logger, setup_logging
 from utils.project import get_project_root

src/openai_batch_sdk/__init__.py

@@ -0,0 +1,4 @@
+from .core import init_monitoring, graceful_shutdown, retrieve_batches_results_handler, retrieve_batches_results
+from event_handler import EventHandler
+
+__all__ = ["init_monitoring", "EventHandler", "graceful_shutdown", "retrieve_batches_results_handler", "retrieve_batches_results"]

tests/test_batch_processor.py

@@ -8,7 +8,7 @@
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src')))
 
 from src.advanced_batch_processor import init_monitoring
-from src.event_handler import EventHandler
+from src.openai_batch_sdk.event_handler import EventHandler
 
 
 class TestBatchProcessor(unittest.TestCase):