fixed typign issue

landing-ai · Mar 16, 2024 · 7bfb5c1 · 7bfb5c1
1 parent 2436284
commit 7bfb5c1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/vision_agent/agent/reflexion.py b/vision_agent/agent/reflexion.py
@@ -1,5 +1,5 @@
 import re
-from typing import Dict, List, Tuple, Union
+from typing import Dict, List, Optional, Tuple, Union
 
 from vision_agent import LLM, OpenAILLM
 
@@ -62,8 +62,6 @@ def __init__(
  self.finsh_prompt = finsh_prompt
  self.cot_examples = cot_examples
  self.refelct_examples = reflect_examples
- self.self_reflect_llm = self_reflect_llm
- self.action_agent = action_agent
  self.reflections: List[str] = []
 
  if self_reflect_llm is None: