blob: 7524bd8b91f33a746043710ed70df792a21c3a52 (
plain)
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
|
from ..forge_log import ForgeLogger
from .registry import ability
logger = ForgeLogger(__name__)
@ability(
name="finish",
description="Use this to shut down once you have accomplished all of your goals,"
" or when there are insurmountable problems that make it impossible"
" for you to finish your task.",
parameters=[
{
"name": "reason",
"description": "A summary to the user of how the goals were accomplished",
"type": "string",
"required": True,
}
],
output_type="None",
)
async def finish(
agent,
task_id: str,
reason: str,
) -> str:
"""
A function that takes in a string and exits the program
Parameters:
reason (str): A summary to the user of how the goals were accomplished.
Returns:
A result string from create chat completion. A list of suggestions to
improve the code.
"""
logger.info(reason, extra={"title": "Shutting down...\n"})
return reason
|