From 546c858a8c8a1845d44f4a2cb17e4adc3e1d00a8 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Fri, 22 Mar 2024 11:18:12 -0700 Subject: [PATCH] fix imports --- vision_agent/agent/easytool.py | 3 ++- vision_agent/agent/reflexion.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/vision_agent/agent/easytool.py b/vision_agent/agent/easytool.py index b86400d9..6a3c6c77 100644 --- 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 index b28e0a34..f13984b1 100644 --- 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 (