feat: + annotations

This commit is contained in:
莘权 马 2023-08-28 19:47:57 +08:00
parent 27561765cf
commit 946e6fa8b3
2 changed files with 6 additions and 7 deletions

View file

@ -4,7 +4,7 @@
@Time : 2023/5/1 11:59
@Author : alexanderwu
@File : const.py'
@Modified By: mashenquan, 2023/8/28. Add 'OPTIONS', 'DEFAULT_LANGUAGE', 'DEFAULT_MAX_TOKENS'
@Modified By: mashenquan, 2023/8/28. Add 'OPTIONS', 'DEFAULT_LANGUAGE', 'DEFAULT_MAX_TOKENS'...
"""
import contextvars
from pathlib import Path
@ -40,4 +40,7 @@ MEM_TTL = 24 * 30 * 3600
OPTIONS = contextvars.ContextVar("OPTIONS")
DEFAULT_LANGUAGE = "English"
DEFAULT_MAX_TOKENS = 1500
DEFAULT_MAX_TOKENS = 1500
COMMAND_TOKENS = 500
BRAIN_MEMORY = "BRAIN_MEMORY"
SKILL_PATH = "SKILL_PATH"

View file

@ -21,6 +21,7 @@ from metagpt.actions import ActionOutput
from metagpt.actions.skill_action import SkillAction, ArgumentsParingAction
from metagpt.actions.talk_action import TalkAction
from metagpt.config import Config
from metagpt.const import BRAIN_MEMORY, SKILL_PATH
from metagpt.learn.skill_loader import SkillLoader
from metagpt.logs import logger
from metagpt.memory.brain_memory import BrainMemory, MessageType
@ -28,11 +29,6 @@ from metagpt.provider.openai_api import CostManager
from metagpt.roles import Role
from metagpt.schema import Message
DEFAULT_MAX_TOKENS = 1500
COMMAND_TOKENS = 500
BRAIN_MEMORY = "BRAIN_MEMORY"
SKILL_PATH = "SKILL_PATH"
class Assistant(Role):
"""Assistant for solving common issues."""