> ## Documentation Index
> Fetch the complete documentation index at: /docs/llms.txt
> Use this file to discover all available pages before exploring further.

# task_workers

# `prefect.server.models.task_workers`

## Functions

### `observe_worker` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L86](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L86)</sup>

```python theme={null}
observe_worker(task_keys: List[TaskKey], worker_id: WorkerId) -> None
```

### `forget_worker` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L93](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L93)</sup>

```python theme={null}
forget_worker(worker_id: WorkerId) -> None
```

### `get_workers_for_task_keys` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L99](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L99)</sup>

```python theme={null}
get_workers_for_task_keys(task_keys: List[TaskKey]) -> List[TaskWorkerResponse]
```

### `get_all_workers` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L105](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L105)</sup>

```python theme={null}
get_all_workers() -> List[TaskWorkerResponse]
```

## Classes

### `TaskWorkerResponse` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L16](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L16)</sup>

### `InMemoryTaskWorkerTracker` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L22](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L22)</sup>

**Methods:**

#### `forget_worker` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L39](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L39)</sup>

```python theme={null}
forget_worker(self, worker_id: WorkerId) -> None
```

#### `get_all_workers` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L60](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L60)</sup>

```python theme={null}
get_all_workers(self) -> List[TaskWorkerResponse]
```

#### `get_workers_for_task_keys` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L51](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L51)</sup>

```python theme={null}
get_workers_for_task_keys(self, task_keys: List[TaskKey]) -> List[TaskWorkerResponse]
```

#### `observe_worker` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L28](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L28)</sup>

```python theme={null}
observe_worker(self, task_keys: List[TaskKey], worker_id: WorkerId) -> None
```

#### `reset` <sup>[https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task\_workers.py#L74](https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L74)</sup>

```python theme={null}
reset(self) -> None
```

Testing utility to reset the state of the task worker tracker