License check code, pre-commit, license header updates across the board

Signed-off-by: Mark Sze <[email protected]>

Author: marklysze

.pre-commit-config.yaml

@@ -64,6 +64,16 @@ repos:
       require_serial: true
       # Print the number of files as a sanity-check
       verbose: true
+    - id: check-license-headers
+      name: check license headers
+      entry: python ./scripts/pre-commit-license-check.py
+      language: python
+      types: [python]
+      exclude: |
+        (?x)^(
+          notebook/.* |
+          website/.*
+        )$
   - repo: https://github.com/nbQA-dev/nbQA
     rev: 1.8.5
     hooks:

autogen/agentchat/contrib/captainagent.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import hashlib
 import json
 import os

autogen/agentchat/contrib/captainagent/tools/data_analysis/calculate_correlation.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def calculate_correlation(csv_path: str, column1: str, column2: str, method: str = "pearson") -> float:
     """
     Calculate the correlation between two columns in a CSV file.

autogen/agentchat/contrib/captainagent/tools/data_analysis/calculate_skewness_and_kurtosis.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def calculate_skewness_and_kurtosis(csv_file: str, column_name: str) -> tuple:
     """
     Calculate the skewness and kurtosis of a specified column in a CSV file. The kurtosis is calculated using the Fisher definition.

autogen/agentchat/contrib/captainagent/tools/data_analysis/detect_outlier_iqr.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def detect_outlier_iqr(csv_file: str, column_name: str):
     """
     Detect outliers in a specified column of a CSV file using the IQR method.

autogen/agentchat/contrib/captainagent/tools/data_analysis/detect_outlier_zscore.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def detect_outlier_zscore(csv_file, column_name, threshold=3):
     """
     Detect outliers in a CSV file based on a specified column. The outliers are determined by calculating the z-score of the data points in the column.

autogen/agentchat/contrib/captainagent/tools/data_analysis/explore_csv.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def explore_csv(file_path, num_lines=5):
     """
     Reads a CSV file and prints the column names, shape, data types, and the first few lines of data.

autogen/agentchat/contrib/captainagent/tools/data_analysis/shapiro_wilk_test.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/information_retrieval/arxiv_download.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import arxiv
 
 from autogen.coding.func_with_reqs import with_requirements

autogen/agentchat/contrib/captainagent/tools/information_retrieval/arxiv_search.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import arxiv
 
 from autogen.coding.func_with_reqs import with_requirements

autogen/agentchat/contrib/captainagent/tools/information_retrieval/extract_pdf_image.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import os
 
 from autogen.coding.func_with_reqs import with_requirements

autogen/agentchat/contrib/captainagent/tools/information_retrieval/extract_pdf_text.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/information_retrieval/get_wikipedia_text.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def get_wikipedia_text(title):
     """
     Retrieves the text content of a Wikipedia page. It does not support tables and other complex formatting.

autogen/agentchat/contrib/captainagent/tools/information_retrieval/get_youtube_caption.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 # alternative api: https://rapidapi.com/omarmhaimdat/api/youtube-v2
 
 

autogen/agentchat/contrib/captainagent/tools/information_retrieval/image_qa.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import os
 
 from PIL import Image

autogen/agentchat/contrib/captainagent/tools/information_retrieval/optical_character_recognition.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 import os
 
 from autogen.coding.func_with_reqs import with_requirements

autogen/agentchat/contrib/captainagent/tools/information_retrieval/perform_web_search.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def perform_web_search(query, count=10, offset=0):
     """
     Perform a web search using Bing API.

autogen/agentchat/contrib/captainagent/tools/information_retrieval/scrape_wikipedia_tables.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def scrape_wikipedia_tables(url: str, header_keyword: str):
     """
     Scrapes Wikipedia tables based on a given URL and header keyword.

autogen/agentchat/contrib/captainagent/tools/information_retrieval/transcribe_audio_file.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/information_retrieval/youtube_download.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def youtube_download(url: str):
     """
     Downloads a YouTube video and returns the download link.

autogen/agentchat/contrib/captainagent/tools/math/calculate_circle_area_from_diameter.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/math/calculate_day_of_the_week.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def calculate_day_of_the_week(total_days: int, starting_day: str):
     """
     Calculates the day of the week after a given number of days starting from a specified day.

autogen/agentchat/contrib/captainagent/tools/math/calculate_fraction_sum.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def calculate_fraction_sum(
     fraction1_numerator: int, fraction1_denominator: int, fraction2_numerator: int, fraction2_denominator: int
 ):

