mypy, flake8 fixes

vision_agent/agent/vision_agent.py

@@ -6,7 +6,11 @@
 
 from vision_agent.agent import Agent
 from vision_agent.agent.agent_utils import extract_json
-from vision_agent.agent.vision_agent_prompts import EXAMPLES_CODE1, EXAMPLES_CODE2, VA_CODE
+from vision_agent.agent.vision_agent_prompts import (
+ EXAMPLES_CODE1,
+ EXAMPLES_CODE2,
+ VA_CODE,
+)
 from vision_agent.lmm import LMM, Message, OpenAILMM
 from vision_agent.tools import META_TOOL_DOCSTRING
 from vision_agent.utils import CodeInterpreterFactory
@@ -80,7 +84,9 @@ def run_code_action(code: str, code_interpreter: CodeInterpreter) -> str:
  # for log in result.logs.stderr:
  # return_str += log.replace("\\n", "\n")
  if result.error:
- return_str += "\n" + result.error.value + "\n".join(result.error.traceback_raw)
+ return_str += (
+ "\n" + result.error.value + "\n".join(result.error.traceback_raw)
+ )
 
  return return_str
 

vision_agent/lmm/lmm.py

@@ -5,14 +5,15 @@
 import os
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Sequence, Union, cast
+from typing import Any, Callable, Dict, List, Optional, Union, cast
 
 import requests
 from openai import AzureOpenAI, OpenAI
 from PIL import Image
 
 import vision_agent.tools as T
 from vision_agent.tools.prompts import CHOOSE_PARAMS, SYSTEM_PROMPT
+
 from .types import Message
 
 _LOGGER = logging.getLogger(__name__)

vision_agent/tools/meta_tools.py

@@ -1,9 +1,9 @@
 import subprocess
 from pathlib import Path
 from typing import List
-from vision_agent.lmm.types import Message
 
 import vision_agent as va
+from vision_agent.lmm.types import Message
 from vision_agent.tools.tool_utils import get_tool_documentation
 
 CURRENT_FILE = None

vision_agent/tools/tool_utils.py

@@ -1,9 +1,9 @@
+import inspect
 import logging
 import os
-import pandas as pd
-import inspect
-from typing import Any, Dict, List, Callable
+from typing import Any, Callable, Dict, List
 
+import pandas as pd
 from requests import Session
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry