How To Use FileWriterTool To Write Code To File

[jamesbyars]

I'm working on a crew that translates business requirements into working software (python). What I have is an Agent that writes the code to satisfy the requirements but the agent is not able to write the code to a file.

I don't want to hardcode a specific file to write the code to because the application can be made up of multiple files (think DAOs, views, controllers, tests, etc).

I want the crew to be able to:

• read the existing filesystem to comprehend the current state of the application
• update or write the necessary code to satisfy the requirements
• write the changes to the file system to reflect the changes

I've looked at the LangChain FileManagementToolkit and the CrewAI Tool FileWriterTool.

I'm unable to get either tool working to write the task generated output to a file using the agent. Sample code (incomplete) below.

from crewai import LLM, Agent, Task, Crew
from langchain.agents.agent_toolkits import FileManagementToolkit
from crewai_tools.tools.file_writer_tool.file_writer_tool import FileWriterToolInput

llm = LLM(model="bedrock/anthropic.claude-3-haiku-20240307-v1:0")

filesystem_tools = FileManagementToolkit(
    root_dir='/Users/[dir]/code/crewai_poc/src',
).get_tools()

software_engineer = Agent(
    role='Senior Python Engineer',
    goal='''Design and implement solutions based on user stories and requirements as defined. Deliver high quality code with an emphasis on SOLID software engineering techniques''',
    backstory='''You have worked in high pressure environments your entire career. You understand how to use Python to meet deadlines and delivery high quality, working software.''',
    verbose=True,
    allow_delegation=False,
    llm=llm,
    tool=[
        filesystem_tools,
    ],
)

create_test = Task(
    description='''
    YOU MUST USE the tool to write an updated 
      version of the test to the file 
      system in the following path: '.' 
      replacing the text content with the suggestions 
      provided.
      
      You first write the file then your final answer 
      MUST be the updated test content.

      RULES
      -----
      - NEVER WRITE \\n (newlines as string) on the file, just the code.
      - Keep the same component imports and don't use new components.
      - NEVER USE COMPONENTS THAT ARE NOT IMPORTED.
      - ALL COMPONENTS USED SHOULD BE IMPORTED, don't make up components.
      - Save the file as with `.py` extension.

      If you follow the rules I'll give you a $100 tip!!! 
      MY LIFE DEPEND ON YOU FOLLOWING IT!  
    
    Using the requirements provided below, implement unit tests in python using the unittest module.
    
    {requirements}''',
    expected_output='''A file containing the test code''',
    agent=software_engineer,
    verbose=True,
    output_pydantic=FileWriterToolInput
)

crew = Crew(
    agents=[software_engineer],
    tasks=[
        create_test,
    ],
    verbose=True,
)

inputs = {
    'requirements': 'hello world application that prints out "Hello World!"',
}

result = crew.kickoff(inputs=inputs)