new observer thoughts (#1442)

This commit is contained in:
Shuchang Zheng
2024-12-27 09:04:09 -08:00
committed by GitHub
parent d03957d590
commit 9e6c2362bf
6 changed files with 198 additions and 17 deletions

View File

@@ -1,5 +1,6 @@
from datetime import datetime
from enum import StrEnum
from typing import Any
from pydantic import BaseModel, ConfigDict, HttpUrl, field_validator
@@ -35,6 +36,22 @@ class ObserverCruise(BaseModel):
modified_at: datetime
class ObserverThoughtType(StrEnum):
plan = "plan"
metadata = "metadata"
user_goal_check = "user_goal_check"
internal_plan = "internal_plan"
class ObserverThoughtScenario(StrEnum):
generate_plan = "generate_plan"
user_goal_check = "user_goal_check"
generate_metadata = "generate_metadata"
extract_loop_values = "extract_loop_values"
generate_task_in_loop = "generate_task_in_loop"
generate_task = "generate_general_task"
class ObserverThought(BaseModel):
model_config = ConfigDict(from_attributes=True)
@@ -49,6 +66,9 @@ class ObserverThought(BaseModel):
observation: str | None = None
thought: str | None = None
answer: str | None = None
observer_thought_type: ObserverThoughtType | None = ObserverThoughtType.plan
observer_thought_scenario: ObserverThoughtScenario | None = None
output: dict[str, Any] | None = None
created_at: datetime
modified_at: datetime