move litewrapper

codelion · Oct 24, 2024 · ba2eebe · ba2eebe
1 parent 52d67d5
commit ba2eebe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/optillm.py b/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

diff --git a/litellm_wrapper.py → optillm/litellm_wrapper.py b/litellm_wrapper.py → optillm/litellm_wrapper.py