oxygent 1.0.8__py3-none-any.whl

oxygent/__init__.py

@@ -0,0 +1,20 @@
+from dotenv import load_dotenv
+
+from .config import Config
+from .mas import MAS
+from .oxy import Oxy
+from .oxy_factory import OxyFactory
+from .schemas import OxyOutput, OxyRequest, OxyResponse, OxyState
+
+load_dotenv(".env")
+
+__all__ = [
+    "Oxy",
+    "MAS",
+    "OxyState",
+    "OxyRequest",
+    "OxyOutput",
+    "OxyResponse",
+    "OxyFactory",
+    "Config",
+]

oxygent/banner.py

@@ -0,0 +1,54 @@
+oxygent_larry3d = """
+ _____                    ____                   __      
+/\  __`\                 /\  _`\                /\ \__   
+\ \ \/\ \   __  _  __  __\ \ \L\_\     __    ___\ \ ,_\  
+ \ \ \ \ \ /\ \/'\/\ \/\ \\ \ \L_L   /'__`\/' _ `\ \ \/  
+  \ \ \_\ \\/>  </\ \ \_\ \\ \ \/, \/\  __//\ \/\ \ \ \_ 
+   \ \_____\/\_/\_\\/`____ \\ \____/\ \____\ \_\ \_\ \__\
+    \/_____/\//\/_/ `/___/> \\/___/  \/____/\/_/\/_/\/__/
+                       /\___/                            
+                       \/__/                             
+"""
+
+oxygent_slant = """
+   ____             ______           __ 
+  / __ \_  ____  __/ ____/__  ____  / /_
+ / / / / |/_/ / / / / __/ _ \/ __ \/ __/
+/ /_/ />  </ /_/ / /_/ /  __/ / / / /_  
+\____/_/|_|\__, /\____/\___/_/ /_/\__/  
+          /____/                        
+"""
+
+oxygent_standard = """
+  ___              ____            _   
+ / _ \__  ___   _ / ___| ___ _ __ | |_ 
+| | | \ \/ / | | | |  _ / _ \ '_ \| __|
+| |_| |>  <| |_| | |_| |  __/ | | | |_ 
+ \___//_/\_\\__, |\____|\___|_| |_|\__|
+            |___/                      
+"""
+
+oxygent_smslant = """
+  ____            _____         __ 
+ / __ \__ ____ __/ ___/__ ___  / /_
+/ /_/ /\ \ / // / (_ / -_) _ \/ __/
+\____//_\_\\_, /\___/\__/_//_/\__/ 
+          /___/                    
+"""
+
+oxygent_speed = """
+_______               _________           _____ 
+__  __ \___  ______  ___  ____/_____________  /_
+_  / / /_  |/_/_  / / /  / __ _  _ \_  __ \  __/
+/ /_/ /__>  < _  /_/ // /_/ / /  __/  / / / /_  
+\____/ /_/|_| _\__, / \____/  \___//_/ /_/\__/  
+              /____/                           
+"""
+
+oxygent_chunky = """
+ _______               _______               __   
+|       |.--.--.--.--.|     __|.-----.-----.|  |_ 
+|   -   ||_   _|  |  ||    |  ||  -__|     ||   _|
+|_______||__.__|___  ||_______||_____|__|__||____|
+               |_____|                            
+"""

oxygent/chart/__init__.py

@@ -0,0 +1,71 @@
+import importlib
+import os
+
+from oxygent.oxy import FunctionHub
+
+# List of tool modules to import
+tool_modules = [
+    "flow_image_gen_tools",
+    "open_chart_tools",
+    "flowchart_api"
+]
+
+__all__ = []
+
+# Get the current package directory path
+package_dir = os.path.dirname(__file__)
+
+for module_name in tool_modules:
+    module_path = os.path.join(package_dir, f"{module_name}.py")
+
+    # First check if the module file exists
+    if not os.path.exists(module_path):
+        print(
+            f"Warning: Failed to import tool '{module_name}': Module file does not exist, please check '{module_path}'"
+        )
+        globals()[module_name] = None
+        continue
+
+    try:
+        # Dynamically import the module
+        module = importlib.import_module(f".{module_name}", __package__)
+
+        # Filter out all non-private attributes that are instances of FunctionHub
+        function_hub_instances = [
+            attr
+            for attr in dir(module)
+            if not attr.startswith("_")  # Exclude private attributes
+            and isinstance(
+                getattr(module, attr), FunctionHub
+            )  # Only keep FunctionHub instances
+        ]
+
+        if function_hub_instances:
+            # Add eligible instances to the current scope and __all__
+            for attr_name in function_hub_instances:
+                attr_value = getattr(module, attr_name)
+                globals()[attr_name] = attr_value
+                __all__.append(attr_name)
+        else:
+            print(f"Warning: No FunctionHub instances found in module '{module_name}'")
+
+    except ImportError as e:
+        # Catch import errors and extract the missing package name
+        error_msg = str(e)
+        missing_package = None
+
+        if "No module named" in error_msg:
+            parts = error_msg.split("'")
+            if len(parts) >= 2:
+                missing_package = parts[-2]
+
+        # Print a clear prompt message
+        if missing_package and not missing_package.startswith(__package__):
+            print(
+                f"Warning: Failed to import tool '{module_name}': Missing dependency package '{missing_package}', please run: pip install {missing_package}"
+            )
+        else:
+            print(f"Warning: Failed to import tool '{module_name}': {error_msg}")
+
+        # Set the module entry to None to prevent errors in subsequent use
+        globals()[module_name] = None