chore: Add framework name into UserAgent header for bedrock integration (run-llama#19615)

Author: 0x-fang

llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py

@@ -147,6 +147,7 @@ def __init__(
                     retries={"max_attempts": max_retries, "mode": "standard"},
                     connect_timeout=timeout,
                     read_timeout=timeout,
+                    user_agent_extra="x-client-framework:llama_index",
                 )
                 if botocore_config is None
                 else botocore_config
@@ -239,6 +240,7 @@ def set_credentials(
 
         try:
             import boto3
+            from botocore.config import Config
 
             session = boto3.Session(**session_kwargs)
         except ImportError:
@@ -247,7 +249,8 @@ def set_credentials(
             )
 
         if "bedrock-runtime" in session.get_available_services():
-            self._client = session.client("bedrock-runtime")
+            config = Config(user_agent_extra="x-client-framework:llama_index")
+            self._client = session.client("bedrock-runtime", config=config)
         else:
             self._client = session.client("bedrock")
 
@@ -311,6 +314,7 @@ def from_credentials(
 
         try:
             import boto3
+            from botocore.config import Config
 
             session = boto3.Session(**session_kwargs)
         except ImportError:
@@ -319,7 +323,8 @@ def from_credentials(
             )
 
         if "bedrock-runtime" in session.get_available_services():
-            client = session.client("bedrock-runtime")
+            config = Config(user_agent_extra="x-client-framework:llama_index")
+            client = session.client("bedrock-runtime", config=config)
         else:
             client = session.client("bedrock")
         return cls(

llama-index-integrations/embeddings/llama-index-embeddings-bedrock/pyproject.toml

@@ -27,7 +27,7 @@ dev = [
 
 [project]
 name = "llama-index-embeddings-bedrock"
-version = "0.6.0"
+version = "0.6.1"
 description = "llama-index embeddings bedrock integration"
 authors = [{name = "Your Name", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"

llama-index-integrations/llms/llama-index-llms-bedrock-converse/llama_index/llms/bedrock_converse/base.py

@@ -252,6 +252,7 @@ def __init__(
                     retries={"max_attempts": max_retries, "mode": "standard"},
                     connect_timeout=timeout,
                     read_timeout=timeout,
+                    user_agent_extra="x-client-framework:llama_index",
                 )
                 if botocore_config is None
                 else botocore_config

llama-index-integrations/llms/llama-index-llms-bedrock-converse/pyproject.toml

@@ -29,7 +29,7 @@ dev = [
 
 [project]
 name = "llama-index-llms-bedrock-converse"
-version = "0.8.0"
+version = "0.8.1"
 description = "llama-index llms bedrock converse integration"
 authors = [{name = "Your Name", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"

llama-index-integrations/llms/llama-index-llms-bedrock/llama_index/llms/bedrock/base.py

@@ -182,6 +182,7 @@ def __init__(
                     retries={"max_attempts": max_retries, "mode": "standard"},
                     connect_timeout=timeout,
                     read_timeout=timeout,
+                    user_agent_extra="x-client-framework:llama_index",
                 )
                 if botocore_config is None
                 else botocore_config

llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml

@@ -26,7 +26,7 @@ dev = [
 
 [project]
 name = "llama-index-llms-bedrock"
-version = "0.4.0"
+version = "0.4.1"
 description = "llama-index llms bedrock integration"
 authors = [{name = "Your Name", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"