jarvis-ai-assistant 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

jarvis/tools/sub_agent.py

@@ -1,83 +0,0 @@
-from typing import Dict, Any
-from ..agent import Agent
-from ..models import BaseModel
-from ..utils import PrettyOutput, OutputType
-from .base import ToolRegistry
-
-class SubAgentTool:
-    name = "create_sub_agent"
-    description = """创建一个子代理来处理独立任务。（重要：子代理启动时没有任何上下文！必须提供完整的步骤和上下文。）"""
-
-    parameters = {
-        "type": "object",
-        "properties": {
-            "name": {
-                "type": "string",
-                "description": "子代理名称（例如：'文件分析器'）"
-            },
-            "task": {
-                "type": "string",
-                "description": "需要明确步骤和目标的任务"
-            },
-            "context": {
-                "type": "string",
-                "description": "必填：背景信息、执行步骤和预期结果",
-                "default": ""
-            }
-        },
-        "required": ["name", "task", "context"]
-    }
-
-    def __init__(self, model: BaseModel):
-        """Initialize with the same model as parent agent"""
-        self.model = model
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """Create and run a sub-agent for the specified task"""
-        try:
-            name = args["name"]
-            task = args["task"]
-            context = args.get("context")
-
-            if not context:
-                return {
-                    "success": False,
-                    "error": "必须提供上下文信息，包括完整的背景和执行步骤。"
-                }
-
-            PrettyOutput.print(f"正在创建子代理 '{name}'...", OutputType.INFO)
-            
-            # Create a new tool registry for the sub-agent
-            tool_registry = ToolRegistry(self.model)
-            
-            # Create the sub-agent with the specified name
-            sub_agent = Agent(self.model, tool_registry, name=name, is_sub_agent=True)
-            
-            # Prepare the task with context
-            full_task = f"""背景和步骤：
-{context}
-
-主要任务：
-{task}
-
-要求：
-1. 严格按照提供的步骤执行
-2. 每完成一个步骤都要报告进度
-3. 突出显示任何问题或不明确的点
-4. 提供符合预期输出的详细结果"""
-            
-            PrettyOutput.print(f"子代理 '{name}' 正在执行任务...", OutputType.INFO)
-            
-            # Execute the task and get the summary
-            summary = sub_agent.run(full_task)
-            return {
-                "success": True,
-                "stdout": f"子代理 '{name}' 已完成。\n\n结果：\n{summary}",
-                "stderr": ""
-            }
-
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"子代理执行失败：{str(e)}"
-            } 

jarvis/tools/webpage.py

@@ -1,76 +0,0 @@
-from typing import Dict, Any
-import requests
-from bs4 import BeautifulSoup
-from ..utils import PrettyOutput, OutputType
-
-class WebpageTool:
-    name = "read_webpage"
-    description = "读取网页内容，提取标题和正文文本"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "url": {
-                "type": "string",
-                "description": "需要读取的网页URL"
-            }
-        },
-        "required": ["url"]
-    }
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """读取网页内容"""
-        try:
-            url = args["url"]
-            
-            # 设置请求头
-            headers = {
-                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
-            }
-            
-            # 发送请求
-            PrettyOutput.print(f"正在读取网页: {url}", OutputType.INFO)
-            response = requests.get(url, headers=headers, timeout=10)
-            response.raise_for_status()
-            
-            # 使用正确的编码
-            response.encoding = response.apparent_encoding
-            
-            # 解析HTML
-            soup = BeautifulSoup(response.text, 'html.parser')
-            
-            # 移除script和style标签
-            for script in soup(["script", "style"]):
-                script.decompose()
-            
-            # 提取标题
-            title = soup.title.string if soup.title else ""
-            title = title.strip() if title else "无标题"
-            
-            # 提取正文
-            text = soup.get_text(separator='\n', strip=True)
-            lines = [line.strip() for line in text.splitlines() if line.strip()]
-            
-            # 构建输出
-            output = [
-                f"标题: {title}",
-                "",
-                "正文内容:",
-                "\n".join(lines)
-            ]
-            
-            return {
-                "success": True,
-                "stdout": "\n".join(output),
-                "stderr": ""
-            }
-
-        except requests.RequestException as e:
-            return {
-                "success": False,
-                "error": f"网页请求失败: {str(e)}"
-            }
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"解析网页失败: {str(e)}"
-            } 

jarvis/zte_llm.py

