Skip to content

Chat

Chat

Source code in griptape/griptape/utils/chat.py 
 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
@define(frozen=True)
class Chat:
    structure: Structure = field()
    exit_keywords: list[str] = field(default=["exit"], kw_only=True)
    exiting_text: str = field(default="exiting...", kw_only=True)
    processing_text: str = field(default="processing...", kw_only=True)
    intro_text: Optional[str] = field(default=None, kw_only=True)
    prompt_prefix: str = field(default="Q: ", kw_only=True)
    response_prefix: str = field(default="A: ", kw_only=True)
    output_fn: Callable[[str], None] = field(default=Factory(lambda: print), kw_only=True)

    def start(self) -> None:
        if self.intro_text:
            self.output_fn(self.intro_text)
        while True:
            question = input(self.prompt_prefix)

            if question.lower() in self.exit_keywords:
                self.output_fn(self.exiting_text)

                break
            else:
                self.output_fn(self.processing_text)

            self.output_fn(f"{self.response_prefix}{self.structure.run(question).output_task.output.to_text()}")

exit_keywords: list[str] = field(default=['exit'], kw_only=True) class-attribute instance-attribute

exiting_text: str = field(default='exiting...', kw_only=True) class-attribute instance-attribute

intro_text: Optional[str] = field(default=None, kw_only=True) class-attribute instance-attribute

output_fn: Callable[[str], None] = field(default=Factory(lambda : print), kw_only=True) class-attribute instance-attribute

processing_text: str = field(default='processing...', kw_only=True) class-attribute instance-attribute

prompt_prefix: str = field(default='Q: ', kw_only=True) class-attribute instance-attribute

response_prefix: str = field(default='A: ', kw_only=True) class-attribute instance-attribute

structure: Structure = field() class-attribute instance-attribute

start()

Source code in griptape/griptape/utils/chat.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
def start(self) -> None:
    if self.intro_text:
        self.output_fn(self.intro_text)
    while True:
        question = input(self.prompt_prefix)

        if question.lower() in self.exit_keywords:
            self.output_fn(self.exiting_text)

            break
        else:
            self.output_fn(self.processing_text)

        self.output_fn(f"{self.response_prefix}{self.structure.run(question).output_task.output.to_text()}")