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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import os
from textwrap import dedent
from typing import Type, Union
from ...core.config import ContinueConfig
from ...core.main import History, Policy, Step
from ...core.observation import UserInputObservation
from ...libs.util.paths import getServerFolderPath
from ..steps.chat import SimpleChatStep
from ..steps.core.core import MessageStep
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 spcecified in the config
if history.get_current() is None:
shown_welcome_file = os.path.join(getServerFolderPath(), ".shown_welcome")
if os.path.exists(shown_welcome_file):
return StepsOnStartupStep()
with open(shown_welcome_file, "w") as f:
f.write("")
return (
MessageStep(
name="Welcome to Continue",
message=dedent(
"""\
- Highlight code section and ask a question or give instructions
- Use `cmd+m` (Mac) / `ctrl+m` (Windows) to open Continue
- Use `/help` to ask questions about how to use Continue
- [Customize Continue](https://continue.dev/docs/customization) (e.g. use your own API key) by typing '/config'."""
),
)
>> 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
|
