jarvis-ai-assistant 0.1.111__py3-none-any.whl → 0.1.112__py3-none-any.whl

jarvis/jarvis_platform/kimi.py

@@ -26,24 +26,24 @@ class KimiModel(BasePlatform):
         self.api_key = os.getenv("KIMI_API_KEY")
         if not self.api_key:
             message = (
-                "Need to set KIMI_API_KEY to use Jarvis. Please follow the steps below:\n"
-                "1. Get Kimi API Key:\n"
-                "   • Visit Kimi AI platform: https://kimi.moonshot.cn\n"
-                "   • Login to your account\n" 
-                "   • Open browser developer tools (F12 or right-click -> Inspect)\n"
-                "   • Switch to the Network tab\n"
-                "   • Send any message\n"
-                "   • Find the Authorization header in the request\n"
-                "   • Copy the token value (remove the 'Bearer ' prefix)\n"
-                "2. Set environment variable:\n"
-                "   • Method 1: Create or edit ~/.jarvis/env file:\n"
+                "需要设置 KIMI_API_KEY 才能使用 Jarvis。请按照以下步骤操作：\n"
+                "1. 获取 Kimi API Key:\n"
+                "   • 访问 Kimi AI 平台: https://kimi.moonshot.cn\n"
+                "   • 登录您的账户\n" 
+                "   • 打开浏览器开发者工具 (F12 或右键 -> 检查)\n"
+                "   • 切换到网络标签\n"
+                "   • 发送任意消息\n"
+                "   • 在请求中找到 Authorization 头\n"
+                "   • 复制 token 值（去掉 'Bearer ' 前缀）\n"
+                "2. 设置环境变量:\n"
+                "   • 方法 1: 创建或编辑 ~/.jarvis/env 文件:\n"
                 "   echo 'KIMI_API_KEY=your_key_here' > ~/.jarvis/env\n"
-                "   • Method 2: Set environment variable directly:\n"
+                "   • 方法 2: 直接设置环境变量:\n"
                 "   export KIMI_API_KEY=your_key_here\n"
-                "After setting, run Jarvis again."
+                "设置后，重新运行 Jarvis。"
             )
             PrettyOutput.print(message, OutputType.INFO)
-            PrettyOutput.print("KIMI_API_KEY is not set", OutputType.WARNING)
+            PrettyOutput.print("KIMI_API_KEY 未设置", OutputType.WARNING)
         self.auth_header = f"Bearer {self.api_key}"
         self.chat_id = ""
         self.uploaded_files = []  # 存储已上传文件的信息
@@ -75,7 +75,7 @@ class KimiModel(BasePlatform):
             self.chat_id = response.json()["id"]
             return True
         except Exception as e:
-            PrettyOutput.print(f"Error: Failed to create chat: {e}", OutputType.ERROR)
+            PrettyOutput.print(f"错误：创建会话失败：{e}", OutputType.ERROR)
             return False
 
     def _get_presigned_url(self, filename: str, action: str) -> Dict:
@@ -105,7 +105,7 @@ class KimiModel(BasePlatform):
                 response = while_success(lambda: requests.put(presigned_url, data=content), sleep_time=5)
                 return response.status_code == 200
         except Exception as e:
-            PrettyOutput.print(f"Error: Failed to upload file: {e}", OutputType.ERROR)
+            PrettyOutput.print(f"错误：上传文件失败：{e}", OutputType.ERROR)
             return False
 
     def _get_file_info(self, file_data: Dict, name: str, file_type: str) -> Dict:
@@ -202,7 +202,7 @@ class KimiModel(BasePlatform):
                     else:
                         uploaded_files.append(file_info)
                 else:
-                    PrettyOutput.print(f"Error: 文件上传失败: {file_path}", OutputType.ERROR)
+                    PrettyOutput.print(f"错误：文件上传失败: {file_path}", OutputType.ERROR)
                     
             except Exception as e:
                 PrettyOutput.print(f"✗ 处理文件出错 {file_path}: {str(e)}", OutputType.ERROR)

jarvis/jarvis_platform/ollama.py

