Skip to content

Commit

Permalink
fix mypy issues
Browse files Browse the repository at this point in the history
  • Loading branch information
shankar-vision-eng committed Aug 28, 2024
1 parent a9b8c28 commit 160941d
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions vision_agent/agent/vision_agent_coder.py
Original file line number Diff line number Diff line change
Expand Up @@ -741,12 +741,12 @@ def chat_with_workflow(
plan = []
success = False

plans = self.create_plans(
plans = self._create_plans(
int_chat, customized_tool_names, working_memory, self.planner
)

if test_multi_plan:
self.log_plans(plans, self.verbosity)
self._log_plans(plans, self.verbosity)

tool_infos = retrieve_tools(
plans,
Expand Down Expand Up @@ -841,7 +841,13 @@ def log_progress(self, data: Dict[str, Any]) -> None:
if self.report_progress_callback is not None:
self.report_progress_callback(data)

def create_plans(self, int_chat, customized_tool_names, working_memory, planner):
def _create_plans(
self,
int_chat: List[Message],
customized_tool_names: Optional[List[str]],
working_memory: List[Dict[str, str]],
planner: LMM,
) -> Dict[str, Any]:
self.log_progress(
{
"type": "log",
Expand All @@ -859,7 +865,7 @@ def create_plans(self, int_chat, customized_tool_names, working_memory, planner)
)
return plans

def log_plans(self, plans, verbosity):
def _log_plans(self, plans: Dict[str, Any], verbosity: int) -> None:
if verbosity >= 1:
for p in plans:
# tabulate will fail if the keys are not the same for all elements
Expand Down

0 comments on commit 160941d

Please sign in to comment.