feat(env): update input api for env

[lwaekfjlk]

Closes #

📑 Description

✅ 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

[codecov]

Codecov Report

Attention: Patch coverage is 55.26316% with 17 lines in your changes missing coverage. Please review.

Project coverage is 86.83%. Comparing base (5b62fe6) to head (11f9fc2).
Report is 4 commits behind head on main.

Files with missing lines	Patch %	Lines
...arch_town/envs/env_proposal_writing_without_rag.py	6.25%	15 Missing ⚠️
...esearch_town/envs/env_proposal_writing_with_rag.py	89.47%	2 Missing ⚠️

@@            Coverage Diff             @@
##             main     #761      +/-   ##
==========================================
+ Coverage   86.72%   86.83%   +0.11%     
==========================================
  Files          62       62              
  Lines        2696     2697       +1     
==========================================
+ Hits         2338     2342       +4     
+ Misses        358      355       -3     

Files with missing lines	Coverage Δ
research_town/envs/env_review_writing.py	97.72% <100.00%> (ø)
...esearch_town/envs/env_proposal_writing_with_rag.py	96.36% <89.47%> (-1.75%)	⬇️
...arch_town/envs/env_proposal_writing_without_rag.py	32.65% <6.25%> (+0.65%)	⬆️

... and 1 file with indirect coverage changes