refactor code

[XuhuiZhou]

Closes #5

📑 Description

In order to keep the agent outputting actions to call the tools, we can basically utilize an internal validator like this:

HAI-Cosys/haicosystem/generation_utils/validation.py

Line 10 in 49ae78f

@beartype

Using such a mechanism could ensure the structure of the agent output is correct.

To realize this, basically we need to have the tool parser:

HAI-Cosys/haicosystem/tools/tool_parser.py

Line 30 in 49ae78f

class VenmoSendMoneyReturns(BaseModel):

This format can do a rule-based check of the output of LLMs.

✅ Checks

• My pull request adheres to the code style of this project
• My code requires changes to the documentation
• I have updated the documentation as required
• All the tests have passed
• Branch name follows type/descript (e.g. feature/add-llm-agents)
• Ready for code review

ℹ Additional Information

[ProKil]

You can try out pydantic validator? In sotopia, we may want to use it in place of beartype.

sotopia-lab/sotopia#115

[GloriaXimingLu]

thanks for adding the interface for checking the tool! I'll edit my pull request based on this.

[XuhuiZhou]

You can try out pydantic validator? In sotopia, we may want to use it in place of beartype.

sotopia-lab/sotopia#115

Good point! I will think this further and probably implement this on other pull requests