isort

vision_agent/__init__.py

@@ -1,4 +1,4 @@
-from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm
-from .llm import LLM, OpenAILLM
-from .emb import Embedder, SentenceTransformerEmb, OpenAIEmb, get_embedder
 from .data import DataStore, build_data_store
+from .emb import Embedder, OpenAIEmb, SentenceTransformerEmb, get_embedder
+from .llm import LLM, OpenAILLM
+from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm

vision_agent/data/data.py

@@ -2,7 +2,7 @@
 
 import uuid
 from pathlib import Path
-from typing import Dict, List, Optional, Union, cast, Callable
+from typing import Callable, Dict, List, Optional, Union, cast
 
 import faiss
 import numpy as np

vision_agent/image_utils.py

@@ -1,7 +1,7 @@
 import base64
 from io import BytesIO
 from pathlib import Path
-from typing import Union, Tuple
+from typing import Tuple, Union
 
 import numpy as np
 from PIL import Image

vision_agent/llm/llm.py

@@ -1,6 +1,6 @@
 import json
-from typing import cast
 from abc import ABC, abstractmethod
+from typing import cast
 
 from vision_agent.tools import (
  CHOOSE_PARAMS,

vision_agent/tools/__init__.py

@@ -1,2 +1,2 @@
-from .prompts import SYSTEM_PROMPT, CHOOSE_PARAMS
-from .tools import ImageTool, CLIP, GroundingDINO, GroundingSAM
+from .prompts import CHOOSE_PARAMS, SYSTEM_PROMPT
+from .tools import CLIP, GroundingDINO, GroundingSAM, ImageTool