Merge pull request #78 from codelion/fix-init-py

Fix init py

Author: codelion

optillm.py

@@ -13,7 +13,7 @@
 from concurrent.futures import ThreadPoolExecutor
 
 # Import the LiteLLM wrapper
-from litellm_wrapper import LiteLLMWrapper
+from optillm.litellm_wrapper import LiteLLMWrapper
 
 # Import approach modules
 from optillm.mcts import chat_with_mcts

optillm/__init__.py

@@ -0,0 +1,13 @@
+from importlib import util
+import os
+
+# Get the path to the root optillm.py
+spec = util.spec_from_file_location(
+    "optillm.root",
+    os.path.join(os.path.dirname(os.path.dirname(__file__)), "optillm.py")
+)
+module = util.module_from_spec(spec)
+spec.loader.exec_module(module)
+
+# Export the main function
+main = module.main

litellm_wrapper.py renamed to optillm/litellm_wrapper.py

@@ -2,7 +2,7 @@
 
 setup(
     name="optillm",
-    version="0.0.4",
+    version="0.0.5",
     packages=find_packages(),
     py_modules=['optillm'],
     install_requires=[