@@ -26,21 +26,21 @@ class OllamaPlatform(BasePlatform):
             
             if not available_models:
                 message = (
-                    "Need to download Ollama model first to use:\n"
-                    "1. Install Ollama: https://ollama.ai\n"
-                    "2. Download model:\n"
+                    "需要先下载 Ollama 模型才能使用:\n"
+                    "1. 安装 Ollama: https://ollama.ai\n"
+                    "2. 下载模型:\n"
                     f"   ollama pull {self.model_name}"
                 )
                 PrettyOutput.print(message, OutputType.INFO)
-                PrettyOutput.print("Ollama has no available models", OutputType.WARNING)
+                PrettyOutput.print("Ollama 没有可用的模型", OutputType.WARNING)
                 
         except requests.exceptions.ConnectionError:
             message = (
-                "Ollama service is not started or cannot be connected\n"
-                "Please ensure that you have:\n"
-                "1. Installed Ollama: https://ollama.ai\n"
-                "2. Started Ollama service\n"
-                "3. Service address configured correctly (default: http://localhost:11434)"
+                "Ollama 服务未启动或无法连接\n"
+                "请确保您已:\n"
+                "1. 安装 Ollama: https://ollama.ai\n"
+                "2. 启动 Ollama 服务\n"
+                "3. 正确配置服务地址 (默认: http://localhost:11434)"
             )
             PrettyOutput.print(message, OutputType.WARNING)
             
@@ -108,12 +108,12 @@ class OllamaPlatform(BasePlatform):
             return full_response
             
         except Exception as e:
