Skip to content

Commit 665823b

Browse files
committed
refactor common module
1 parent 1828b03 commit 665823b

File tree

13 files changed

+23
-17
lines changed

13 files changed

+23
-17
lines changed
File renamed without changes.

patchwork/steps/ResolveIssue/multiturn_strategy/analyze_implement.py renamed to patchwork/common/multiturn_strategy/analyze_implement.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,8 @@
1010
from openai.types.chat import ChatCompletionMessageParam
1111

1212
from patchwork.common.client.llm.protocol import LlmClient
13-
from patchwork.steps.ResolveIssue.multiturn_strategy.multiturn_strategy import (
14-
MultiturnStrategy,
15-
)
16-
from patchwork.steps.ResolveIssue.tools.tool import Tool
13+
from patchwork.common.multiturn_strategy.multiturn_strategy import MultiturnStrategy
14+
from patchwork.common.tools import Tool
1715

1816

1917
class STAGE(Enum):

patchwork/steps/ResolveIssue/multiturn_strategy/multiturn_strategy.py renamed to patchwork/common/multiturn_strategy/multiturn_strategy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from openai.types.chat.chat_completion_message_param import ChatCompletionMessageParam
1111
from openai.types.chat.chat_completion_tool_param import ChatCompletionToolParam
1212

13-
from patchwork.steps.ResolveIssue.tools.tool import Tool
13+
from patchwork.common.tools import Tool
1414

1515
logger = logging.getLogger("patched")
1616

patchwork/common/tools/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from patchwork.common.tools.bash_tool import BashTool
2+
from patchwork.common.tools.code_edit_tools import CodeEditTool
3+
from patchwork.common.tools.tool import Tool
4+
5+
__all__ = [
6+
"Tool",
7+
"CodeEditTool",
8+
"BashTool",
9+
]

patchwork/steps/ResolveIssue/tools/bash_tool.py renamed to patchwork/common/tools/bash_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import subprocess
44
from pathlib import Path
55

6-
from patchwork.steps.ResolveIssue.tools.tool import Tool
6+
from patchwork.common.tools import Tool
77

88

99
class BashTool(Tool, tool_name="bash"):

patchwork/steps/ResolveIssue/tools/code_edit_tools.py renamed to patchwork/common/tools/code_edit_tools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from pathlib import Path
55
from typing import Literal
66

7-
from patchwork.steps.ResolveIssue.tools.tool import Tool
7+
from patchwork.common.tools import Tool
88

99

1010
class CodeEditTool(Tool, tool_name="code_edit_tool"):
File renamed without changes.

patchwork/steps/ResolveIssue/ResolveIssue.py renamed to patchwork/steps/FixIssue/FixIssue.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,14 +8,13 @@
88
from patchwork.common.client.llm.aio import AioLlmClient
99
from patchwork.common.client.llm.protocol import LlmClient
1010
from patchwork.common.constants import TOKEN_URL
11-
from patchwork.step import Step
12-
from patchwork.steps.ResolveIssue.multiturn_strategy.analyze_implement import (
11+
from patchwork.common.multiturn_strategy.analyze_implement import (
1312
STAGE,
1413
AnalyzeImplementStrategy,
1514
)
16-
from patchwork.steps.ResolveIssue.tools.code_edit_tools import CodeEditTool
17-
from patchwork.steps.ResolveIssue.tools.tool import Tool
18-
from patchwork.steps.ResolveIssue.typed import ResolveIssueInputs, ResolveIssueOutputs
15+
from patchwork.common.tools import CodeEditTool, Tool
16+
from patchwork.step import Step
17+
from patchwork.steps.FixIssue.typed import FixIssueInputs, FixIssueOutputs
1918

2019

2120
class _ResolveIssue(AnalyzeImplementStrategy):
@@ -93,7 +92,7 @@ def is_stop(self, messages: list[ChatCompletionMessageParam]) -> bool:
9392
return "</DONE>" in last_message.get("content")
9493

9594

96-
class ResolveIssue(Step, input_class=ResolveIssueInputs, output_class=ResolveIssueOutputs):
95+
class FixIssue(Step, input_class=FixIssueInputs, output_class=FixIssueOutputs):
9796
def __init__(self, inputs):
9897
super().__init__(inputs)
9998
self.base_path = inputs.get("base_path")

patchwork/steps/ResolveIssue/multiturn_strategy/__init__.py renamed to patchwork/steps/FixIssue/__init__.py

File renamed without changes.

0 commit comments

Comments
 (0)