fix bugs in webui

Author: vinci-grape

MANIFEST.in

@@ -1,6 +1,19 @@
-recursive-include ms_agent/tools/code_interpreter *.ttf
-recursive-include ms_agent/utils *.tiktoken
-recursive-include ms_agent/utils/nltk *.zip
-recursive-include ms_agent/ *.yaml
+include README.md
+include LICENSE
+include requirements.txt
+recursive-include requirements *.txt
+
+# Include projects
+recursive-include projects *
+
+# Include webui backend
 recursive-include webui/backend *.py
+
+# Include webui frontend dist (will be built during setup)
 recursive-include webui/frontend/dist *
+
+# Exclude development files
+global-exclude *.pyc
+global-exclude __pycache__
+global-exclude .DS_Store
+global-exclude *.so

ms_agent/cli/run.py

@@ -2,20 +2,44 @@
 import argparse
 import asyncio
 import os
+from importlib import resources as importlib_resources
 
 from ms_agent.config import Config
-from ms_agent.utils import strtobool
+from ms_agent.utils import get_logger, strtobool
 from ms_agent.utils.constants import AGENT_CONFIG_FILE, MS_AGENT_ASCII
 
 from .base import CLICommand
 
+logger = get_logger()
+
 
 def subparser_func(args):
     """ Function which will be called for a specific sub parser.
     """
     return RunCMD(args)
 
 
+def list_builtin_projects():
+    try:
+        root = importlib_resources.files('ms_agent').joinpath('projects')
+        if not root.exists():
+            return []
+        return sorted([p.name for p in root.iterdir() if p.is_dir()])
+    except Exception as e:
+        # Fallback: don't let help crash just because a resource is unavailable.
+        logger.warning(f'Could not list built-in projects: {e}')
+        return []
+
+
+def project_help_text():
+    projects = list_builtin_projects()
+    if projects:
+        return (
+            'Built-in bundled project name under package ms_agent/projects. '
+            f'Available: {", ".join(projects)}')
+    return 'Built-in bundled project name under package ms_agent/projects.'
+
+
 class RunCMD(CLICommand):
     name = 'run'
 
@@ -25,6 +49,8 @@ def __init__(self, args):
     @staticmethod
     def define_args(parsers: argparse.ArgumentParser):
         """Define args for run command."""
+        projects = list_builtin_projects()
+
         parser: argparse.ArgumentParser = parsers.add_parser(RunCMD.name)
         parser.add_argument(
             '--query',
@@ -39,6 +65,14 @@ def define_args(parsers: argparse.ArgumentParser):
             type=str,
             default=None,
             help='The directory or the repo id of the config file')
+        parser.add_argument(
+            '--project',
+            required=False,
+            type=str,
+            default=None,
+            choices=projects,
+            help=project_help_text(),
+        )
         parser.add_argument(
             '--trust_remote_code',
             required=False,
@@ -89,6 +123,33 @@ def define_args(parsers: argparse.ArgumentParser):
         parser.set_defaults(func=subparser_func)
 
     def execute(self):
+        if getattr(self.args, 'project', None):
+            if self.args.config:
+                raise ValueError(
+                    'Please specify only one of --config or --project')
+
+            project = self.args.project
+            project_trav = importlib_resources.files('ms_agent').joinpath(
+                'projects', project)
+
+            if not project_trav.exists():
+                projects_root = importlib_resources.files('ms_agent').joinpath(
+                    'projects')
+                available = []
+                if projects_root.exists():
+                    available = [
+                        p.name for p in projects_root.iterdir() if p.is_dir()
+                    ]
+                raise ValueError(
+                    f'Unknown project: {project}. Available: {available}')
+
+            # as_file ensures we get a real filesystem path even if installed as zip
+            with importlib_resources.as_file(project_trav) as project_dir:
+                self.args.config = str(project_dir)
+                return self._execute_with_config()
+        return self._execute_with_config()
+
+    def _execute_with_config(self):
         if not self.args.config:
             current_dir = os.getcwd()
             if os.path.exists(os.path.join(current_dir, AGENT_CONFIG_FILE)):

ms_agent/cli/ui.py

@@ -59,7 +59,7 @@ def execute(self):
 
         if not webui_dir.exists():
             import ms_agent
-            ms_agent_path = Path(ms_agent.__file__).parent.parent
+            ms_agent_path = Path(ms_agent.__file__).parent
             webui_dir = ms_agent_path / 'webui'
 
         if not webui_dir.exists():

ms_agent/config/config.py

@@ -201,6 +201,23 @@ def traverse_config(_config: Union[DictConfig, ListConfig, Any],
                             _config[idx] = extra[value[1:-1]]
 
         traverse_config(config)
+
+        for key, value in extra.items():
+            if '.' in key and not key.startswith('tools.'):
+                parts = key.split('.')
+                current = config
+                # Navigate/create nested structure
+                for i, part in enumerate(parts[:-1]):
+                    if not hasattr(current,
+                                   part) or getattr(current, part) is None:
+                        setattr(current, part, DictConfig({}))
+                    current = getattr(current, part)
+                final_key = parts[-1]
+                if not hasattr(current, final_key) or getattr(
+                        current, final_key) is None:
+                    logger.info(f'Adding new config key: {key}')
+                    setattr(current, final_key, value)
+
         return None
 
     @staticmethod

ms_agent/llm/openai_llm.py

@@ -42,9 +42,10 @@ def __init__(
         self.model: str = config.llm.model
         self.max_continue_runs = getattr(config.llm, 'max_continue_runs',
                                          None) or MAX_CONTINUE_RUNS
-        base_url = base_url or config.llm.openai_base_url or get_service_config(
-            'openai').base_url
-        api_key = api_key or config.llm.openai_api_key
+        base_url = base_url or getattr(
+            config.llm, 'openai_base_url',
+            None) or get_service_config('openai').base_url
+        api_key = api_key or getattr(config.llm, 'openai_api_key', None)
 
         self.client = openai.OpenAI(
             api_key=api_key,