autogen/agentchat/contrib/captainagent/tools/math/calculate_matrix_power.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/math/calculate_reflected_point.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def calculate_reflected_point(point):
     """
     Calculates the reflection point of a given point about the line y=x.

autogen/agentchat/contrib/captainagent/tools/math/complex_numbers_product.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/math/compute_currency_conversion.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 from autogen.coding.func_with_reqs import with_requirements
 
 

autogen/agentchat/contrib/captainagent/tools/math/count_distinct_permutations.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def count_distinct_permutations(sequence):
     """
     Counts the number of distinct permutations of a sequence where items may be indistinguishable.

autogen/agentchat/contrib/captainagent/tools/math/evaluate_expression.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def evaluate_expression(expression):
     """
     Evaluates a mathematical expression with support for floor function notation and power notation.

autogen/agentchat/contrib/captainagent/tools/math/find_continuity_point.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def find_continuity_point(f_leq, f_gt, x_value):
     """
     Find the value 'a' that ensures the continuity of a piecewise function at a given point.

autogen/agentchat/contrib/captainagent/tools/math/fraction_to_mixed_numbers.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def fraction_to_mixed_numbers(numerator, denominator):
     """
     Simplifies a fraction to its lowest terms and returns it as a mixed number.

autogen/agentchat/contrib/captainagent/tools/math/modular_inverse_sum.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def modular_inverse_sum(expressions, modulus):
     """
     Calculates the sum of modular inverses of the given expressions modulo the specified modulus.

autogen/agentchat/contrib/captainagent/tools/math/simplify_mixed_numbers.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def simplify_mixed_numbers(numerator1, denominator1, numerator2, denominator2, whole_number1, whole_number2):
     """
     Simplifies the sum of two mixed numbers and returns the result as a string in the format 'a b/c'.

autogen/agentchat/contrib/captainagent/tools/math/sum_of_digit_factorials.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def sum_of_digit_factorials(number):
     """
     Calculates the sum of the factorial of each digit in a number, often used in problems involving curious numbers like 145.

autogen/agentchat/contrib/captainagent/tools/math/sum_of_primes_below.py

@@ -1,3 +1,6 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
 def sum_of_primes_below(threshold):
     """
     Calculates the sum of all prime numbers below a given threshold.

autogen/agentchat/contrib/graph_rag/document.py

@@ -1,3 +1,9 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
 from dataclasses import dataclass
 from enum import Enum, auto
 from typing import Optional

autogen/agentchat/contrib/graph_rag/falkor_graph_query_engine.py

@@ -1,3 +1,9 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
 import os
 from dataclasses import dataclass, field
 from typing import List

autogen/agentchat/contrib/graph_rag/graph_query_engine.py

@@ -1,3 +1,9 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
 from dataclasses import dataclass, field
 from typing import List, Optional, Protocol
 

autogen/agentchat/contrib/graph_rag/graph_rag_capability.py

@@ -1,3 +1,9 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
 from autogen.agentchat.contrib.capabilities.agent_capability import AgentCapability
 from autogen.agentchat.conversable_agent import ConversableAgent
 

autogen/agentchat/contrib/tool_retriever.py

@@ -1,3 +1,9 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
+# SPDX-License-Identifier: MIT
 import importlib.util
 import inspect
 import os

autogen/coding/base.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from __future__ import annotations
 

autogen/coding/docker_commandline_code_executor.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from __future__ import annotations
 

autogen/coding/factory.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from .base import CodeExecutionConfig, CodeExecutor
 

autogen/coding/func_with_reqs.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from __future__ import annotations
 

autogen/coding/jupyter/base.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from dataclasses import dataclass
 from typing import Optional, Protocol, runtime_checkable

autogen/coding/jupyter/docker_jupyter_server.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from __future__ import annotations
 

autogen/coding/jupyter/embedded_ipython_code_executor.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 import base64
 import json

autogen/coding/jupyter/jupyter_client.py

@@ -1,8 +1,8 @@
-# Copyright (c) 2023 - 2024, Owners of https://github.com/autogen-ai
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
 #
 # SPDX-License-Identifier: Apache-2.0
 #
-# Original portions of this file are derived from https://github.com/microsoft/autogen under the MIT License.
+# Portions derived from https://github.com/microsoft/autogen are under the MIT License.
 # SPDX-License-Identifier: MIT
 from __future__ import annotations