From 14e2b89b810172bcb2fe627a360266c47f11ee6b Mon Sep 17 00:00:00 2001 From: Asankhaya Sharma Date: Mon, 4 Nov 2024 16:47:13 +0800 Subject: [PATCH] Update init file to expose methods --- optillm/__init__.py | 46 +++++++++++++++++++++++++++++++++++++++++++-- setup.py | 2 +- 2 files changed, 45 insertions(+), 3 deletions(-) diff --git a/optillm/__init__.py b/optillm/__init__.py index 26177f9..ed380a9 100644 --- a/optillm/__init__.py +++ b/optillm/__init__.py @@ -9,5 +9,47 @@ module = util.module_from_spec(spec) spec.loader.exec_module(module) -# Export the main function -main = module.main \ No newline at end of file +# Export the main entry point +main = module.main + +# Export the core configuration and server components +server_config = module.server_config +app = module.app +known_approaches = module.known_approaches +plugin_approaches = module.plugin_approaches + +# Export utility functions +parse_combined_approach = module.parse_combined_approach +parse_conversation = module.parse_conversation +extract_optillm_approach = module.extract_optillm_approach +get_config = module.get_config +load_plugins = module.load_plugins + +# Export execution functions +execute_single_approach = module.execute_single_approach +execute_combined_approaches = module.execute_combined_approaches +execute_parallel_approaches = module.execute_parallel_approaches + +# Export streaming response generation +generate_streaming_response = module.generate_streaming_response + +# Version information +__version__ = "0.0.8" # Match with setup.py + +# List of exported symbols +__all__ = [ + 'main', + 'server_config', + 'app', + 'known_approaches', + 'plugin_approaches', + 'parse_combined_approach', + 'parse_conversation', + 'extract_optillm_approach', + 'get_config', + 'load_plugins', + 'execute_single_approach', + 'execute_combined_approaches', + 'execute_parallel_approaches', + 'generate_streaming_response', +] \ No newline at end of file diff --git a/setup.py b/setup.py index 7d91dfe..c90ea4b 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="optillm", - version="0.0.7", + version="0.0.8", packages=find_packages(), py_modules=['optillm'], package_data={