refactor: use snake case instead of camel case in types (#1894)
This commit is contained in:
parent
7080fcf1e6
commit
e94b386a13
5 changed files with 1063 additions and 1057 deletions
|
|
@ -48,8 +48,8 @@ def check_tasks_capability(
|
|||
if required.requests.sampling is not None:
|
||||
if client.requests.sampling is None:
|
||||
return False
|
||||
if required.requests.sampling.createMessage is not None:
|
||||
if client.requests.sampling.createMessage is None:
|
||||
if required.requests.sampling.create_message is not None:
|
||||
if client.requests.sampling.create_message is None:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
|
@ -74,7 +74,7 @@ def has_task_augmented_sampling(caps: ClientCapabilities) -> bool:
|
|||
return False
|
||||
if caps.tasks.requests.sampling is None:
|
||||
return False
|
||||
return caps.tasks.requests.sampling.createMessage is not None
|
||||
return caps.tasks.requests.sampling.create_message is not None
|
||||
|
||||
|
||||
def require_task_augmented_elicitation(client_caps: ClientCapabilities | None) -> None:
|
||||
|
|
|
|||
|
|
@ -41,7 +41,7 @@ class TaskContext:
|
|||
@property
|
||||
def task_id(self) -> str:
|
||||
"""The task identifier."""
|
||||
return self._task.taskId
|
||||
return self._task.task_id
|
||||
|
||||
@property
|
||||
def task(self) -> Task:
|
||||
|
|
|
|||
|
|
@ -125,12 +125,12 @@ def create_task_state(
|
|||
"""
|
||||
now = datetime.now(timezone.utc)
|
||||
return Task(
|
||||
taskId=task_id or generate_task_id(),
|
||||
task_id=task_id or generate_task_id(),
|
||||
status=TASK_STATUS_WORKING,
|
||||
createdAt=now,
|
||||
lastUpdatedAt=now,
|
||||
created_at=now,
|
||||
last_updated_at=now,
|
||||
ttl=metadata.ttl,
|
||||
pollInterval=500, # Default 500ms poll interval
|
||||
poll_interval=500, # Default 500ms poll interval
|
||||
)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -79,14 +79,14 @@ class InMemoryTaskStore(TaskStore):
|
|||
|
||||
task = create_task_state(metadata, task_id)
|
||||
|
||||
if task.taskId in self._tasks:
|
||||
raise ValueError(f"Task with ID {task.taskId} already exists")
|
||||
if task.task_id in self._tasks:
|
||||
raise ValueError(f"Task with ID {task.task_id} already exists")
|
||||
|
||||
stored = StoredTask(
|
||||
task=task,
|
||||
expires_at=self._calculate_expiry(metadata.ttl),
|
||||
)
|
||||
self._tasks[task.taskId] = stored
|
||||
self._tasks[task.task_id] = stored
|
||||
|
||||
# Return a copy to prevent external modification
|
||||
return Task(**task.model_dump())
|
||||
|
|
@ -124,10 +124,10 @@ class InMemoryTaskStore(TaskStore):
|
|||
status_changed = True
|
||||
|
||||
if status_message is not None:
|
||||
stored.task.statusMessage = status_message
|
||||
stored.task.status_message = status_message
|
||||
|
||||
# Update lastUpdatedAt on any change
|
||||
stored.task.lastUpdatedAt = datetime.now(timezone.utc)
|
||||
# Update last_updated_at on any change
|
||||
stored.task.last_updated_at = datetime.now(timezone.utc)
|
||||
|
||||
# If task is now terminal and has TTL, reset expiry timer
|
||||
if status is not None and is_terminal(status) and stored.task.ttl is not None:
|
||||
|
|
|
|||
|
|
@ -41,5 +41,5 @@ async def poll_until_terminal(
|
|||
if is_terminal(status.status):
|
||||
break
|
||||
|
||||
interval_ms = status.pollInterval if status.pollInterval is not None else default_interval_ms
|
||||
interval_ms = status.poll_interval if status.poll_interval is not None else default_interval_ms
|
||||
await anyio.sleep(interval_ms / 1000)
|
||||
|
|
|
|||
Loading…
Reference in a new issue