@@ -28,7 +28,7 @@ classifiers = [
 grpc = ["grpcio>=1.48.2,<2"]
 opentelemetry = ["opentelemetry-api>=1.11.1,<2", "opentelemetry-sdk>=1.11.1,<2"]
 pydantic = ["pydantic>=2.0.0,<3"]
-openai-agents = ["openai-agents>=0.14.0", "mcp>=1.9.4, <2"]
+openai-agents = ["openai-agents>=0.17.1", "mcp>=1.9.4, <2"]
 google-adk = ["google-adk>=1.27.0,<2"]
 langsmith = ["langsmith>=0.7.0,<0.8"]
 lambda-worker-otel = [
@@ -79,7 +79,7 @@ dev = [
   "pytest-rerunfailures>=16.1",
   "pytest-xdist>=3.6,<4",
   "moto[s3,server]>=5",
-  "langsmith>=0.7.0,<0.8",
+  "langsmith>=0.7.0,<0.7.34",
   "setuptools<82",
   "opentelemetry-exporter-otlp-proto-grpc>=1.11.1,<2",
   "opentelemetry-semantic-conventions>=0.40b0,<1",
 
@@ -684,6 +684,7 @@ impl WorkerRef {
     }
 
     fn initiate_shutdown(&self) -> PyResult<()> {
+        enter_sync!(self.runtime);
         let worker = self.worker.as_ref().unwrap().clone();
         worker.initiate_shutdown();
         Ok(())
 
@@ -124,8 +124,8 @@ def on_span_end(self, span: Span[Any]) -> None:
 
         self._impl.on_span_end(span)
 
-    def shutdown(self) -> None:
-        self._impl.shutdown()
+    def shutdown(self, timeout: float | None = None) -> None:
+        self._impl.shutdown(timeout)
 
     def force_flush(self) -> None:
         self._impl.force_flush()
Original file line number	Diff line number	Diff line change
`@@ -684,6 +684,7 @@ impl WorkerRef {`
`684`	`684`	`}`
`685`	`685`
`686`	`686`	`fn initiate_shutdown(&self) -> PyResult<()> {`
	`687`	`+ enter_sync!(self.runtime);`
`687`	`688`	`let worker = self.worker.as_ref().unwrap().clone();`
`688`	`689`	`worker.initiate_shutdown();`
`689`	`690`	`Ok(())`