cranberrypy/chat.py at master · exante/cranberrypy · GitHub

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
# Copyright (C) 2025 EXANTE

import sys

from core.configuration.chat import ChatConfig
from core.logger import Logger

CONFIG = ChatConfig('cranberrypy.ini')

LOGGER = Logger(config=CONFIG, name=__name__)
LOGGER.setup_logger(stream=sys.stdout)


SYSTEM_PROMPT = """You are a helpful assistant, you will use the provided context to answer user question.
Read the given context before answering questions and think step by step. If you can`t answer a user question based on
the provided context, inform the user. Do not use any other information for answering user."""

ADDITIONAL_PROMPT = """"""

MODULE_PATHS = """
/path_to/root_directory/your_project/module_1
/path_to/root_directory/your_project/module_2
"""

TEST_CONDITIONS = """
For configuration use function 'get_config' in module 'your_project.tests.utils'.
"""


def run_chat():
    from core.assistant.ai import AI

    ai = AI(config=CONFIG)

    # ai.chat()
    # ai.chat_with_persistent_context(module_paths=MODULE_PATHS)
    # ai.chat_with_current_context(module_paths=MODULE_PATHS)
    #
    # ai.generate_code_according_to_diagram(using_project_context=True)

    # ai.generate_documentation_with_current_context()
    ai.generate_documentation(
        description=True,
        code=True,
        dependencies=True,
        contain_code_text=False
    )

    # ai.generate_unit_tests(
    #     conditions=TEST_CONDITIONS,
    #     using_pytest=True
    # )
    # ai.generate_unit_tests_with_persistent_context(
    #     conditions=TEST_CONDITIONS,
    #     using_pytest=True
    # )
    # ai.generate_unit_tests_with_current_context(
    #     conditions=TEST_CONDITIONS,
    #     using_pytest=True
    # )


if __name__ == "__main__":
    run_chat()