diff --git a/poetry.lock b/poetry.lock index afccb50..3942c49 100644 --- a/poetry.lock +++ b/poetry.lock @@ -16,13 +16,13 @@ pydantic = ">=2.8.2,<3.0.0" [[package]] name = "agentdesk" -version = "0.2.134" +version = "0.2.135" description = "A desktop for AI agents" optional = false python-versions = "<4.0,>=3.10" files = [ - {file = "agentdesk-0.2.134-py3-none-any.whl", hash = "sha256:ce5f2b17b4aa66a30c2827c851cd303adb132fffdd7bc2a97621d7279b9bc6b8"}, - {file = "agentdesk-0.2.134.tar.gz", hash = "sha256:8f710fa57650e1fd9e22531c719feabe9b095a1fadf3d6a4707ef5dcb50b8b71"}, + {file = "agentdesk-0.2.135-py3-none-any.whl", hash = "sha256:4ad33fe8ec59174be527217b291a8eedde7fbcb95f7dcc6f9a50c7eb172fa6bc"}, + {file = "agentdesk-0.2.135.tar.gz", hash = "sha256:cfd60b529d703371fb593d433b85c155cee9ce5721b272aa1494bdb977fb4271"}, ] [package.dependencies] @@ -5894,4 +5894,4 @@ type = ["pytest-mypy"] [metadata] lock-version = "2.0" python-versions = "^3.10" -content-hash = "4e505d7d47ba64625b7a5827234838ea94b26ea6aa9ecdc4266ce85af29ab0b8" +content-hash = "ec2adb03def2c79abe6054d9e46b6d34f5bf346d2b852e1c604f51cbd41824d2" diff --git a/pyproject.toml b/pyproject.toml index 63e894a..5ea86fa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "surfkit" -version = "0.1.421" +version = "0.1.422" description = "A toolkit for building AI agents that use devices" authors = ["Patrick Barker ", "Jeffrey Huckabay "] license = "MIT" @@ -19,7 +19,7 @@ litellm = "^1.35.8" rich = "^13.7.1" tqdm = "^4.66.4" agentcore = "^0.1.2" -agentdesk = "^0.2.134" +agentdesk = "^0.2.135" taskara = "^0.1.244" diff --git a/surfkit/runtime/agent/kube.py b/surfkit/runtime/agent/kube.py index 1445ced..43ed3ed 100644 --- a/surfkit/runtime/agent/kube.py +++ b/surfkit/runtime/agent/kube.py @@ -1227,6 +1227,8 @@ def learn_task_with_job( auth_server: str = AGENTSEA_AUTH_URL, nebu_server: str = NEBU_SERVER, orign_server: str = ORIGN_SERVER, + labels: Optional[Dict[str, str]] = None, + extra_spec: Optional[Dict[str, str]] = None, debug: bool = False, ) -> None: """ @@ -1323,6 +1325,7 @@ def learn_task_with_job( pod_spec = client.V1PodSpec( containers=[container], restart_policy="Never", + **(extra_spec or {}) ) # Incorporate any relevant annotations/labels @@ -1332,7 +1335,7 @@ def learn_task_with_job( "agent_type": agent_type.name, "agent_model": agent_type.to_v1().model_dump_json(), } - labels = {"provisioner": "surfkit"} + labels = {"provisioner": "surfkit", **(labels or {})} # Construct the pod template template = client.V1PodTemplateSpec(