1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
from typing import Type, Union
from ...core.config import ContinueConfig
from ...core.main import History, Policy, Step
from ...core.observation import UserInputObservation
from ..steps.chat import SimpleChatStep
from ..steps.custom_command import CustomCommandStep
from ..steps.main import EditHighlightedCodeStep
from ..steps.steps_on_startup import StepsOnStartupStep
def parse_slash_command(inp: str, config: ContinueConfig) -> Union[None, Step]:
"""
Parses a slash command, returning the command name and the rest of the input.
"""
if inp.startswith("/"):
command_name = inp.split(" ")[0].strip()
after_command = " ".join(inp.split(" ")[1:])
for slash_command in config.slash_commands:
if slash_command.name == command_name[1:]:
params = slash_command.params
params["user_input"] = after_command
try:
return slash_command.step(**params)
except TypeError as e:
raise Exception(
f"Incorrect params used for slash command '{command_name}': {e}"
)
return None
def parse_custom_command(inp: str, config: ContinueConfig) -> Union[None, Step]:
command_name = inp.split(" ")[0].strip()
after_command = " ".join(inp.split(" ")[1:])
for custom_cmd in config.custom_commands:
if custom_cmd.name == command_name[1:]:
slash_command = parse_slash_command(custom_cmd.prompt, config)
if slash_command is not None:
return slash_command
return CustomCommandStep(
name=custom_cmd.name,
description=custom_cmd.description,
prompt=custom_cmd.prompt,
user_input=after_command,
slash_command=command_name,
)
return None
class DefaultPolicy(Policy):
default_step: Type[Step] = SimpleChatStep
default_params: dict = {}
def next(self, config: ContinueConfig, history: History) -> Step:
# At the very start, run initial Steps specified in the config
if history.get_current() is None:
return StepsOnStartupStep()
observation = history.get_current().observation
if observation is not None and isinstance(observation, UserInputObservation):
# This could be defined with ObservationTypePolicy. Ergonomics not right though.
user_input = observation.user_input
slash_command = parse_slash_command(user_input, config)
if slash_command is not None:
if (
getattr(slash_command, "user_input", None) is None
and history.get_current().step.user_input is not None
):
history.get_current().step.user_input = (
history.get_current().step.user_input.split()[0]
)
return slash_command
custom_command = parse_custom_command(user_input, config)
if custom_command is not None:
return custom_command
if user_input.startswith("/edit"):
return EditHighlightedCodeStep(user_input=user_input[5:])
return self.default_step(**self.default_params)
return None
|
