8ea243c572
Agent应用中添加知识库的配置字段（提示词修改、反思给默认值） * fix/develop_kj_knowledge: (1 commits squashed) - Agent应用中添加知识库的配置字段（提示词修改、反思给默认值） Signed-off-by: aliyun8644380055 <accounts_68c0f5d519f260d93ee2997e@mail.teambition.com> Reviewed-by: aliyun6762716068 <accounts_68cb7c6b61f5dcc4200d6251@mail.teambition.com> Merged-by: aliyun6762716068 <accounts_68cb7c6b61f5dcc4200d6251@mail.teambition.com> CR-link: https://codeup.aliyun.com/redbearai/python/redbear-mem-open/change/87
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from pydantic import BaseModel, Field
|
|
from typing import Optional
|
|
from enum import Enum
|
|
|
|
|
|
class OptimizationStrategy(str, Enum):
|
|
"""优化策略枚举"""
|
|
SPEED_FIRST = "speed_first"
|
|
ACCURACY_FIRST = "accuracy_first"
|
|
BALANCED = "balanced"
|
|
class Memory_Reflection(BaseModel):
|
|
config_id: Optional[int] = None
|
|
reflection_enabled: bool
|
|
reflection_period_in_hours: str
|
|
reflexion_range: Optional[str] = "partial"
|
|
baseline: Optional[str] = "TIME"
|
|
reflection_model_id: str
|
|
memory_verify: bool
|
|
quality_assessment: bool
|
|
|
|
# 新增快速引擎优化参数
|
|
optimization_strategy: Optional[OptimizationStrategy] = OptimizationStrategy.BALANCED
|
|
use_fast_model: Optional[bool] = True
|
|
enable_caching: Optional[bool] = True
|
|
enable_streaming: Optional[bool] = True
|
|
batch_size: Optional[int] = Field(default=3, ge=1, le=10)
|
|
max_concurrent: Optional[int] = Field(default=5, ge=1, le=20)
|
|
|
|
class Config:
|
|
use_enum_values = True
|
|
|
|
|
|
class FastReflectionRequest(BaseModel):
|
|
"""快速反思请求模型"""
|
|
reflection: Memory_Reflection
|
|
host_id: Optional[str] = "88a459f5_text02"
|
|
optimization_strategy: Optional[OptimizationStrategy] = OptimizationStrategy.BALANCED
|
|
|
|
class Config:
|
|
use_enum_values = True
|
|
|
|
|
|
class ReflectionBenchmarkRequest(BaseModel):
|
|
"""反思基准测试请求模型"""
|
|
reflection: Memory_Reflection
|
|
host_id: Optional[str] = "88a459f5_text02"
|
|
iterations: Optional[int] = Field(default=3, ge=1, le=10)
|
|
|
|
class Config:
|
|
use_enum_values = True
|
|
|
|
|