diff --git a/poetry.lock b/poetry.lock
index 89b1ef3..ad22e2d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1752,13 +1752,13 @@ adal = ["adal (>=1.0.2)"]
 
 [[package]]
 name = "litellm"
-version = "1.48.7"
+version = "1.48.11"
 description = "Library to easily interface with LLM API providers"
 optional = false
 python-versions = "!=2.7.*,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,!=3.7.*,>=3.8"
 files = [
-    {file = "litellm-1.48.7-py3-none-any.whl", hash = "sha256:4971a9e681188635c2ee6dc44fe35bb2774586e9018682adcccdbb516b839c64"},
-    {file = "litellm-1.48.7.tar.gz", hash = "sha256:ff1fef7049e9afa09598f98d1e510a6d5f252ec65c0526b8bfaf13eadfcf65e5"},
+    {file = "litellm-1.48.11-py3-none-any.whl", hash = "sha256:6e38c83430b40a9d4afc2974daad4ead198e18facb9fd8923b0590ae56b30acd"},
+    {file = "litellm-1.48.11.tar.gz", hash = "sha256:7c0a4622dfbab4099effb11fce02eb0ba64788b6465ed5f2de5ddcea8a4e80d2"},
 ]
 
 [package.dependencies]
@@ -1767,7 +1767,7 @@ click = "*"
 importlib-metadata = ">=6.8.0"
 jinja2 = ">=3.1.2,<4.0.0"
 jsonschema = ">=4.22.0,<5.0.0"
-openai = ">=1.45.0"
+openai = ">=1.51.0"
 pydantic = ">=2.0.0,<3.0.0"
 python-dotenv = ">=0.2.0"
 requests = ">=2.31.0,<3.0.0"
@@ -2080,13 +2080,13 @@ signedtoken = ["cryptography (>=3.0.0)", "pyjwt (>=2.0.0,<3)"]
 
 [[package]]
 name = "openai"
-version = "1.50.2"
+version = "1.51.0"
 description = "The official Python library for the openai API"
 optional = false
 python-versions = ">=3.7.1"
 files = [
-    {file = "openai-1.50.2-py3-none-any.whl", hash = "sha256:822dd2051baa3393d0d5406990611975dd6f533020dc9375a34d4fe67e8b75f7"},
-    {file = "openai-1.50.2.tar.gz", hash = "sha256:3987ae027152fc8bea745d60b02c8f4c4a76e1b5c70e73565fa556db6f78c9e6"},
+    {file = "openai-1.51.0-py3-none-any.whl", hash = "sha256:d9affafb7e51e5a27dce78589d4964ce4d6f6d560307265933a94b2e3f3c5d2c"},
+    {file = "openai-1.51.0.tar.gz", hash = "sha256:8dc4f9d75ccdd5466fc8c99a952186eddceb9fd6ba694044773f3736a847149d"},
 ]
 
 [package.dependencies]
@@ -3275,13 +3275,13 @@ files = [
 
 [[package]]
 name = "skillpacks"
-version = "0.1.60"
+version = "0.1.64"
 description = "Pluggable skills for AI agents"
 optional = false
 python-versions = "<4.0,>=3.10"
 files = [
-    {file = "skillpacks-0.1.60-py3-none-any.whl", hash = "sha256:a9b75721e7f4da6d94a23b52ded78ab41ac696fdc0723e7a816184a17de62a66"},
-    {file = "skillpacks-0.1.60.tar.gz", hash = "sha256:0fb989fd2114a556a245d6f7b3bdbf53e1af30f52370dcd6720518e16a67a45b"},
+    {file = "skillpacks-0.1.64-py3-none-any.whl", hash = "sha256:5d9c209c6b22eaecfc0a332420dfb5d62ae91ac1f1395d2860d66a337c97b711"},
+    {file = "skillpacks-0.1.64.tar.gz", hash = "sha256:8cf63c2a7d1b2563f5abe1ba3a46ddc6b158a5aefbf2f2af828298b72adf3eb0"},
 ]
 
 [package.dependencies]
@@ -4270,4 +4270,4 @@ runtime = ["docker", "google-auth", "google-cloud-container", "kubernetes"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "69cc416208fcc7104af707a757276878fdbb4ed629d01f0dbb14d38206326b46"
+content-hash = "dc727237ce84c8488b77bc3364e3915e0bd5de36ab00d469d8b1aef2a32101dc"
diff --git a/pyproject.toml b/pyproject.toml
index d745321..6a3a58c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "taskara"
-version = "0.1.152"
+version = "0.1.153"
 description = "Task management for AI agents"
 authors = ["Patrick Barker <patrickbarkerco@gmail.com>"]
 license = "MIT"
@@ -21,7 +21,7 @@ tabulate = {version = "^0.9.0", optional = true}
 shortuuid = "^1.0.13"
 tqdm = "^4.66.4"
 cryptography = "^43.0.1"
-skillpacks = "^0.1.60"
+skillpacks = "^0.1.64"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.1.1"
diff --git a/taskara/db/models.py b/taskara/db/models.py
index 218b26e..c5f2a8e 100644
--- a/taskara/db/models.py
+++ b/taskara/db/models.py
@@ -22,7 +22,6 @@
     Column("task_id", String, ForeignKey("tasks.id"), primary_key=True),
 )
 
-
 # Association table for many-to-many between tasks and tags
 task_tag_association = Table(
     "task_tag_association",
diff --git a/taskara/task.py b/taskara/task.py
index 055307e..7ba5a10 100644
--- a/taskara/task.py
+++ b/taskara/task.py
@@ -22,8 +22,8 @@
     V1Action,
     V1Episode,
     V1ToolRef,
-    V1EnvState,
     Review,
+    EnvState,
 )
 from threadmem import RoleMessage, RoleThread, V1RoleThreads
 from threadmem.server.models import V1RoleMessage
@@ -700,11 +700,11 @@ def _get_episode(
 
     def record_action(
         self,
-        state: V1EnvState,
+        state: EnvState,
         action: V1Action,
         tool: V1ToolRef,
         result: Optional[Any] = None,
-        end_state: Optional[V1EnvState] = None,
+        end_state: Optional[EnvState] = None,
         prompt: Optional[Prompt | str] = None,
         namespace: str = "default",
         metadata: dict = {},
diff --git a/tests/test_runtime.py b/tests/test_runtime.py
index 9576dbd..2bb8219 100644
--- a/tests/test_runtime.py
+++ b/tests/test_runtime.py
@@ -5,7 +5,7 @@
 from mllm import Prompt, RoleMessage, RoleThread
 from namesgenerator import get_random_name
 from openai import BaseModel
-from skillpacks import ActionEvent, V1Action, V1EnvState
+from skillpacks import ActionEvent, V1Action, V1EnvState, EnvState
 from skillpacks.server.models import V1Episode, V1ActionEvents
 from toolfuse.models import V1ToolRef
 
@@ -245,7 +245,7 @@ def test_process_tracker_runtime():
 
         # Store an action event
         action_event = ActionEvent(
-            state=V1EnvState(images=["https://test.img"]),
+            state=EnvState(images=["https://test.img"]),
             action=V1Action(name="test", parameters={}),
             tool=V1ToolRef(module="test", type="test"),
             prompt=prompt,
@@ -294,7 +294,7 @@ class Expected(BaseModel):
         print("created a new task: ", new_task.id)
 
         action_event = ActionEvent(
-            state=V1EnvState(images=["https://test.img"]),
+            state=EnvState(images=["https://test.img"]),
             action=V1Action(name="test", parameters={}),
             tool=V1ToolRef(module="test", type="test"),
             prompt=prompt,