Skip to content

Commit

Permalink
fix type and lint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
dillonalaird committed Sep 22, 2024
1 parent 64df1e8 commit 14fc101
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions vision_agent/agent/vision_agent_coder.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,10 @@
import logging
import os
import sys
import tempfile
from json import JSONDecodeError
from pathlib import Path
from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union, cast

from PIL import Image
from rich.console import Console
from rich.style import Style
from rich.syntax import Syntax
Expand Down Expand Up @@ -776,11 +774,11 @@ def chat_with_workflow(
verbosity=self.verbosity,
)
best_plan = plan_thoughts["best_plan"]
plan_thoughts = plan_thoughts["thoughts"]
plan_thoughts_str = plan_thoughts["thoughts"]
else:
best_plan = list(plans.keys())[0]
tool_output_str = ""
plan_thoughts = ""
plan_thoughts_str = ""

if best_plan in plans and best_plan in tool_infos:
plan_i = plans[best_plan]
Expand Down Expand Up @@ -815,7 +813,7 @@ def chat_with_workflow(
+ "\n-".join([e for e in plan_i["instructions"]]),
tool_info=tool_info,
tool_output=tool_output_str,
plan_thoughts=plan_thoughts,
plan_thoughts=plan_thoughts_str,
tool_utils=T.UTILITIES_DOCSTRING,
working_memory=working_memory,
coder=self.coder,
Expand Down

0 comments on commit 14fc101

Please sign in to comment.