Merge branch 'main' into u/#4354

python/packages/autogen-studio/autogenstudio/__init__.py

@@ -1,3 +1,18 @@
-from .datamodel import *
+from .database.db_manager import DatabaseManager
+from .datamodel import Agent, AgentConfig, Model, ModelConfig, Team, TeamConfig, Tool, ToolConfig
+from .teammanager import TeamManager
 from .version import __version__
-from .teammanager import *
+
+__all__ = [
+    "Tool",
+    "Model",
+    "DatabaseManager",
+    "Team",
+    "Agent",
+    "ToolConfig",
+    "ModelConfig",
+    "TeamConfig",
+    "AgentConfig",
+    "TeamManager",
+    "__version__",
+]

python/packages/autogen-studio/autogenstudio/cli.py

@@ -15,7 +15,7 @@ def ui(
     host: str = "127.0.0.1",
     port: int = 8081,
     workers: int = 1,
-    reload: Annotated[bool, typer.Option("--reload")] = True,
+    reload: Annotated[bool, typer.Option("--reload")] = False,
     docs: bool = True,
     appdir: str = None,
     database_uri: Optional[str] = None,
@@ -48,11 +48,7 @@ def ui(
         port=port,
         workers=workers,
         reload=reload,
-        reload_excludes=[
-            "**/alembic/*",
-            "**/alembic.ini",
-            "**/versions/*"
-        ] if reload else None
+        reload_excludes=["**/alembic/*", "**/alembic.ini", "**/versions/*"] if reload else None,
     )
 
 

python/packages/autogen-studio/autogenstudio/database/__init__.py

@@ -1,3 +1,3 @@
-from .db_manager import DatabaseManager
-from .component_factory import ComponentFactory, Component
+from .component_factory import Component, ComponentFactory
 from .config_manager import ConfigurationManager
+from .db_manager import DatabaseManager