steven_li
6e9e74d1ee
添加 RuntimeContext 类用于捕获模型运行时的日期时间信息， 包括UTC时间、本地时间和时区信息，并在系统提示中显示这些信息。 同时增加最大上下文消息数和工具迭代次数的配置选项， 将验证服务从引擎加载器中移除，并更新相关的数据结构和接口。 BREAKING CHANGE: 移除了验证服务，相关字段被替换为证据状态和接受状态。 - 添加 RuntimeContext 类和相关渲染方法 - 增加 max_context_messages 和 max_tool_iterations 配置 - 移除 ValidationService 相关代码 - 更新消息记录中的验证状态字段 - 添加原始工具调用检测和回退处理
154 lines
4.2 KiB
Python
154 lines
4.2 KiB
Python
"""Chat-related web schemas."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
try:
|
|
from pydantic import BaseModel, Field
|
|
except ModuleNotFoundError: # pragma: no cover - fallback for skeleton-only environments
|
|
class BaseModel:
|
|
"""Very small fallback shim used only so imports work without pydantic."""
|
|
|
|
def __init__(self, **kwargs: Any) -> None:
|
|
annotations = getattr(self.__class__, "__annotations__", {})
|
|
for name in annotations:
|
|
default = getattr(self.__class__, name, None)
|
|
if name in kwargs:
|
|
value = kwargs[name]
|
|
else:
|
|
value = default
|
|
setattr(self, name, value)
|
|
|
|
def model_dump(self, *, exclude_none: bool = False) -> dict[str, Any]:
|
|
data = dict(self.__dict__)
|
|
if exclude_none:
|
|
data = {key: value for key, value in data.items() if value is not None}
|
|
return data
|
|
|
|
def Field(default: Any = None, **kwargs: Any) -> Any:
|
|
default_factory = kwargs.get("default_factory")
|
|
if default_factory is not None:
|
|
return default_factory()
|
|
return default
|
|
|
|
|
|
class WebProviderTarget(BaseModel):
|
|
"""Web-facing provider target shape.
|
|
|
|
先保持和 runtime 里的 `ProviderTarget` 接近，但只暴露 Web 当前需要的字段。
|
|
后面如果 provider 层扩字段，再由这里显式补齐。
|
|
"""
|
|
|
|
provider: str | None = None
|
|
model: str | None = None
|
|
api_key: str | None = None
|
|
api_base: str | None = None
|
|
extra_headers: dict[str, str] | None = None
|
|
|
|
|
|
class WebChatRequest(BaseModel):
|
|
"""最小正式 chat 请求结构。"""
|
|
|
|
message: str = Field(min_length=1)
|
|
session_id: str | None = None
|
|
user_id: str | None = None
|
|
title: str | None = None
|
|
execution_context: str | None = None
|
|
model: str | None = None
|
|
provider_name: str | None = None
|
|
embedding_model: str | None = None
|
|
temperature: float | None = None
|
|
max_tokens: int | None = None
|
|
thinking_enabled: bool | None = None
|
|
max_tool_iterations: int | None = None
|
|
fallback_target: WebProviderTarget | None = None
|
|
auxiliary_target: WebProviderTarget | None = None
|
|
embedding_target: WebProviderTarget | None = None
|
|
reply_to_scheduled_run_id: str | None = None
|
|
scheduled_reply_intent: str | None = None
|
|
|
|
|
|
class WebChatResponse(BaseModel):
|
|
"""最小正式 chat 响应结构。"""
|
|
|
|
session_id: str
|
|
run_id: str
|
|
output_text: str
|
|
finish_reason: str
|
|
tool_iterations: int
|
|
provider_name: str | None = None
|
|
model: str | None = None
|
|
usage: dict[str, Any] = Field(default_factory=dict)
|
|
task_id: str | None = None
|
|
task_status: str | None = None
|
|
evidence_status: str | None = None
|
|
acceptance_state: str | None = None
|
|
validation_result: dict[str, Any] | None = None
|
|
|
|
|
|
class WebChatAcceptanceRequest(BaseModel):
|
|
"""User acceptance on the latest assistant result in chat."""
|
|
|
|
session_id: str
|
|
run_id: str
|
|
acceptance_type: str
|
|
comment: str | None = None
|
|
|
|
|
|
class WebChatAcceptanceResponse(BaseModel):
|
|
"""Acceptance recording result."""
|
|
|
|
session_id: str
|
|
run_id: str
|
|
task_id: str
|
|
task_status: str
|
|
acceptance_type: str
|
|
feedback_type: str
|
|
learning_candidates: list[dict[str, Any]] = Field(default_factory=list)
|
|
|
|
|
|
class WebChatFeedbackRequest(BaseModel):
|
|
"""Backward-compatible feedback payload."""
|
|
|
|
session_id: str
|
|
run_id: str
|
|
feedback_type: str
|
|
comment: str | None = None
|
|
|
|
|
|
class WebChatFeedbackResponse(WebChatAcceptanceResponse):
|
|
"""Backward-compatible feedback response."""
|
|
|
|
|
|
class WebProviderConfigRequest(BaseModel):
|
|
"""Provider config update from the status page."""
|
|
|
|
enabled: bool = True
|
|
model: str | None = None
|
|
api_key: str | None = None
|
|
api_base: str | None = None
|
|
request_timeout_seconds: float | None = None
|
|
|
|
|
|
class WebProviderConfigResponse(BaseModel):
|
|
"""Provider config update result."""
|
|
|
|
ok: bool
|
|
provider: str
|
|
enabled: bool
|
|
|
|
|
|
class WebStatusResponse(BaseModel):
|
|
"""Web 宿主层状态响应。"""
|
|
|
|
status: str
|
|
running: bool
|
|
mode: str
|
|
|
|
|
|
class WebErrorResponse(BaseModel):
|
|
"""统一错误响应结构。"""
|
|
|
|
detail: str
|