ishangsf/workflows
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #105 from devchat-ai/handle-readme

Browse Source 
Handle readme manually
This commit is contained in:
boob.yang 2024-05-16 07:33:08 +00:00 committed by GitHub
commit cd7b56b4fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
merico/unit_tests/main.py
View File

@ -298,25 +298,33 @@ def main(input: str):
"/unit_tests {a}:::{b}:::{c}:::{d}:::{e}:::{f}"
"""
# Parse input
params = input.strip().split(":::")
assert len(params) == 6, f"Invalid input: {input}, number of params: {len(params)}"
(
file_path,
func_name,
func_start_line, # 0-based, inclusive
func_end_line, # 0-based, inclusive
container_start_line, # 0-based, inclusive
container_end_line, # 0-based, inclusive
) = params
try:
params = input.strip().split(":::")
assert len(params) == 6, f"Invalid input: {input}, number of params: {len(params)}"
(
file_path,
func_name,
func_start_line, # 0-based, inclusive
func_end_line, # 0-based, inclusive
container_start_line, # 0-based, inclusive
container_end_line, # 0-based, inclusive
) = params
func_start_line = int(func_start_line)
func_end_line = int(func_end_line)
container_start_line = int(container_start_line)
container_end_line = int(container_end_line)
except Exception as e:
raise Exception(f"Invalid input: {input}, error: {e}")
readme = os.path.join(os.path.dirname(__file__), "README.md")
if os.path.exists(readme):
with open(readme, "r", encoding="utf-8") as f:
readme_text = f.read()
print(readme_text)
return
else:
raise Exception(f"Invalid input: {input}, error: {e}")
user_prompt = f"Help me write unit tests for the `{func_name}` function"