diff --git a/langchain/llms/amazon_api_gateway.py b/langchain/llms/amazon_api_gateway.py
index a2c0d7ec492..72b269dcba5 100644
--- a/langchain/llms/amazon_api_gateway.py
+++ b/langchain/llms/amazon_api_gateway.py
@@ -52,7 +52,7 @@ class AmazonAPIGateway(LLM):
         """Get the identifying parameters."""
         _model_kwargs = self.model_kwargs or {}
         return {
-            **{"endpoint_name": self.api_url},
+            **{"api_url": self.api_url, "headers": self.headers},
             **{"model_kwargs": _model_kwargs},
         }