xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 12:50:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

publish_output

Browse Source
This commit is contained in:
mrT23
2024-04-08 14:00:41 +03:00
parent a543d7ed1a
commit 84d8f78d0c
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/utils.py
View File

@ -575,7 +575,7 @@ def clip_tokens(text: str, max_tokens: int, add_three_dots=True) -> str:
num_output_chars = int(chars_per_token * max_tokens) num_output_chars = int(chars_per_token * max_tokens)
clipped_text = text[:num_output_chars] clipped_text = text[:num_output_chars]
if add_three_dots: if add_three_dots:
clipped_text += " ...(truncated)" clipped_text += "\n...(truncated)"
return clipped_text return clipped_text
except Exception as e: except Exception as e:
get_logger().warning(f"Failed to clip tokens: {e}") get_logger().warning(f"Failed to clip tokens: {e}")

3
pr_agent/tools/pr_reviewer.py
View File

@ -357,6 +357,9 @@ class PRReviewer:
return True return True
def set_review_labels(self, data): def set_review_labels(self, data):
if not get_settings().config.publish_output:
return
if (get_settings().pr_reviewer.enable_review_labels_security or if (get_settings().pr_reviewer.enable_review_labels_security or
get_settings().pr_reviewer.enable_review_labels_effort): get_settings().pr_reviewer.enable_review_labels_effort):
try: try:

2
tests/unittest/test_clip_tokens.py
View File

@ -15,5 +15,5 @@ class TestClipTokens:
max_tokens = 10 max_tokens = 10
result = clip_tokens(text, max_tokens) result = clip_tokens(text, max_tokens)
expected_results = 'line1\nline2\nline3\nli ...(truncated)' expected_results = 'line1\nline2\nline3\nli\n...(truncated)'
assert result == expected_results assert result == expected_results