mirror of
https://github.com/TheR1D/shell_gpt.git
synced 2026-07-03 14:10:18 +02:00
Farkhod Sadykov
763fef15d3
* Refactoring, more tests, lint fixes, optimise * Lint CI fixes * Changing version, readme help fixes * Validation rule, validation tests, minor improvements
31 lines
845 B
Python
31 lines
845 B
Python
from pathlib import Path
|
|
|
|
from sgpt import OpenAIClient, config, make_prompt
|
|
from sgpt.utils import CompletionModes
|
|
from .handler import Handler
|
|
|
|
CHAT_CACHE_LENGTH = int(config.get("CHAT_CACHE_LENGTH"))
|
|
CHAT_CACHE_PATH = Path(config.get("CHAT_CACHE_PATH"))
|
|
|
|
|
|
class DefaultHandler(Handler):
|
|
def __init__(
|
|
self,
|
|
client: OpenAIClient,
|
|
shell: bool = False,
|
|
code: bool = False,
|
|
model: str = "gpt-3.5-turbo",
|
|
) -> None:
|
|
super().__init__(client)
|
|
self.client = client
|
|
self.mode = CompletionModes.get_mode(shell, code)
|
|
self.model = model
|
|
|
|
def make_prompt(self, prompt) -> str:
|
|
prompt = prompt.strip()
|
|
return make_prompt.initial(
|
|
prompt,
|
|
self.mode == CompletionModes.SHELL,
|
|
self.mode == CompletionModes.CODE,
|
|
)
|