| | 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 |
| | ) |
| |
|
| |
|