@@ -1,135 +0,0 @@
-import requests
-import json
-import os
-from pathlib import Path
-from typing import Dict, Any, List, Optional
-
-from jarvis.utils import OutputType, PrettyOutput
-from .models import BaseModel
-
-class ZteLLM(BaseModel):
-    """ZTE Nebula LLM implementation"""
-    
-    def __init__(self, 
-                 app_id: str,
-                 app_key: str,
-                 emp_no: str,
-                 auth_value: str,
-                 model: str = "nebulacoder",
-                 ):
-        """Initialize ZTE LLM with required credentials"""
-        self.app_id = str(app_id)
-        self.app_key = str(app_key)
-        self.emp_no = str(emp_no)
-        self.auth_value = str(auth_value)
-        self.model = model
-        self.base_url = "https://studio.zte.com.cn/zte-studio-ai-platform/openapi/v1"
-        
-    def _make_request(self, endpoint: str, data: Dict[str, Any]) -> Dict[str, Any]:
-        """Make request to ZTE API"""
-        headers = {
-            'Content-Type': 'application/json',
-            'Authorization': f'Bearer {self.app_id}-{self.app_key}',
-            'X-Emp-No': self.emp_no,
-            'X-Auth-Value': self.auth_value
-        }
-        
-        response = requests.post(
-            f"{self.base_url}/{endpoint}",
-            headers=headers,
-            json=data,
-            stream=True  # 启用流式传输
-        )
-        
-        response.raise_for_status()
-        
-        full_content = []
-        for line in response.iter_lines():
-            if line:
-                # 解析 SSE 数据
-                line = line.decode('utf-8')
-                if line.startswith('data: '):
-                    try:
-                        data = json.loads(line[6:])  # 跳过 "data: " 前缀
-                        if "result" in data:
-                            result = data["result"]
-                            if result:  # 只处理非空结果
-                                full_content.append(result)
-                                PrettyOutput.print_stream(result, OutputType.SYSTEM)
-                            if data.get("finishReason") == "stop":
-                                break
-                    except json.JSONDecodeError:
-                        continue
-        PrettyOutput.print_stream_end()
-        
-        return "".join(full_content)
-
-    def chat(self, messages: List[Dict[str, Any]]) -> str:
-        """Chat with ZTE LLM"""
-        # Convert messages to prompt
-        prompt = self._convert_messages_to_prompt(messages)
-        
-        # Prepare data for API call
-        data = {
-            "chatUuid": "",
-            "chatName": "",
-            "stream": True,  # 启用流式响应
-            "keep": False,
-            "text": prompt,
-            "model": self.model
-        }
-        
-        try:
-            return self._make_request("chat", data)
-
-            
-        except Exception as e:
-            raise Exception(f"ZTE LLM chat failed: {str(e)}")
-
-    def _convert_messages_to_prompt(self, messages: List[Dict[str, Any]]) -> str:
-        """Convert message list to a single prompt string"""
-        prompt_parts = []
-        
-        for message in messages:
-            role = message["role"]
-            content = message.get("content", "")
-            
-            if role == "system":
-                prompt_parts.append(f"System: {content}")
-            elif role == "user":
-                prompt_parts.append(f"User: {content}")
-            elif role == "assistant":
-                prompt_parts.append(f"Assistant: {content}")
-            elif role == "tool":
-                prompt_parts.append(f"Tool Result: {content}")
-                
-        return "\n\n".join(prompt_parts)
-
-
-def create_zte_llm(model_name: str = "NebulaBiz") -> ZteLLM:
-    """Create ZTE LLM instance with provided parameters"""
-    # Load environment variables from file
-    
-    # Get credentials from parameters, env file, or system environment variables
-    app_id = os.getenv('ZTE_APP_ID') 
-    app_key = os.getenv('ZTE_APP_KEY') 
-    emp_no = os.getenv('ZTE_EMP_NO') 
-    auth_value = os.getenv('ZTE_AUTH_VALUE') 
-    
-    # Validate required credentials
-    if not all([app_id, app_key, emp_no, auth_value]):
-        raise ValueError(
-            "Missing required credentials. Please provide through either:\n"
-            "1. Function parameters\n"
-            "2. ~/.jarvis_env file\n"
-            "3. System environment variables\n\n"
-            "Required variables: ZTE_APP_ID, ZTE_APP_KEY, ZTE_EMP_NO, ZTE_AUTH_VALUE"
-        )
-    
-    return ZteLLM(
-        app_id=app_id,
-        app_key=app_key,
-        emp_no=emp_no,
-        auth_value=auth_value,
-        model=model_name
-    ) 

jarvis_ai_assistant-0.1.9.dist-info/RECORD

