From 645ca02cd34377053a22e13446d18624a642e8ea Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 10 Oct 2024 21:21:22 -0700 Subject: [PATCH] fixed type issues --- vision_agent/agent/vision_agent_coder.py | 2 +- vision_agent/tools/meta_tools.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py index 4211ca34..9a30f334 100644 --- a/vision_agent/agent/vision_agent_coder.py +++ b/vision_agent/agent/vision_agent_coder.py @@ -5,7 +5,7 @@ from pathlib import Path from typing import Any, Callable, Dict, List, Optional, Sequence, Union, cast -from redbaron import RedBaron +from redbaron import RedBaron # type: ignore from tabulate import tabulate import vision_agent.tools as T diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index 8706d2a0..7f59c685 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -11,7 +11,7 @@ import numpy as np from IPython.display import display -from redbaron import RedBaron +from redbaron import RedBaron # type: ignore import vision_agent as va from vision_agent.agent.agent_utils import extract_json @@ -25,8 +25,6 @@ from vision_agent.utils.image_utils import convert_to_b64, numpy_to_bytes from vision_agent.utils.video import frames_to_bytes -# These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent - CURRENT_FILE = None CURRENT_LINE = 0 DEFAULT_WINDOW_SIZE = 100 @@ -155,6 +153,9 @@ def __contains__(self, name: str) -> bool: return name in self.artifacts +# These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent + + def format_lines(lines: List[str], start_idx: int) -> str: output = "" for i, line in enumerate(lines):