add headers

Signed-off-by: Yuchen Zhang <[email protected]>

Author: yczhang-nv

tests/aiq/llm_providers/test_langchain_agents.py

@@ -1,3 +1,17 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import pytest
 from langchain_core.messages import AIMessage
 from langchain_core.prompts import ChatPromptTemplate
@@ -12,22 +26,20 @@
 @pytest.mark.integration
 @pytest.mark.asyncio
 async def test_nim_langchain_agent():
-    """Test NIM LLM with LangChain agent. Requires NVIDIA_API_KEY to be set."""
+    """
+    Test NIM LLM with LangChain agent. Requires NVIDIA_API_KEY to be set.
+    """
 
     prompt = ChatPromptTemplate.from_messages([("system", "You are a helpful AI assistant."), ("human", "{input}")])
 
-
     llm_config = NIMModelConfig(model_name="meta/llama-3.1-70b-instruct", temperature=0.0)
 
-
     async with WorkflowBuilder() as builder:
         await builder.add_llm("nim_llm", llm_config)
         llm = await builder.get_llm("nim_llm", wrapper_type=LLMFrameworkEnum.LANGCHAIN)
 
-
         agent = prompt | llm
 
-
         response = await agent.ainvoke({"input": "What is 1+2?"})
         assert isinstance(response, AIMessage)
         assert response.content is not None
@@ -38,7 +50,9 @@ async def test_nim_langchain_agent():
 @pytest.mark.integration
 @pytest.mark.asyncio
 async def test_openai_langchain_agent():
-    """Test OpenAI LLM with LangChain agent. Requires OPENAI_API_KEY to be set."""
+    """
+    Test OpenAI LLM with LangChain agent. Requires OPENAI_API_KEY to be set.
+    """
     prompt = ChatPromptTemplate.from_messages([("system", "You are a helpful AI assistant."), ("human", "{input}")])
 
     llm_config = OpenAIModelConfig(model_name="gpt-3.5-turbo", temperature=0.0)
@@ -59,7 +73,11 @@ async def test_openai_langchain_agent():
 @pytest.mark.integration
 @pytest.mark.asyncio
 async def test_aws_bedrock_langchain_agent():
-    """Test AWS Bedrock LLM with LangChain agent. Requires AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY to be set."""
+    """
+    Test AWS Bedrock LLM with LangChain agent.
+    Requires AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY to be set.
+    See https://docs.aws.amazon.com/bedrock/latest/userguide/setting-up.html for more information.
+    """
     prompt = ChatPromptTemplate.from_messages([("system", "You are a helpful AI assistant."), ("human", "{input}")])
 
     llm_config = AWSBedrockModelConfig(model_name="meta.llama3-3-70b-instruct-v1:0",

tests/aiq/llm_providers/test_llama_index_agents.py

@@ -1,3 +1,17 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 from typing import Any
 
 import pytest
@@ -74,8 +88,11 @@ async def test_openai_minimal_agent():
 @pytest.mark.integration
 @pytest.mark.asyncio
 async def test_aws_bedrock_minimal_agent():
-    """Test AWS Bedrock LLM with minimal LlamaIndex agent.
-    Requires AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY to be set."""
+    """
+    Test AWS Bedrock LLM with LangChain agent.
+    Requires AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY to be set.
+    See https://docs.aws.amazon.com/bedrock/latest/userguide/setting-up.html for more information.
+    """
     llm_config = AWSBedrockModelConfig(model_name="us.meta.llama3-1-405b-instruct-v1:0",
                                        temperature=0.0,
                                        region_name="us-east-2",