@@ -1,39 +0,0 @@
-jarvis/__init__.py,sha256=-pqI_MJPVrCN3BBcW2YIg8KbW5B7vfbpb-ESMyNKpVg,49
-jarvis/agent.py,sha256=ePgdOCWdkw35KITvZOqcrJOGl-F4wzvrltHTw7N4p88,11014
-jarvis/main.py,sha256=4IqebMKGoBGJMSNFE6E6GQ_Ay-uuOT-BLsQPnQVW47k,6705
-jarvis/models.py,sha256=G9QkUqoyOyUx6eSqwbn3hCSwHguN0bUm7Nbd6Ql0IQs,3771
-jarvis/utils.py,sha256=3hLtv-HcBL8Ngw69cowhARuIFrjcQ6yRP3Y1o9CvtsI,5992
-jarvis/zte_llm.py,sha256=Us5D6zMdZT6SoSnUgKWQbxthGadXkulLs_oSmW5MPDo,4666
-jarvis/__pycache__/__init__.cpython-313.pyc,sha256=GF-b9JNTGgyKROLNqZojLYFIrgPShHKlZADEFHP_YfE,208
-jarvis/__pycache__/agent.cpython-313.pyc,sha256=itbFLk6qJ9FTjbiyj6DsQhu5NbdxVaP4g-SygRguvSc,11625
-jarvis/__pycache__/main.cpython-313.pyc,sha256=6hWRW-RG_bZrqwhSQXyhKaJBXexgLqKa-_SZzAk6_eo,8479
-jarvis/__pycache__/models.cpython-313.pyc,sha256=uWuRIjGrY4YDB3dGW5PGDLWaS03et8g11O725TjY_eU,5960
-jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
-jarvis/__pycache__/utils.cpython-313.pyc,sha256=k4jyAlx4tlW0MKLMLzV7OOH9zsKrK0H955kY6M2SuFU,8772
-jarvis/__pycache__/zte_llm.cpython-313.pyc,sha256=R_HT_Gc6OQelxRLRQ-yB8Tf0sYHcb18dh9E4RF6ivbE,5662
-jarvis/tools/__init__.py,sha256=tH6YaApKpqs1YSjEllRzOZZUd-OTFKSZ5oA0zs7HdSE,297
-jarvis/tools/base.py,sha256=zK-JteBwj9d2BD_7BSAp-zD7wgJbVBppAWyfFxaPObI,4114
-jarvis/tools/bing_search.py,sha256=SvYeXM83eP9qREDcWguh0_r0m0npbcwXIyzEwwfreKU,1444
-jarvis/tools/file_ops.py,sha256=zTksx45NZm3iz9itN5iQGZ8DoxnSeTHdrnF08_ix7PU,3770
-jarvis/tools/search.py,sha256=P4vEbwcXTP6gpc4nwLEuERX0vVszGaer9xvaLfFI0Uc,4836
-jarvis/tools/shell.py,sha256=7q52lA3slf0TdjBjP1bkwugoO5pB0eqh6cYjAzAXNtI,2547
-jarvis/tools/sub_agent.py,sha256=QJYWdil1goZfo95CQkuhkjZBl-Rx6UvyeFMiGnDyjOA,2832
-jarvis/tools/webpage.py,sha256=DBnh9gye6oL2nVjzU2SU4Jupsck8x9g9On-rbjV6d8Y,2386
-jarvis/tools/__pycache__/__init__.cpython-313.pyc,sha256=1d1eUI0FlaydGB8TFBln6s7cBKd6gcPdER2_DIfX2Aw,443
-jarvis/tools/__pycache__/base.cpython-313.pyc,sha256=2OJWYNsGJjKJs6Vdj9NcN_Snjd2bA_OLlB-zr2j4wfs,6462
-jarvis/tools/__pycache__/bing_search.cpython-313.pyc,sha256=1G_wPbk5wcQYh7H0drLIS2Aw0XOG2ZM8ztgfQaqu3P8,2031
-jarvis/tools/__pycache__/file_ops.cpython-313.pyc,sha256=LbOp31JUzoRp5XVazy1VBqCQhpFg0qQYmVftFVY90V4,3628
-jarvis/tools/__pycache__/python_script.cpython-313.pyc,sha256=8JpryqTovEiTvBlWAK1KjZmPvHUuPc9GT9rTXBEQoJc,6693
-jarvis/tools/__pycache__/rag.cpython-313.pyc,sha256=JH6-PSZRMKAvTZqCwlRXJGClxYXNMs-vetU0q7hBLz0,6064
-jarvis/tools/__pycache__/search.cpython-313.pyc,sha256=wLMIkFwT-h4NGHgssytT4xme7sGO6ZhEnex7kjcy0-k,5990
-jarvis/tools/__pycache__/shell.cpython-313.pyc,sha256=QMaLUc1MtZXWod3msv_x7iMq2IybwMwz1OoaKsbm6U4,3271
-jarvis/tools/__pycache__/sub_agent.cpython-313.pyc,sha256=qzNLnQxqFZoAiYOPH4GrCGxaQVs-QWfTLspW5a1Xa6c,3144
-jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRvoYmi7A0GzxYkzU-C4Wfhs5qW_HBqs,2271
-jarvis/tools/__pycache__/user_input.cpython-313.pyc,sha256=JjTFOhObKsKF4Pn8KBRuKfV1_Ssj083fjU7Mfc_5z7c,2531
-jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
-jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=BjzSfnNzsKCrLETCcWjt32lNDLzwnjqcVGg4JfWd9OM,3008
-jarvis_ai_assistant-0.1.9.dist-info/METADATA,sha256=d5LRsaxuZB6V0m8klCq07sxghlARFo7K8VM36TyCnvk,3753
-jarvis_ai_assistant-0.1.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jarvis_ai_assistant-0.1.9.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
-jarvis_ai_assistant-0.1.9.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.9.dist-info/RECORD,,