fix imports

landing-ai · Mar 22, 2024 · 546c858 · 546c858
1 parent 7b27549
commit 546c858
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/vision_agent/agent/easytool.py b/vision_agent/agent/easytool.py
@@ -4,7 +4,8 @@
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-from vision_agent import LLM, LMM, OpenAILLM
+from vision_agent.llm import LLM, OpenAILLM
+from vision_agent.lmm import LMM
 from vision_agent.tools import TOOLS
 
 from .agent import Agent

diff --git a/vision_agent/agent/reflexion.py b/vision_agent/agent/reflexion.py
@@ -4,7 +4,8 @@
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Union
 
-from vision_agent import LLM, LMM, OpenAILLM
+from vision_agent.llm import LLM, OpenAILLM
+from vision_agent.lmm import LMM
 
 from .agent import Agent
 from .reflexion_prompts import (