-            PrettyOutput.print(f"Chat failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"对话失败: {str(e)}", OutputType.ERROR)
             raise Exception(f"Chat failed: {str(e)}")
 
     def upload_files(self, file_list: List[str]) -> List[Dict]:
         """Upload files (Ollama does not support file upload)"""
-        PrettyOutput.print("Ollama does not support file upload", output_type=OutputType.WARNING)
+        PrettyOutput.print("Ollama 不支持文件上传", output_type=OutputType.WARNING)
         return []
         
     def reset(self):
@@ -142,10 +142,10 @@ if __name__ == "__main__":
         ollama = OllamaPlatform()
         while True:
             try:
-                message = get_single_line_input("Input question (Ctrl+C to exit)")
+                message = get_single_line_input("输入问题 (Ctrl+C 退出)")
                 ollama.chat_until_success(message)
             except KeyboardInterrupt:
-                print("Goodbye!")
+                print("再见!")
                 break
     except Exception as e:
-        PrettyOutput.print(f"Program exited with an exception: {str(e)}", OutputType.ERROR)
+        PrettyOutput.print(f"程序异常退出: {str(e)}", OutputType.ERROR)

jarvis/jarvis_platform/openai.py

@@ -20,21 +20,21 @@ class OpenAIModel(BasePlatform):
         self.api_key = os.getenv("OPENAI_API_KEY")
         if not self.api_key:
             message = (
-                "Need to set the following environment variables to use OpenAI model:\n"
-                "  • OPENAI_API_KEY: API key\n" 
-                "  • OPENAI_API_BASE: (optional) API base address, default using https://api.openai.com/v1\n"
-                "You can set them in the following ways:\n"
-                "1. Create or edit ~/.jarvis/env file:\n"
+                "需要设置以下环境变量才能使用 OpenAI 模型:\n"
+                "  • OPENAI_API_KEY: API 密钥\n" 
+                "  • OPENAI_API_BASE: (可选) API 基础地址, 默认使用 https://api.openai.com/v1\n"
+                "您可以通过以下方式设置它们:\n"
+                "1. 创建或编辑 ~/.jarvis/env 文件:\n"
                 "   OPENAI_API_KEY=your_api_key\n"
                 "   OPENAI_API_BASE=your_api_base\n" 
                 "   OPENAI_MODEL_NAME=your_model_name\n"
-                "2. Or set the environment variables directly:\n"
+                "2. 直接设置环境变量:\n"
                 "   export OPENAI_API_KEY=your_api_key\n"
                 "   export OPENAI_API_BASE=your_api_base\n"
                 "   export OPENAI_MODEL_NAME=your_model_name"
             )
             PrettyOutput.print(message, OutputType.INFO)
-            PrettyOutput.print("OPENAI_API_KEY is not set", OutputType.WARNING)
+            PrettyOutput.print("OPENAI_API_KEY 未设置", OutputType.WARNING)
             
         self.base_url = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
         self.model_name =  os.getenv("JARVIS_MODEL") or "gpt-4o"
@@ -92,7 +92,7 @@ class OpenAIModel(BasePlatform):
             return full_response
             
         except Exception as e:
-            PrettyOutput.print(f"Chat failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"对话失败：{str(e)}", OutputType.ERROR)
             raise Exception(f"Chat failed: {str(e)}")
 
     def name(self) -> str:

jarvis/jarvis_platform/oyi.py

@@ -29,11 +29,11 @@ class OyiModel(BasePlatform):
         
         self.token = os.getenv("OYI_API_KEY")
         if not self.token:
-            PrettyOutput.print("OYI_API_KEY is not set", OutputType.WARNING)
+            PrettyOutput.print("OYI_API_KEY 未设置", OutputType.WARNING)
         
         self.model_name = os.getenv("JARVIS_MODEL") or "deepseek-chat"
         if self.model_name not in [m.split()[0] for m in self.get_available_models()]:
-            PrettyOutput.print(f"Warning: The selected model {self.model_name} is not in the available list", OutputType.WARNING)
+            PrettyOutput.print(f"警告: 选择的模型 {self.model_name} 不在可用列表中", OutputType.WARNING)
         
 
     def set_model_name(self, model_name: str):
@@ -83,14 +83,14 @@ class OyiModel(BasePlatform):
                     self.conversation = data
                     return True
                 else:
-                    PrettyOutput.print(f"Create conversation failed: {data['message']}", OutputType.ERROR)
+                    PrettyOutput.print(f"创建会话失败: {data['message']}", OutputType.ERROR)
                     return False
             else:
-                PrettyOutput.print(f"Create conversation failed: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"创建会话失败: {response.status_code}", OutputType.ERROR)
                 return False
                 
         except Exception as e:
-            PrettyOutput.print(f"Create conversation failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"创建会话失败: {str(e)}", OutputType.ERROR)
             return False
     
     def set_system_message(self, message: str):
@@ -156,13 +156,13 @@ class OyiModel(BasePlatform):
             )
             
             if response.status_code != 200:
-                error_msg = f"Chat request failed: {response.status_code}"
+                error_msg = f"聊天请求失败: {response.status_code}"
                 PrettyOutput.print(error_msg, OutputType.ERROR)
                 raise Exception(error_msg)
             
             data = response.json()
             if data['code'] != 200 or data['type'] != 'success':
-                error_msg = f"Chat failed: {data.get('message', 'Unknown error')}"
+                error_msg = f"聊天失败: {data.get('message', '未知错误')}"
                 PrettyOutput.print(error_msg, OutputType.ERROR)
                 raise Exception(error_msg)
             
@@ -195,11 +195,11 @@ class OyiModel(BasePlatform):
                 self.messages.append({"role": "assistant", "content": full_response})
                 return full_response
             else:
-                error_msg = f"Get response failed: {response.status_code}"
+                error_msg = f"获取响应失败: {response.status_code}"
                 PrettyOutput.print(error_msg, OutputType.ERROR)
                 raise Exception(error_msg)
         except Exception as e:
-            PrettyOutput.print(f"Chat failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"聊天失败: {str(e)}", OutputType.ERROR)
             raise e
             
     def name(self) -> str:
@@ -240,16 +240,16 @@ class OyiModel(BasePlatform):
                     self.reset()
                     return True
                 else:
-                    error_msg = f"Delete conversation failed: {data.get('message', 'Unknown error')}"
+                    error_msg = f"删除会话失败: {data.get('message', '未知错误')}"
                     PrettyOutput.print(error_msg, OutputType.ERROR)
                     return False
             else:
-                error_msg = f"Delete conversation request failed: {response.status_code}"
+                error_msg = f"删除会话请求失败: {response.status_code}"
                 PrettyOutput.print(error_msg, OutputType.ERROR)
                 return False
             
         except Exception as e:
-            PrettyOutput.print(f"Delete conversation failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"删除会话失败: {str(e)}", OutputType.ERROR)
             return False
     
     def upload_files(self, file_list: List[str]) -> List[Dict]:
@@ -265,7 +265,7 @@ class OyiModel(BasePlatform):
             # 检查当前模型是否支持文件上传
             model_info = self.models.get(self.model_name)
             if not model_info or not model_info.get('uploadFile', False):
-                PrettyOutput.print(f"The current model {self.model_name} does not support file upload", OutputType.WARNING)
+                PrettyOutput.print(f"当前模型 {self.model_name} 不支持文件上传", OutputType.WARNING)
                 return []
             
             headers = {
@@ -281,7 +281,7 @@ class OyiModel(BasePlatform):
                 # 检查文件类型
                 file_type = mimetypes.guess_type(file_path)[0]
                 if not file_type or not file_type.startswith(('image/', 'text/', 'application/')):
-                    PrettyOutput.print(f"The file type {file_type} is not supported", OutputType.ERROR)
+                    PrettyOutput.print(f"文件类型 {file_type} 不支持", OutputType.ERROR)
                     continue
                 
                 with open(file_path, 'rb') as f:
@@ -300,15 +300,15 @@ class OyiModel(BasePlatform):
                         if data.get('code') == 200:
                             self.files.append(data)
                         else:
-                            PrettyOutput.print(f"File upload failed: {data.get('message')}", OutputType.ERROR)
+                            PrettyOutput.print(f"文件上传失败: {data.get('message')}", OutputType.ERROR)
                             return []
                     else:
-                        PrettyOutput.print(f"File upload failed: {response.status_code}", OutputType.ERROR)
+                        PrettyOutput.print(f"文件上传失败: {response.status_code}", OutputType.ERROR)
                         return []
                 
             return self.files
         except Exception as e:
-            PrettyOutput.print(f"File upload failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"文件上传失败: {str(e)}", OutputType.ERROR)
             return []
 
     def get_available_models(self) -> List[str]:
@@ -335,7 +335,7 @@ class OyiModel(BasePlatform):
             )
             
             if response.status_code != 200:
-                PrettyOutput.print(f"Get model list failed: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"获取模型列表失败: {response.status_code}", OutputType.ERROR)
                 return []
             
             data = response.json()
@@ -379,5 +379,5 @@ class OyiModel(BasePlatform):
             return sorted(models)
             
         except Exception as e:
-            PrettyOutput.print(f"Get model list failed: {str(e)}", OutputType.WARNING)
+            PrettyOutput.print(f"获取模型列表失败: {str(e)}", OutputType.WARNING)
             return []

jarvis/jarvis_platform/registry.py

@@ -36,7 +36,7 @@ class PlatformRegistry:
 
                 pass
             except Exception as e:
-                PrettyOutput.print(f"Create platform directory failed: {str(e)}", OutputType.ERROR)
+                PrettyOutput.print(f"创建平台目录失败: {str(e)}", OutputType.ERROR)
                 return ""
         return user_platform_dir
 
@@ -71,7 +71,7 @@ class PlatformRegistry:
         
         if missing_methods:
             PrettyOutput.print(
-                f"Platform {platform_class.__name__} is missing necessary methods: {', '.join(missing_methods)}", 
+                f"平台 {platform_class.__name__} 缺少必要的方法: {', '.join(missing_methods)}", 
                 OutputType.ERROR
             )
             return False
@@ -92,7 +92,7 @@ class PlatformRegistry:
         
         # 确保目录存在
         if not os.path.exists(directory):
-            PrettyOutput.print(f"Platform directory does not exist: {directory}", OutputType.ERROR)
+            PrettyOutput.print(f"平台目录不存在: {directory}", OutputType.ERROR)
             return platforms
             
         # 获取目录的包名
@@ -128,7 +128,7 @@ class PlatformRegistry:
                             platforms[obj.platform_name] = obj # type: ignore
                             break
                 except Exception as e:
-                    PrettyOutput.print(f"Load platform {module_name} failed: {str(e)}", OutputType.ERROR)
+                    PrettyOutput.print(f"加载平台 {module_name} 失败: {str(e)}", OutputType.ERROR)
         
         return platforms
 
@@ -201,7 +201,7 @@ class PlatformRegistry:
             BasePlatform: Platform instance
         """
         if name not in self.platforms:
-            PrettyOutput.print(f"Platform not found: {name}", OutputType.ERROR)
+            PrettyOutput.print(f"未找到平台: {name}", OutputType.ERROR)
             return None
             
         try:
@@ -209,7 +209,7 @@ class PlatformRegistry:
             platform = self.platforms[name]()
             return platform
         except Exception as e:
-            PrettyOutput.print(f"Create platform failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"创建平台失败: {str(e)}", OutputType.ERROR)
             return None
 
     def get_available_platforms(self) -> List[str]:

jarvis/jarvis_platform_manager/main.py

@@ -31,10 +31,10 @@ def list_platforms():
                         output += f"  • {model_name}\n"
                 PrettyOutput.print(output, OutputType.SUCCESS, lang="markdown")
             else:
-                PrettyOutput.print("  • No available model information", OutputType.WARNING)
+                PrettyOutput.print("  • 没有可用的模型信息", OutputType.WARNING)
                 
         except Exception as e:
-            PrettyOutput.print(f"Failed to get model list for {platform_name}: {str(e)}", OutputType.WARNING)
+            PrettyOutput.print(f"获取 {platform_name} 的模型列表失败: {str(e)}", OutputType.WARNING)
 
 def chat_with_model(platform_name: str, model_name: str):
     """Chat with specified platform and model"""
@@ -43,13 +43,13 @@ def chat_with_model(platform_name: str, model_name: str):
     # Create platform instance
     platform = registry.create_platform(platform_name)
     if not platform:
-        PrettyOutput.print(f"Failed to create platform {platform_name}", OutputType.ERROR)
+        PrettyOutput.print(f"创建平台 {platform_name} 失败", OutputType.ERROR)
         return
     
     try:
         # Set model
         platform.set_model_name(model_name)
-        PrettyOutput.print(f"Connected to {platform_name} platform {model_name} model", OutputType.SUCCESS)
+        PrettyOutput.print(f"连接到 {platform_name} 平台 {model_name} 模型", OutputType.SUCCESS)
         
         # Start conversation loop
         while True:
@@ -58,7 +58,7 @@ def chat_with_model(platform_name: str, model_name: str):
             
             # Check if input is cancelled
             if user_input.strip() == "/bye":
-                PrettyOutput.print("Bye!", OutputType.SUCCESS)
+                PrettyOutput.print("再见!", OutputType.SUCCESS)
                 break
                 
             # Check if input is empty
@@ -70,22 +70,22 @@ def chat_with_model(platform_name: str, model_name: str):
                 try:
                     platform.delete_chat()
                     platform.set_model_name(model_name)  # Reinitialize session
-                    PrettyOutput.print("Session cleared", OutputType.SUCCESS)
+                    PrettyOutput.print("会话已清除", OutputType.SUCCESS)
                 except Exception as e:
-                    PrettyOutput.print(f"Failed to clear session: {str(e)}", OutputType.ERROR)
+                    PrettyOutput.print(f"清除会话失败: {str(e)}", OutputType.ERROR)
                 continue
                 
             try:
                 # Send to model and get reply
                 response = platform.chat_until_success(user_input)
                 if not response:
-                    PrettyOutput.print("No valid reply", OutputType.WARNING)
+                    PrettyOutput.print("没有有效的回复", OutputType.WARNING)
                     
             except Exception as e:
-                PrettyOutput.print(f"Failed to chat: {str(e)}", OutputType.ERROR)
+                PrettyOutput.print(f"聊天失败: {str(e)}", OutputType.ERROR)
                 
     except Exception as e:
-        PrettyOutput.print(f"Failed to initialize conversation: {str(e)}", OutputType.ERROR)
+        PrettyOutput.print(f"初始化会话失败: {str(e)}", OutputType.ERROR)
     finally:
         # Clean up resources
         try:
@@ -100,7 +100,7 @@ def info_command(args):
 def chat_command(args):
     """Process chat subcommand"""
     if not args.platform or not args.model:
-        PrettyOutput.print("Please specify platform and model. Use 'jarvis info' to view available platforms and models.", OutputType.ERROR)
+        PrettyOutput.print("请指定平台和模型。使用 'jarvis info' 查看可用平台和模型。", OutputType.ERROR)
         return
     chat_with_model(args.platform, args.model)
 
@@ -110,16 +110,16 @@ def main():
 
     init_env()
     
-    parser = argparse.ArgumentParser(description='Jarvis AI Platform')
-    subparsers = parser.add_subparsers(dest='command', help='Available subcommands')
+    parser = argparse.ArgumentParser(description='Jarvis AI 平台')
+    subparsers = parser.add_subparsers(dest='command', help='可用子命令')
     
     # info subcommand
-    info_parser = subparsers.add_parser('info', help='Display supported platforms and models information')
+    info_parser = subparsers.add_parser('info', help='显示支持的平台和模型信息')
     
     # chat subcommand
-    chat_parser = subparsers.add_parser('chat', help='Chat with specified platform and model')
-    chat_parser.add_argument('--platform', '-p', help='Specify the platform to use')
-    chat_parser.add_argument('--model', '-m', help='Specify the model to use')
+    chat_parser = subparsers.add_parser('chat', help='与指定平台和模型聊天')
+    chat_parser.add_argument('--platform', '-p', help='指定要使用的平台')
+    chat_parser.add_argument('--model', '-m', help='指定要使用的模型')
     
     args = parser.parse_args()
     

jarvis/jarvis_rag/main.py

@@ -154,9 +154,9 @@ class RAGTool:
         try:
             self.embedding_model = load_embedding_model()
             self.vector_dim = self.embedding_model.get_sentence_embedding_dimension()
-            PrettyOutput.print("Model loaded", output_type=OutputType.SUCCESS)
+            PrettyOutput.print("模型加载完成", output_type=OutputType.SUCCESS)
         except Exception as e:
-            PrettyOutput.print(f"Failed to load model: {str(e)}", output_type=OutputType.ERROR)
+            PrettyOutput.print(f"加载模型失败: {str(e)}", output_type=OutputType.ERROR)
             raise
 
         # 修改缓存相关初始化
@@ -218,7 +218,7 @@ class RAGTool:
                                 file_cache = pickle.load(f)
                                 self.documents.extend(file_cache["documents"])
                         except Exception as e:
-                            PrettyOutput.print(f"Failed to load cache for {file_path}: {str(e)}", 
+                            PrettyOutput.print(f"加载缓存失败,文件路径:{file_path}: {str(e)}", 
                                             output_type=OutputType.WARNING)
                 
                 # 重建向量索引
@@ -238,11 +238,11 @@ class RAGTool:
                         vectors = np.vstack(vectors)
                         self._build_index(vectors)
                         
-                PrettyOutput.print(f"Loaded {len(self.documents)} document fragments", 
+                PrettyOutput.print(f"加载 {len(self.documents)} 个文档片段", 
                                 output_type=OutputType.INFO)
                                 
             except Exception as e:
-                PrettyOutput.print(f"Failed to load cache index: {str(e)}", 
+                PrettyOutput.print(f"加载缓存索引失败: {str(e)}", 
                                 output_type=OutputType.WARNING)
                 self.documents = []
                 self.index = None
@@ -276,7 +276,7 @@ class RAGTool:
                 pickle.dump(index_data, f)
                             
         except Exception as e:
-            PrettyOutput.print(f"Failed to save cache: {str(e)}", output_type=OutputType.ERROR)
+            PrettyOutput.print(f"保存缓存失败: {str(e)}", output_type=OutputType.ERROR)
 
     def _build_index(self, vectors: np.ndarray):
         """Build FAISS index"""
@@ -382,7 +382,7 @@ class RAGTool:
             
             return get_embedding_batch(self.embedding_model, texts)
         except Exception as e:
-            PrettyOutput.print(f"Batch processing failed: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"批量处理失败: {str(e)}", OutputType.ERROR)
             return np.zeros((0, self.vector_dim), dtype=np.float32) # type: ignore
 
     def _process_file(self, file_path: str) -> List[Document]:
@@ -436,7 +436,7 @@ class RAGTool:
             return documents
             
         except Exception as e:
-            PrettyOutput.print(f"Failed to process file {file_path}: {str(e)}", 
+            PrettyOutput.print(f"处理文件失败: {file_path}: {str(e)}", 
                             output_type=OutputType.ERROR)
             return []
 
@@ -517,7 +517,7 @@ class RAGTool:
                             new_vectors.append(file_vectors)
                             
                     except Exception as e:
-                        PrettyOutput.print(f"Failed to process {file_path}: {str(e)}", OutputType.ERROR)
+                        PrettyOutput.print(f"处理文件失败: {file_path}: {str(e)}", OutputType.ERROR)
                     
                     pbar.update(1)
 
@@ -542,9 +542,9 @@ class RAGTool:
                 self._build_index(final_vectors)
 
             PrettyOutput.print(
-                f"Indexed {len(self.documents)} documents "
-                f"(New/Modified: {len(new_documents)}, "
-                f"Unchanged: {len(unchanged_documents)})", 
+                f"索引 {len(self.documents)} 个文档 "
+                f"(新/修改: {len(new_documents)}, "
+                f"不变: {len(unchanged_documents)})", 
                 OutputType.SUCCESS
             )
 
@@ -566,13 +566,13 @@ class RAGTool:
             return np.vstack(unchanged_vectors) if unchanged_vectors else None
             
         except Exception as e:
-            PrettyOutput.print(f"Failed to get unchanged vectors: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"获取不变向量失败: {str(e)}", OutputType.ERROR)
             return None
 
     def search(self, query: str, top_k: int = 30) -> List[Tuple[Document, float]]:
         """Search documents with context window"""
         if not self.index:
-            PrettyOutput.print("Index not built, building...", output_type=OutputType.INFO)
+            PrettyOutput.print("索引未构建,正在构建...", output_type=OutputType.INFO)
             self.build_index(self.root_dir)
             
         # Get query vector
@@ -670,7 +670,7 @@ Relevant documents (ordered by relevance):
                 prompt += "3. Reference the documents in the answer.\n"
                 if current_count + get_context_token_count(doc_content) > available_count:
                     PrettyOutput.print(
-                        "Due to context length limit, some fragments were omitted", 
+                        "由于上下文长度限制，部分内容被省略",
                         output_type=OutputType.WARNING
                     )
                     break
@@ -687,7 +687,7 @@ Relevant documents (ordered by relevance):
             return response
             
         except Exception as e:
-            PrettyOutput.print(f"Failed to answer: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"回答失败：{str(e)}", OutputType.ERROR)
             return None
 
     def is_index_built(self) -> bool:
@@ -724,7 +724,7 @@ def main():
             args.dir = current_dir
 
         if args.dir and args.build:
-            PrettyOutput.print(f"Processing directory: {args.dir}", output_type=OutputType.INFO)
+            PrettyOutput.print(f"正在处理目录: {args.dir}", output_type=OutputType.INFO)
             rag.build_index(args.dir)
             return 0
 
@@ -733,13 +733,13 @@ def main():
             if args.search:
                 results = rag.query(args.search)
                 if not results:
-                    PrettyOutput.print("No related content found", output_type=OutputType.WARNING)
+                    PrettyOutput.print("未找到相关内容", output_type=OutputType.WARNING)
                     return 1
                     
                 for doc in results:
-                    output = f"""File: {doc.metadata['file_path']}\n"""
-                    output += f"""Fragment {doc.metadata['chunk_index'] + 1}/{doc.metadata['total_chunks']}\n"""
-                    output += f"""Content:\n{doc.content}\n"""
+                    output = f"""文件: {doc.metadata['file_path']}\n"""
+                    output += f"""片段 {doc.metadata['chunk_index'] + 1}/{doc.metadata['total_chunks']}\n"""
+                    output += f"""内容:\n{doc.content}\n"""
                     PrettyOutput.print(output, output_type=OutputType.INFO, lang="markdown")
                 return 0
 
@@ -747,19 +747,19 @@ def main():
                 # Call ask method
                 response = rag.ask(args.ask)
                 if not response:
-                    PrettyOutput.print("Failed to get answer", output_type=OutputType.WARNING)
+                    PrettyOutput.print("获取答案失败", output_type=OutputType.WARNING)
                     return 1
                     
                 # Display answer
-                output = f"""Answer:\n{response}"""
+                output = f"""答案:\n{response}"""
                 PrettyOutput.print(output, output_type=OutputType.INFO)
                 return 0
 
-        PrettyOutput.print("Please specify operation parameters. Use -h to view help.", output_type=OutputType.WARNING)
+        PrettyOutput.print("请指定操作参数。使用 -h 查看帮助。", output_type=OutputType.WARNING)
         return 1
 
     except Exception as e:
-        PrettyOutput.print(f"Failed to execute: {str(e)}", output_type=OutputType.ERROR)
+        PrettyOutput.print(f"执行失败: {str(e)}", output_type=OutputType.ERROR)
         return 1
 
 if __name__ == "__main__":