|
|
from typing import Any, Dict, List |
|
|
from flow_modules.Tachi67.ExtendLibraryFlowModule import ControllerFlow_ExtLib |
|
|
|
|
|
|
|
|
from dataclasses import dataclass |
|
|
|
|
|
|
|
|
@dataclass |
|
|
class Command: |
|
|
name: str |
|
|
description: str |
|
|
input_args: List[str] |
|
|
|
|
|
class Controller_CoderFlow(ControllerFlow_ExtLib): |
|
|
def __init__( |
|
|
self, |
|
|
commands: List[Command], |
|
|
**kwargs): |
|
|
super().__init__(**kwargs) |
|
|
self.system_message_prompt_template = self.system_message_prompt_template.partial( |
|
|
commands=self.build_commands_manual(commands), |
|
|
plan="no plans yet", |
|
|
plan_file_location="no location yet", |
|
|
code_library="no code library yet", |
|
|
code_library_location="no location yet", |
|
|
logs="no logs yet", |
|
|
) |
|
|
self.hint_for_model = """ |
|
|
Make sure your response is in the following format: |
|
|
Response Format: |
|
|
{ |
|
|
"command": "call one of the subordinates", |
|
|
"command_args": { |
|
|
"arg name": "value" |
|
|
} |
|
|
} |
|
|
""" |
|
|
|
|
|
def _get_code_library_location(self, input_data): |
|
|
assert "memory_files" in input_data, "memory_files not passed to Coder/Controller" |
|
|
assert "code_library" in input_data["memory_files"], "code_library not in memory files" |
|
|
return input_data["memory_files"]["code_library"] |
|
|
|
|
|
def _get_code_library_content(self, input_data): |
|
|
assert "code_library" in input_data, "code_library not passed to Coder/Controller" |
|
|
code_library_content = input_data["code_library"] |
|
|
if len(code_library_content) == 0: |
|
|
code_library_content = 'No plan yet' |
|
|
return code_library_content |
|
|
|
|
|
def _update_prompts_and_input(self, input_data: Dict[str, Any]): |
|
|
if 'goal' in input_data: |
|
|
input_data['goal'] += self.hint_for_model |
|
|
if 'result' in input_data: |
|
|
input_data['result'] += self.hint_for_model |
|
|
plan_file_location = self._get_plan_file_location(input_data) |
|
|
plan_content = self._get_plan_content(input_data) |
|
|
code_library_location = self._get_code_library_location(input_data) |
|
|
code_library_content = self._get_code_library_content(input_data) |
|
|
logs_content = self._get_logs_content(input_data) |
|
|
self.system_message_prompt_template = self.system_message_prompt_template.partial( |
|
|
plan_file_location=plan_file_location, |
|
|
plan=plan_content, |
|
|
code_library_location=code_library_location, |
|
|
code_library=code_library_content, |
|
|
logs=logs_content |
|
|
) |
|
|
|
|
|
|
