SigmaFlow 0.0.76__tar.gz → 0.0.77__tar.gz

{sigmaflow-0.0.76 → sigmaflow-0.0.77}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SigmaFlow
-Version: 0.0.76
+Version: 0.0.77
 Summary: SigmaFlow is a Python package designed to optimize the performance of task-flow related to LLMs/MLLMs or Multi-agent
 Author-email: Kangkun Mao <maokangkun@pjlab.org.cn>
 License-Expression: MIT

{sigmaflow-0.0.76 → sigmaflow-0.0.77}/src/SigmaFlow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SigmaFlow
-Version: 0.0.76
+Version: 0.0.77
 Summary: SigmaFlow is a Python package designed to optimize the performance of task-flow related to LLMs/MLLMs or Multi-agent
 Author-email: Kangkun Mao <maokangkun@pjlab.org.cn>
 License-Expression: MIT

{sigmaflow-0.0.76 → sigmaflow-0.0.77}/src/sigmaflow/agent/agent.py

@@ -235,7 +235,7 @@ class Agent:
                             print(f"{Prompt.Tokens}inp: {response.usage.prompt_tokens}, out: {response.usage.completion_tokens}, time: {cost_time:.2f}s, {(response.usage.completion_tokens or 0) / cost_time:.2f} tokens/s, tools: {sum(not i[-1].startswith('Error:') for i in info['used_tools'])}/{len(info['used_tools'])}, skills: {sum(not i[-1].startswith('Error:') for i in info['used_skills'])}/{len(info['used_skills'])}")
                             break
                     else:
-                        if msg.content.strip(): print(f"{Prompt.Assistant}{msg.content}")
+                        if msg.content and msg.content.strip(): print(f"{Prompt.Assistant}{msg.content}")
 
             results = []
             for block in response.content if self.method == 'anthropic' else msg.tool_calls:
@@ -272,12 +272,12 @@ class Agent:
                         console.rule("Subagent End", style="bold red")
                     else:
                         handler = TOOL_HANDLERS.get(func)
-                        output = str(handler(**inp)) if handler else f"Error: unknown tool {func}"
-                        if len(output) > 1000: output = output[:400]+'\n......\n'+output[-400:]
+                        output = out_prt = str(handler(**inp)) if handler else f"Error: unknown tool {func}"
+                        if len(output) > 1000: out_prt = output[:400]+'\n......\n'+output[-400:]
                     if func in MCP.tools:
-                        print(f"{Prompt.MCP}[magenta]{func.upper()}[/]\n{output}")
+                        print(f"{Prompt.MCP}[magenta]{func.upper()}[/]\n{out_prt}")
                     else:
-                        print(f"{Prompt.Tool}[magenta]{func.upper()}[/]\n{output}")
+                        print(f"{Prompt.Tool}[magenta]{func.upper()}[/]\n{out_prt}")
 
                     match self.method:
                         case "anthropic":

{sigmaflow-0.0.76 → sigmaflow-0.0.77}/src/sigmaflow/agent/conf.py

@@ -135,11 +135,12 @@ CHILD_TOOLS = [
     },
     {
         "name": "read_file", 
-        "description": "Read file contents.",
+        "description": "Reads a file from the local filesystem. If the file is too long, you can specify a line offset (0-indexed) and limit to read a portion of the file.",
         "input_schema": {
             "type": "object", 
             "properties": {
-                "path": {"type": "string"}, 
+                "path": {"type": "string"},
+                "offset": {"type": "integer"},
                 "limit": {"type": "integer"}
             }, 
             "required": ["path"]

{sigmaflow-0.0.76 → sigmaflow-0.0.77}/src/sigmaflow/agent/tools.py

@@ -68,6 +68,7 @@ def run_grep(tool: str, args: str) -> str:
         r = subprocess.run(
             [tool, *argv],
             cwd=WORKDIR,
+            stdin=subprocess.DEVNULL,
             capture_output=True,
             text=True,
             timeout=CMD_TIMEOUT,
@@ -79,13 +80,19 @@ def run_grep(tool: str, args: str) -> str:
     except Exception as e:
         return f"Error: {e}"
 
-def run_read(path: str, limit: int = None) -> str:
+def run_read(path: str, offset: int = None, limit: int = None) -> str:
     try:
         path = (WORKDIR / path).resolve()
         text = path.read_text()
         lines = text.splitlines()
+
+        if offset and 0 < offset < len(lines):
+            lines = [f"[{offset} lines skipped] ..."] + lines[offset:]
+            if limit: limit += 1
+
         if limit and limit < len(lines):
-            lines = lines[:limit] + [f"... ({len(lines) - limit} more lines)"]
+            lines = lines[:limit] + [f"... [{len(lines) - limit} more lines]"]
+
         return "\n".join(lines)[:50000]
     except Exception as e:
         return f"Error: {e}"
@@ -620,7 +627,7 @@ class MCPClient:
 TOOL_HANDLERS = {
     "bash":       lambda **kw: run_bash(kw.get("command", "")),
     "grep":       lambda **kw: run_grep(kw.get("tool", ""), kw.get("args", "")),
-    "read_file":  lambda **kw: run_read(kw["path"], kw.get("limit")),
+    "read_file":  lambda **kw: run_read(kw["path"], offset=kw.get("offset"), limit=kw.get("limit")),
     "write_file": lambda **kw: run_write(kw["path"], kw.get("content", "")),
     "edit_file":  lambda **kw: run_edit(kw["path"], kw.get("old_text", ""), kw.get("new_text", "")),
     "load_skill": lambda **kw: skill_loader.get_content(kw.get("name", "")),