jarvis-ai-assistant 0.3.2__tar.gz → 0.3.4__tar.gz

{jarvis_ai_assistant-0.3.2/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.3.2
+Version: 0.3.4
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -202,27 +202,66 @@ iex ((New-Object System.Net.WebClient).DownloadString('https://raw.githubusercon
 
 #### 手动安装
 ```bash
+# 1. 克隆仓库
 git clone https://github.com/skyfireitdiy/Jarvis
+
+# 2. 进入项目目录
 cd Jarvis
+
+# 3. 安装项目为可编辑模式
 pip3 install -e .
 ```
+> **提示**: 使用 `-e .` (可编辑模式) 安装后，您对源码的任何修改都会立刻生效，非常适合开发者。
+
 或者从PyPI安装 (可能不是最新版):
 ```bash
 pip3 install jarvis-ai-assistant
 ```
 
+**通过 uv 安装 (推荐)**
+```bash
+# 1. 安装 uv (如果未安装)
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# 2. 克隆仓库
+git clone https://github.com/skyfireitdiy/Jarvis
+
+# 3. 进入项目目录
+cd Jarvis
+
+# 4. 创建虚拟环境并安装
+uv venv
+
+# 安装基础功能
+uv pip install .
+
+# 可选: 安装RAG功能(包含PyTorch等较重的依赖)
+# uv pip install .[rag]
+```
+
+> **提示**: 安装完成后，建议将虚拟环境激活命令添加到您的 shell 配置文件中:
+> - Bash/Zsh: 在 ~/.bashrc 或 ~/.zshrc 中添加 `source /path/to/Jarvis/.venv/bin/activate`
+> - Fish: 在 ~/.config/fish/config.fish 中添加 `source /path/to/Jarvis/.venv/bin/activate.fish`
+
 ### 基本使用
 Jarvis 包含一系列专注于不同任务的工具。以下是主要命令及其快捷方式：
 
 | 命令 | 快捷方式 | 功能描述 |
 |------|----------|----------|
 | `jarvis` | `jvs` | 通用AI代理，适用于多种任务 |
+| `jarvis-agent` | `ja` | AI代理基础功能，处理会话和任务 |
 | `jarvis-code-agent` | `jca` | 专注于代码分析、修改和生成的代码代理 |
+| `jarvis-code-review` | `jcr` | 智能代码审查工具 |
 | `jarvis-git-commit` | `jgc` | 自动化分析代码变更并生成规范的Git提交信息 |
+| `jarvis-git-squash` | `jgs` | Git提交历史整理工具 |
 | `jarvis-platform-manager` | `jpm` | 管理和测试不同的大语言模型平台 |
+| `jarvis-multi-agent` | `jma` | 多智能体协作系统 |
+| `jarvis-tool` | `jt` | 工具管理与调用系统 |
+| `jarvis-methodology` | `jm` | 方法论知识库管理 |
 | `jarvis-rag` | `jrg` | 构建和查询本地化的RAG知识库 |
 | `jarvis-smart-shell` | `jss` | 实验性的智能Shell功能 |
 | `jarvis-stats` | `jst` | 通用统计模块，支持记录和可视化任意指标数据 |
+| `jarvis-memory-organizer` | `jmo` | 记忆管理工具，支持整理、合并、导入导出记忆 |
 
 更多详细用法和参数，请查阅我们的 [**使用指南**](docs/jarvis_book/4.使用指南.md)。
 

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/README.md

@@ -117,27 +117,66 @@ iex ((New-Object System.Net.WebClient).DownloadString('https://raw.githubusercon
 
 #### 手动安装
 ```bash
+# 1. 克隆仓库
 git clone https://github.com/skyfireitdiy/Jarvis
+
+# 2. 进入项目目录
 cd Jarvis
+
+# 3. 安装项目为可编辑模式
 pip3 install -e .
 ```
+> **提示**: 使用 `-e .` (可编辑模式) 安装后，您对源码的任何修改都会立刻生效，非常适合开发者。
+
 或者从PyPI安装 (可能不是最新版):
 ```bash
 pip3 install jarvis-ai-assistant
 ```
 
+**通过 uv 安装 (推荐)**
+```bash
+# 1. 安装 uv (如果未安装)
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# 2. 克隆仓库
+git clone https://github.com/skyfireitdiy/Jarvis
+
+# 3. 进入项目目录
+cd Jarvis
+
+# 4. 创建虚拟环境并安装
+uv venv
+
+# 安装基础功能
+uv pip install .
+
+# 可选: 安装RAG功能(包含PyTorch等较重的依赖)
+# uv pip install .[rag]
+```
+
+> **提示**: 安装完成后，建议将虚拟环境激活命令添加到您的 shell 配置文件中:
+> - Bash/Zsh: 在 ~/.bashrc 或 ~/.zshrc 中添加 `source /path/to/Jarvis/.venv/bin/activate`
+> - Fish: 在 ~/.config/fish/config.fish 中添加 `source /path/to/Jarvis/.venv/bin/activate.fish`
+
 ### 基本使用
 Jarvis 包含一系列专注于不同任务的工具。以下是主要命令及其快捷方式：
 
 | 命令 | 快捷方式 | 功能描述 |
 |------|----------|----------|
 | `jarvis` | `jvs` | 通用AI代理，适用于多种任务 |
+| `jarvis-agent` | `ja` | AI代理基础功能，处理会话和任务 |
 | `jarvis-code-agent` | `jca` | 专注于代码分析、修改和生成的代码代理 |
+| `jarvis-code-review` | `jcr` | 智能代码审查工具 |
 | `jarvis-git-commit` | `jgc` | 自动化分析代码变更并生成规范的Git提交信息 |
+| `jarvis-git-squash` | `jgs` | Git提交历史整理工具 |
 | `jarvis-platform-manager` | `jpm` | 管理和测试不同的大语言模型平台 |
+| `jarvis-multi-agent` | `jma` | 多智能体协作系统 |
+| `jarvis-tool` | `jt` | 工具管理与调用系统 |
+| `jarvis-methodology` | `jm` | 方法论知识库管理 |
 | `jarvis-rag` | `jrg` | 构建和查询本地化的RAG知识库 |
 | `jarvis-smart-shell` | `jss` | 实验性的智能Shell功能 |
 | `jarvis-stats` | `jst` | 通用统计模块，支持记录和可视化任意指标数据 |
+| `jarvis-memory-organizer` | `jmo` | 记忆管理工具，支持整理、合并、导入导出记忆 |
 
 更多详细用法和参数，请查阅我们的 [**使用指南**](docs/jarvis_book/4.使用指南.md)。
 

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.3.2"
+version = "0.3.4"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -80,6 +80,8 @@ jarvis-platform-manager = "jarvis.jarvis_platform_manager.main:main"
 jpm = "jarvis.jarvis_platform_manager.main:main"
 jarvis-git-commit = "jarvis.jarvis_git_utils.git_commiter:main"
 jarvis-code-review = "jarvis.jarvis_code_analysis.code_review:main"
+jarvis-memory-organizer = "jarvis.jarvis_memory_organizer.memory_organizer:main"
+jmo = "jarvis.jarvis_memory_organizer.memory_organizer:main"
 jcr = "jarvis.jarvis_code_analysis.code_review:main"
 jgc = "jarvis.jarvis_git_utils.git_commiter:main"
 jarvis-git-squash = "jarvis.jarvis_git_squash.main:main"
@@ -97,10 +99,16 @@ jrg = "jarvis.jarvis_rag.cli:main"
 jarvis-stats = "jarvis.jarvis_stats.cli:main"
 jst = "jarvis.jarvis_stats.cli:main"
 
-[[tool.uv.index]]
-url = "https://artsz.zte.com.cn/artifactory/api/pypi/public-pypi-virtual/simple"
-default = true
+
 
 [[tool.uv.index]]
 name = "tsinghua"
 url = "https://pypi.tuna.tsinghua.edu.cn/simple"
+
+[[tool.uv.index]]
+name = "aliyun"
+url = "https://mirrors.aliyun.com/pypi/simple/"
+
+[[tool.uv.index]]
+name = "ustc"
+url = "https://pypi.mirrors.ustc.edu.cn/simple/"

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages  # type: ignore
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.3.2",
+    version="0.3.4",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -75,6 +75,8 @@ setup(
             "ja=jarvis.jarvis_agent.main:main",
             "jarvis-tool=jarvis.jarvis_tools.cli.main:main",
             "jt=jarvis.jarvis_tools.cli.main:main",
+            "jarvis-memory-organizer=jarvis.jarvis_memory_organizer.memory_organizer:main",
+            "jmo=jarvis.jarvis_memory_organizer.memory_organizer:main",
             "jarvis-methodology=jarvis.jarvis_methodology.main:main",
             "jm=jarvis.jarvis_methodology.main:main",
             "jarvis-rag=jarvis.jarvis_rag.cli:main",

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.3.2"
+__version__ = "0.3.4"

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/jarvis_agent/__init__.py

@@ -3,6 +3,7 @@
 import datetime
 import os
 import platform
+from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Protocol, Tuple, Union
 
 # 第三方库导入
@@ -14,6 +15,7 @@ from jarvis.jarvis_agent.protocols import OutputHandlerProtocol
 from jarvis.jarvis_agent.session_manager import SessionManager
 from jarvis.jarvis_agent.tool_executor import execute_tool_call
 from jarvis.jarvis_agent.memory_manager import MemoryManager
+from jarvis.jarvis_memory_organizer.memory_organizer import MemoryOrganizer
 from jarvis.jarvis_agent.task_analyzer import TaskAnalyzer
 from jarvis.jarvis_agent.file_methodology_manager import FileMethodologyManager
 from jarvis.jarvis_agent.prompts import (
@@ -28,6 +30,7 @@ from jarvis.jarvis_platform.registry import PlatformRegistry
 
 # jarvis_utils 相关
 from jarvis.jarvis_utils.config import (
+    get_data_dir,
     get_max_token_count,
     get_normal_model_name,
     get_normal_platform_name,
@@ -50,6 +53,98 @@ from jarvis.jarvis_utils.input import get_multiline_input, user_confirm
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 
+
+def show_agent_startup_stats(agent_name: str, model_name: str) -> None:
+    """输出启动时的统计信息
+
+    参数:
+        agent_name: Agent的名称
+        model_name: 使用的模型名称
+    """
+    try:
+        from jarvis.jarvis_utils.methodology import _load_all_methodologies
+        from jarvis.jarvis_tools.registry import ToolRegistry
+        from jarvis.jarvis_utils.config import get_data_dir
+        from pathlib import Path
+        from rich.console import Console
+        from rich.panel import Panel
+        from rich.text import Text
+        from rich.align import Align
+        import os
+
+        methodologies = _load_all_methodologies()
+        methodology_count = len(methodologies)
+
+        # 获取工具数量
+        # 创建一个临时的工具注册表类来获取所有工具（不应用过滤）
+        class TempToolRegistry(ToolRegistry):
+            def _apply_tool_config_filter(self) -> None:
+                """重写过滤方法，不执行任何过滤"""
+                pass
+
+        # 获取所有工具的数量
+        tool_registry_all = TempToolRegistry()
+        total_tool_count = len(tool_registry_all.tools)
+
+        # 获取可用工具的数量（应用过滤）
+        tool_registry = ToolRegistry()
+        available_tool_count = len(tool_registry.get_all_tools())
+
+        global_memory_dir = Path(get_data_dir()) / "memory" / "global_long_term"
+        global_memory_count = 0
+        if global_memory_dir.exists():
+            global_memory_count = len(list(global_memory_dir.glob("*.json")))
+
+        # 检查项目记忆
+        project_memory_dir = Path(".jarvis/memory")
+        project_memory_count = 0
+        if project_memory_dir.exists():
+            project_memory_count = len(list(project_memory_dir.glob("*.json")))
+
+        # 获取当前工作目录
+        current_dir = os.getcwd()
+
+        # 构建欢迎信息
+        welcome_message = f"{agent_name} 初始化完成 - 使用 {model_name} 模型"
+
+        stats_parts = [
+            f"📚  本地方法论: [bold cyan]{methodology_count}[/bold cyan]",
+            f"🛠️  工具: [bold green]{available_tool_count}/{total_tool_count}[/bold green] (可用/全部)",
+            f"🧠  全局记忆: [bold yellow]{global_memory_count}[/bold yellow]",
+        ]
+
+        # 如果有项目记忆，添加到统计信息中
+        if project_memory_count > 0:
+            stats_parts.append(
+                f"📝  项目记忆: [bold magenta]{project_memory_count}[/bold magenta]"
+            )
+
+        stats_text = Text.from_markup(" | ".join(stats_parts), justify="center")
+
+        # 创建包含欢迎信息和统计信息的面板内容
+        panel_content = Text()
+        panel_content.append(welcome_message, style="bold white")
+        panel_content.append("\n")
+        panel_content.append(f"📁  工作目录: {current_dir}", style="dim white")
+        panel_content.append("\n\n")
+        panel_content.append(stats_text)
+        panel_content.justify = "center"
+
+        panel = Panel(
+            panel_content,
+            title="✨ Jarvis 资源概览 ✨",
+            title_align="center",
+            border_style="blue",
+            expand=False,
+        )
+
+        console = Console()
+        console.print(Align.center(panel))
+
+    except Exception as e:
+        PrettyOutput.print(f"加载统计信息失败: {e}", OutputType.WARNING)
+
+
 origin_agent_system_prompt = f"""
 <role>
 # 🤖 角色
@@ -175,9 +270,8 @@ class Agent:
         # 设置系统提示词
         self._setup_system_prompt()
 
-        # 打印欢迎信息
-        welcome_message = f"{name} 初始化完成 - 使用 {self.model.name()} 模型"  # type: ignore
-        PrettyOutput.print(welcome_message, OutputType.SYSTEM)
+        # 输出统计信息（包含欢迎信息）
+        show_agent_startup_stats(name, self.model.name())  # type: ignore
 
     def _init_model(self, llm_type: str, model_group: Optional[str]):
         """初始化模型平台"""
@@ -190,7 +284,9 @@ class Agent:
 
         self.model = PlatformRegistry().create_platform(platform_name)
         if self.model is None:
-            PrettyOutput.print(f"平台 {platform_name} 不存在，将使用普通模型", OutputType.WARNING)
+            PrettyOutput.print(
+                f"平台 {platform_name} 不存在，将使用普通模型", OutputType.WARNING
+            )
             self.model = PlatformRegistry().get_normal_platform()
 
         if model_name:
@@ -512,10 +608,8 @@ class Agent:
 
         if self.use_analysis:
             self.task_analyzer.analysis_task(satisfaction_feedback)
-        else:
-            # 如果没有开启分析，也提示用户是否有值得记忆的信息
-            if self.force_save_memory:
-                self.memory_manager.prompt_memory_save()
+
+        self._check_and_organize_memory()
 
         if self.need_summary:
             print("📄 正在生成总结...")
@@ -689,7 +783,9 @@ class Agent:
         返回:
             str: "continue" 或 "complete"
         """
-        user_input = self.multiline_inputer(f"{self.name}: 请输入，或输入空行来结束当前任务：")
+        user_input = self.multiline_inputer(
+            f"{self.name}: 请输入，或输入空行来结束当前任务："
+        )
 
         if user_input:
             self.session.prompt = user_input
@@ -711,6 +807,71 @@ class Agent:
 
         self.first = False
 
+    def _check_and_organize_memory(self):
+        """
+        检查记忆库状态，如果满足条件则提示用户整理。
+        每天只检测一次。
+        """
+        try:
+            # 检查项目记忆
+            self._perform_memory_check("project_long_term", Path(".jarvis"), "project")
+            # 检查全局记忆
+            self._perform_memory_check(
+                "global_long_term",
+                Path(get_data_dir()),
+                "global",
+            )
+        except Exception as e:
+            PrettyOutput.print(f"检查记忆库时发生意外错误: {e}", OutputType.WARNING)
+
+    def _perform_memory_check(self, memory_type: str, base_path: Path, scope_name: str):
+        """执行特定范围的记忆检查和整理"""
+        check_file = base_path / ".last_memory_organizer_check"
+        now = datetime.datetime.now()
+
+        if check_file.exists():
+            try:
+                last_check_time = datetime.datetime.fromisoformat(
+                    check_file.read_text()
+                )
+                if (now - last_check_time).total_seconds() < 24 * 3600:
+                    return  # 24小时内已检查
+            except (ValueError, FileNotFoundError):
+                # 文件内容无效或文件在读取时被删除，继续执行检查
+                pass
+
+        # 立即更新检查时间，防止并发或重复检查
+        base_path.mkdir(parents=True, exist_ok=True)
+        check_file.write_text(now.isoformat())
+
+        organizer = MemoryOrganizer()
+        # NOTE: 使用受保护方法以避免重复实现逻辑
+        memories = organizer._load_memories(memory_type)
+
+        if len(memories) < 200:
+            return
+
+        # NOTE: 使用受保护方法以避免重复实现逻辑
+        overlap_groups = organizer._find_overlapping_memories(memories, min_overlap=3)
+        has_significant_overlap = any(groups for groups in overlap_groups.values())
+
+        if not has_significant_overlap:
+            return
+
+        prompt = (
+            f"检测到您的 '{scope_name}' 记忆库中包含 {len(memories)} 条记忆，"
+            f"并且存在3个以上标签重叠的记忆。\n"
+            f"是否立即整理记忆库以优化性能和相关性？"
+        )
+        if user_confirm(prompt, default=True):
+            PrettyOutput.print(
+                f"正在开始整理 '{scope_name}' ({memory_type}) 记忆库...",
+                OutputType.INFO,
+            )
+            organizer.organize_memories(memory_type, min_overlap=3)
+        else:
+            PrettyOutput.print(f"已取消 '{scope_name}' 记忆库整理。", OutputType.INFO)
+
     def clear_history(self):
         """
         Clears conversation history by delegating to the session manager.

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/jarvis_agent/file_methodology_manager.py

@@ -80,7 +80,12 @@ class FileMethodologyManager:
 
         memory_manager = MemoryManager(self.agent)
         memory_tags_prompt = memory_manager.prepare_memory_tags_prompt()
-        methodology = load_methodology(msg, self.agent.get_tool_registry())
+        methodology = load_methodology(
+            msg,
+            self.agent.get_tool_registry(),
+            platform_name=self.agent.model.platform_name(),
+            model_name=self.agent.model.name(),
+        )
         self.agent.session.prompt = f"{self.agent.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{methodology}{memory_tags_prompt}"
 
     def handle_history_with_file_upload(self) -> str:

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/jarvis_agent/share_manager.py

@@ -98,6 +98,27 @@ class ShareManager(ABC):
                 )
                 # 如果有远程分支，执行pull
                 if result.stdout.strip():
+                    # 检查是否有未提交的更改
+                    status_result = subprocess.run(
+                        ["git", "status", "--porcelain"],
+                        cwd=self.repo_path,
+                        capture_output=True,
+                        text=True,
+                        check=True,
+                    )
+                    if status_result.stdout:
+                        if user_confirm(
+                            f"检测到中心{self.get_resource_type()}仓库 '{self.repo_name}' 存在未提交的更改，是否放弃这些更改并更新？"
+                        ):
+                            subprocess.run(
+                                ["git", "checkout", "."], cwd=self.repo_path, check=True
+                            )
+                        else:
+                            PrettyOutput.print(
+                                f"跳过更新 '{self.repo_name}' 以保留未提交的更改。",
+                                OutputType.INFO,
+                            )
+                            return
                     subprocess.run(["git", "pull"], cwd=self.repo_path, check=True)
                 else:
                     PrettyOutput.print(

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/jarvis_agent/tool_executor.py

@@ -41,9 +41,13 @@ def execute_tool_call(response: str, agent: "Agent") -> Tuple[bool, Any]:
     if not agent.execute_tool_confirm or user_confirm(
         f"需要执行{tool_to_execute.name()}确认执行？", True
     ):
-        print(f"🔧 正在执行{tool_to_execute.name()}...")
-        result = tool_to_execute.handle(response, agent)
-        print(f"✅ {tool_to_execute.name()}执行完成")
-        return result
+        try:
+            print(f"🔧 正在执行{tool_to_execute.name()}...")
+            result = tool_to_execute.handle(response, agent)
+            print(f"✅ {tool_to_execute.name()}执行完成")
+            return result
+        except Exception as e:
+            PrettyOutput.print(f"工具执行失败: {str(e)}", OutputType.ERROR)
+            return False, str(e)
 
     return False, ""

{jarvis_ai_assistant-0.3.2 → jarvis_ai_assistant-0.3.4}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -398,11 +398,6 @@ class CodeAgent:
             if not user_confirm("是否要提交？", True):
                 return
 
-            # 用户确认修改，统计修改次数
-            from jarvis.jarvis_stats.stats import StatsManager
-
-            StatsManager.increment("code_modification_confirmed", group="code_agent")
-
             try:
                 confirm_add_new_files()
 
@@ -431,9 +426,6 @@ class CodeAgent:
                     ["git", "commit", "-m", f"CheckPoint #{commit_count + 1}"],
                     check=True,
                 )
-
-                # 统计提交次数
-                StatsManager.increment("code_commits_accepted", group="code_agent")
             except subprocess.CalledProcessError as e:
                 PrettyOutput.print(f"提交失败: {str(e)}", OutputType.ERROR)
 
@@ -484,6 +476,10 @@ class CodeAgent:
             )
             git_commiter = GitCommitTool()
             git_commiter.execute({})
+
+            # 在用户接受commit后，根据配置决定是否保存记忆
+            if self.agent.force_save_memory:
+                self.agent.memory_manager.prompt_memory_save()
         elif start_commit:
             os.system(f"git reset --hard {str(start_commit)}")  # 确保转换为字符串
             PrettyOutput.print("已重置到初始提交", OutputType.INFO)
@@ -642,7 +638,8 @@ class CodeAgent:
 def cli(
     llm_type: str = typer.Option(
         "normal",
-        "-t", "--llm_type",
+        "-t",
+        "--llm_type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
@@ -695,9 +692,7 @@ def cli(
             sys.exit(0)
 
     curr_dir = os.getcwd()
-    git_dir = find_git_root_and_cd(curr_dir)
-    PrettyOutput.print(f"当前目录: {git_dir}", OutputType.INFO)
-
+    find_git_root_and_cd(curr_dir)
     try:
         agent = CodeAgent(
             llm_type=llm_type,