mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-16 04:24:27 +01:00
3fe2246468
* Add categories to command registry * Fix tests * Clean up prompt generation * Rename Performance Evaluations to Best Practices * Move specification of response format from system prompt to Agent.construct_base_prompt * Clean up PromptGenerator class * Add debug logging to AIConfig autogeneration * Clarify prompting and add support for multiple thought processes to Agent * WIP: PlanningAgent * Disable message history by default on BaseAgent * Add CommandOutput and ThoughtProcessOutput type aliases * Fix interrupts in main.py * Use custom exceptions and clean up exception/error handling * Remove duplicate agent_history.py * Update PlanningAgent from upstream * WIP: Support for dynamic in-prompt context * WIP: response formats for PlanningAgent three-stage cycle * Remove browsing overlay & separate browsing from extraction code * Fix human feedback * Fix tests * Include history in Agent prompt generation * Code improvements in agent.py * Add ask_user command and revise system prompt
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import functools
|
|
from typing import TYPE_CHECKING, Any, Callable, Optional, TypedDict
|
|
|
|
if TYPE_CHECKING:
|
|
from autogpt.config import Config
|
|
|
|
from autogpt.models.command import Command, CommandOutput, CommandParameter
|
|
|
|
# Unique identifier for auto-gpt commands
|
|
AUTO_GPT_COMMAND_IDENTIFIER = "auto_gpt_command"
|
|
|
|
|
|
class CommandParameterSpec(TypedDict):
|
|
type: str
|
|
description: str
|
|
required: bool
|
|
|
|
|
|
def command(
|
|
name: str,
|
|
description: str,
|
|
parameters: dict[str, CommandParameterSpec],
|
|
enabled: bool | Callable[[Config], bool] = True,
|
|
disabled_reason: Optional[str] = None,
|
|
aliases: list[str] = [],
|
|
) -> Callable[..., CommandOutput]:
|
|
"""The command decorator is used to create Command objects from ordinary functions."""
|
|
|
|
def decorator(func: Callable[..., CommandOutput]) -> Command:
|
|
typed_parameters = [
|
|
CommandParameter(
|
|
name=param_name,
|
|
description=parameter.get("description"),
|
|
type=parameter.get("type", "string"),
|
|
required=parameter.get("required", False),
|
|
)
|
|
for param_name, parameter in parameters.items()
|
|
]
|
|
cmd = Command(
|
|
name=name,
|
|
description=description,
|
|
method=func,
|
|
parameters=typed_parameters,
|
|
enabled=enabled,
|
|
disabled_reason=disabled_reason,
|
|
aliases=aliases,
|
|
)
|
|
|
|
@functools.wraps(func)
|
|
def wrapper(*args, **kwargs) -> Any:
|
|
return func(*args, **kwargs)
|
|
|
|
wrapper.command = cmd
|
|
|
|
setattr(wrapper, AUTO_GPT_COMMAND_IDENTIFIER, True)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|