diff --git a/sphinx/add_copyright_notice.py b/sphinx/add_copyright_notice.py new file mode 100755 index 0000000..4cf560b --- /dev/null +++ b/sphinx/add_copyright_notice.py @@ -0,0 +1,48 @@ +#!/usr/bin/env python3 +import os + +# Define the copyright notice +COPYRIGHT_NOTICE = """\ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- +""" + + +def add_copyright_notice(file_path): + with open(file_path) as file: + content = file.read() + + if ( + COPYRIGHT_NOTICE.strip() not in content + ): # Avoid adding the notice if it's already present + with open(file_path, "w") as file: + file.write(COPYRIGHT_NOTICE + "\n" + content) + + +def recursively_add_notice_to_py_files(directory): + for root, _, files in os.walk(directory): + for file in files: + if file.endswith(".py"): + file_path = os.path.join(root, file) + add_copyright_notice(file_path) + + +# Specify the directory you want to start the search from +start_directory = "../src" # Replace with your directory path + +recursively_add_notice_to_py_files(start_directory) + +print("Copyright notice added to all .py files.") diff --git a/src/artkit/__init__.py b/src/artkit/__init__.py index b9d5d07..a6fb108 100644 --- a/src/artkit/__init__.py +++ b/src/artkit/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ This section of the documentation provides detailed information about the ARTKIT API, including descriptions of all classes, functions, diff --git a/src/artkit/api.py b/src/artkit/api.py index 8d4160b..b1d44cb 100644 --- a/src/artkit/api.py +++ b/src/artkit/api.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Convenience module to import all classes and functions using single package alias. diff --git a/src/artkit/flow.py b/src/artkit/flow.py index 3b97954..576427a 100644 --- a/src/artkit/flow.py +++ b/src/artkit/flow.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ This module provides a functional API for defining and executing complex and asynchronous workflows. The API is inspired by the functional programming diff --git a/src/artkit/model/__init__.py b/src/artkit/model/__init__.py index 02b82ae..ca034f4 100644 --- a/src/artkit/model/__init__.py +++ b/src/artkit/model/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Gen AI model endpoints. diff --git a/src/artkit/model/base/__init__.py b/src/artkit/model/base/__init__.py index 15e602e..0848f7d 100644 --- a/src/artkit/model/base/__init__.py +++ b/src/artkit/model/base/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base classes for the LLM module. """ diff --git a/src/artkit/model/base/_adapter.py b/src/artkit/model/base/_adapter.py index 3702b1d..34f310f 100644 --- a/src/artkit/model/base/_adapter.py +++ b/src/artkit/model/base/_adapter.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of CachedLLM. """ diff --git a/src/artkit/model/base/_model.py b/src/artkit/model/base/_model.py index 087e125..58873fb 100644 --- a/src/artkit/model/base/_model.py +++ b/src/artkit/model/base/_model.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of llm module. """ diff --git a/src/artkit/model/cache/__init__.py b/src/artkit/model/cache/__init__.py index 92b9b7a..c8d1f7c 100644 --- a/src/artkit/model/cache/__init__.py +++ b/src/artkit/model/cache/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Caching for ARTKIT model connections. diff --git a/src/artkit/model/cache/_cache.py b/src/artkit/model/cache/_cache.py index e2448b3..4fad692 100644 --- a/src/artkit/model/cache/_cache.py +++ b/src/artkit/model/cache/_cache.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Internal DB implementation for the cache. """ diff --git a/src/artkit/model/diffusion/__init__.py b/src/artkit/model/diffusion/__init__.py index 3848f9c..3b2e9dc 100644 --- a/src/artkit/model/diffusion/__init__.py +++ b/src/artkit/model/diffusion/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Diffusion model implementations (text-to-image generation) """ diff --git a/src/artkit/model/diffusion/base/__init__.py b/src/artkit/model/diffusion/base/__init__.py index a3d426a..e6546c3 100644 --- a/src/artkit/model/diffusion/base/__init__.py +++ b/src/artkit/model/diffusion/base/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base classes for the diffusion model """ diff --git a/src/artkit/model/diffusion/base/_adapter.py b/src/artkit/model/diffusion/base/_adapter.py index 201d2dc..aff65c7 100644 --- a/src/artkit/model/diffusion/base/_adapter.py +++ b/src/artkit/model/diffusion/base/_adapter.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the ``CachedDiffusionModel`` class. """ diff --git a/src/artkit/model/diffusion/base/_diffusion.py b/src/artkit/model/diffusion/base/_diffusion.py index 33ee764..76a4c3b 100644 --- a/src/artkit/model/diffusion/base/_diffusion.py +++ b/src/artkit/model/diffusion/base/_diffusion.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base implementation of diffusion model """ diff --git a/src/artkit/model/diffusion/openai/__init__.py b/src/artkit/model/diffusion/openai/__init__.py index 6b35de2..817dcda 100644 --- a/src/artkit/model/diffusion/openai/__init__.py +++ b/src/artkit/model/diffusion/openai/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ OpenAI diffusion systems. """ diff --git a/src/artkit/model/diffusion/openai/_openai.py b/src/artkit/model/diffusion/openai/_openai.py index 78ea5e9..5a3e65c 100644 --- a/src/artkit/model/diffusion/openai/_openai.py +++ b/src/artkit/model/diffusion/openai/_openai.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of OpenAIDiffusion """ diff --git a/src/artkit/model/llm/__init__.py b/src/artkit/model/llm/__init__.py index fbde824..e51c704 100644 --- a/src/artkit/model/llm/__init__.py +++ b/src/artkit/model/llm/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Simple LLM adapter classes. """ diff --git a/src/artkit/model/llm/_cached.py b/src/artkit/model/llm/_cached.py index 0aa26a4..0c23326 100644 --- a/src/artkit/model/llm/_cached.py +++ b/src/artkit/model/llm/_cached.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of CachedLLM. """ diff --git a/src/artkit/model/llm/_gen2chat.py b/src/artkit/model/llm/_gen2chat.py index 47c3893..fd528d7 100644 --- a/src/artkit/model/llm/_gen2chat.py +++ b/src/artkit/model/llm/_gen2chat.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of ``ChatFromCompletionModel``. """ diff --git a/src/artkit/model/llm/_generator.py b/src/artkit/model/llm/_generator.py index 8639230..5088e27 100644 --- a/src/artkit/model/llm/_generator.py +++ b/src/artkit/model/llm/_generator.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the text producer. """ diff --git a/src/artkit/model/llm/_historized.py b/src/artkit/model/llm/_historized.py index 4e88529..80b91f5 100644 --- a/src/artkit/model/llm/_historized.py +++ b/src/artkit/model/llm/_historized.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of ``HistorizedChatModel``. """ diff --git a/src/artkit/model/llm/anthropic/__init__.py b/src/artkit/model/llm/anthropic/__init__.py index 38b7ddc..beda7bc 100644 --- a/src/artkit/model/llm/anthropic/__init__.py +++ b/src/artkit/model/llm/anthropic/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Anthropic LLM systems. """ diff --git a/src/artkit/model/llm/anthropic/_anthropic.py b/src/artkit/model/llm/anthropic/_anthropic.py index e656314..431f517 100644 --- a/src/artkit/model/llm/anthropic/_anthropic.py +++ b/src/artkit/model/llm/anthropic/_anthropic.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the Anthropic llm module. """ diff --git a/src/artkit/model/llm/base/__init__.py b/src/artkit/model/llm/base/__init__.py index d22e9aa..45364da 100644 --- a/src/artkit/model/llm/base/__init__.py +++ b/src/artkit/model/llm/base/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base classes for the LLM module. """ diff --git a/src/artkit/model/llm/base/_adapter.py b/src/artkit/model/llm/base/_adapter.py index 04de374..33df1de 100644 --- a/src/artkit/model/llm/base/_adapter.py +++ b/src/artkit/model/llm/base/_adapter.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of CachedLLM. """ diff --git a/src/artkit/model/llm/base/_llm.py b/src/artkit/model/llm/base/_llm.py index 3700b51..fee5a07 100644 --- a/src/artkit/model/llm/base/_llm.py +++ b/src/artkit/model/llm/base/_llm.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of llm module. """ diff --git a/src/artkit/model/llm/gemini/__init__.py b/src/artkit/model/llm/gemini/__init__.py index 249d396..4d59b6d 100644 --- a/src/artkit/model/llm/gemini/__init__.py +++ b/src/artkit/model/llm/gemini/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Gemini LLM systems. """ diff --git a/src/artkit/model/llm/gemini/_gemini.py b/src/artkit/model/llm/gemini/_gemini.py index 75f76c0..d17af58 100644 --- a/src/artkit/model/llm/gemini/_gemini.py +++ b/src/artkit/model/llm/gemini/_gemini.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the Gemini llm module. """ diff --git a/src/artkit/model/llm/groq/__init__.py b/src/artkit/model/llm/groq/__init__.py index f1aa982..b233e61 100644 --- a/src/artkit/model/llm/groq/__init__.py +++ b/src/artkit/model/llm/groq/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Groq LLM systems. """ diff --git a/src/artkit/model/llm/groq/_groq.py b/src/artkit/model/llm/groq/_groq.py index 2c5527a..655d507 100644 --- a/src/artkit/model/llm/groq/_groq.py +++ b/src/artkit/model/llm/groq/_groq.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the Groq llm module. """ diff --git a/src/artkit/model/llm/history/__init__.py b/src/artkit/model/llm/history/__init__.py index 8ec0744..a1f4e1f 100644 --- a/src/artkit/model/llm/history/__init__.py +++ b/src/artkit/model/llm/history/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Classes related to chat history. """ diff --git a/src/artkit/model/llm/history/_history.py b/src/artkit/model/llm/history/_history.py index f6e12f9..4520689 100644 --- a/src/artkit/model/llm/history/_history.py +++ b/src/artkit/model/llm/history/_history.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of ``MyClass``. """ diff --git a/src/artkit/model/llm/history/_message.py b/src/artkit/model/llm/history/_message.py index b396d46..eb06fb4 100644 --- a/src/artkit/model/llm/history/_message.py +++ b/src/artkit/model/llm/history/_message.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of ``ChatMessage``. """ diff --git a/src/artkit/model/llm/huggingface/__init__.py b/src/artkit/model/llm/huggingface/__init__.py index c46f41d..d4efbeb 100644 --- a/src/artkit/model/llm/huggingface/__init__.py +++ b/src/artkit/model/llm/huggingface/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of Huggingface LLMs. """ diff --git a/src/artkit/model/llm/huggingface/_huggingface.py b/src/artkit/model/llm/huggingface/_huggingface.py index ecbe789..3aa41d9 100644 --- a/src/artkit/model/llm/huggingface/_huggingface.py +++ b/src/artkit/model/llm/huggingface/_huggingface.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Huggingface LLM systems. """ diff --git a/src/artkit/model/llm/huggingface/_huggingface_local.py b/src/artkit/model/llm/huggingface/_huggingface_local.py index 2e70ea4..0e44d46 100644 --- a/src/artkit/model/llm/huggingface/_huggingface_local.py +++ b/src/artkit/model/llm/huggingface/_huggingface_local.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Huggingface LLM systems that run locally. """ diff --git a/src/artkit/model/llm/huggingface/base/__init__.py b/src/artkit/model/llm/huggingface/base/__init__.py index ad5ee43..5745b77 100644 --- a/src/artkit/model/llm/huggingface/base/__init__.py +++ b/src/artkit/model/llm/huggingface/base/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base classes for Huggingface LLMs. """ diff --git a/src/artkit/model/llm/huggingface/base/_base.py b/src/artkit/model/llm/huggingface/base/_base.py index 7fed68c..b49a387 100644 --- a/src/artkit/model/llm/huggingface/base/_base.py +++ b/src/artkit/model/llm/huggingface/base/_base.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Huggingface LLM systems. """ diff --git a/src/artkit/model/llm/multi_turn/__init__.py b/src/artkit/model/llm/multi_turn/__init__.py index fb744a5..0b40d30 100644 --- a/src/artkit/model/llm/multi_turn/__init__.py +++ b/src/artkit/model/llm/multi_turn/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Multi-turn conversation functions. """ diff --git a/src/artkit/model/llm/multi_turn/_multi_turn.py b/src/artkit/model/llm/multi_turn/_multi_turn.py index ac37a26..db8b366 100644 --- a/src/artkit/model/llm/multi_turn/_multi_turn.py +++ b/src/artkit/model/llm/multi_turn/_multi_turn.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the multi-turn conversation attack. """ diff --git a/src/artkit/model/llm/openai/__init__.py b/src/artkit/model/llm/openai/__init__.py index 72a253d..7388d53 100644 --- a/src/artkit/model/llm/openai/__init__.py +++ b/src/artkit/model/llm/openai/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ OpenAI LLM systems. """ diff --git a/src/artkit/model/llm/openai/_openai.py b/src/artkit/model/llm/openai/_openai.py index 1f2ac61..8e23b6f 100644 --- a/src/artkit/model/llm/openai/_openai.py +++ b/src/artkit/model/llm/openai/_openai.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of llm module. """ diff --git a/src/artkit/model/llm/util/__init__.py b/src/artkit/model/llm/util/__init__.py index b5b2c45..4205f89 100644 --- a/src/artkit/model/llm/util/__init__.py +++ b/src/artkit/model/llm/util/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ LLM specific utilities """ diff --git a/src/artkit/model/llm/util/_json.py b/src/artkit/model/llm/util/_json.py index 10862ee..11b589c 100644 --- a/src/artkit/model/llm/util/_json.py +++ b/src/artkit/model/llm/util/_json.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the `coerce_json_format` function, which uses an LLM to coerce a JSON-like string into the specified dictionary format. The function is used to diff --git a/src/artkit/model/util/__init__.py b/src/artkit/model/util/__init__.py index 311d9fb..a72ae8a 100644 --- a/src/artkit/model/util/__init__.py +++ b/src/artkit/model/util/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Utilities making use of the model classes. """ diff --git a/src/artkit/model/util/_exception.py b/src/artkit/model/util/_exception.py index badd6de..f92f99b 100644 --- a/src/artkit/model/util/_exception.py +++ b/src/artkit/model/util/_exception.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ This file contains custom exception classes for the models. """ diff --git a/src/artkit/model/util/_retry.py b/src/artkit/model/util/_retry.py index 97dfcf7..16dfef3 100644 --- a/src/artkit/model/util/_retry.py +++ b/src/artkit/model/util/_retry.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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 asyncio import logging import random diff --git a/src/artkit/model/vision/__init__.py b/src/artkit/model/vision/__init__.py index ab7bc0d..ac07ac1 100644 --- a/src/artkit/model/vision/__init__.py +++ b/src/artkit/model/vision/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Vision model implementations (image-to-text generation) """ diff --git a/src/artkit/model/vision/base/__init__.py b/src/artkit/model/vision/base/__init__.py index c39bdb7..ce9272c 100644 --- a/src/artkit/model/vision/base/__init__.py +++ b/src/artkit/model/vision/base/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base classes for the vision model """ diff --git a/src/artkit/model/vision/base/_adapter.py b/src/artkit/model/vision/base/_adapter.py index 65211e8..b4a921e 100644 --- a/src/artkit/model/vision/base/_adapter.py +++ b/src/artkit/model/vision/base/_adapter.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of CachedVisionModel. """ diff --git a/src/artkit/model/vision/base/_vision.py b/src/artkit/model/vision/base/_vision.py index 5fd51c2..ebe068a 100644 --- a/src/artkit/model/vision/base/_vision.py +++ b/src/artkit/model/vision/base/_vision.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Base implementation of vision model """ diff --git a/src/artkit/model/vision/openai/__init__.py b/src/artkit/model/vision/openai/__init__.py index 01fb9f5..7f83608 100644 --- a/src/artkit/model/vision/openai/__init__.py +++ b/src/artkit/model/vision/openai/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ OpenAI vision model. """ diff --git a/src/artkit/model/vision/openai/_openai.py b/src/artkit/model/vision/openai/_openai.py index fd5bf05..28a35bb 100644 --- a/src/artkit/model/vision/openai/_openai.py +++ b/src/artkit/model/vision/openai/_openai.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of OpenAIVision """ diff --git a/src/artkit/util/__init__.py b/src/artkit/util/__init__.py index 0e6e946..6aad06a 100644 --- a/src/artkit/util/__init__.py +++ b/src/artkit/util/__init__.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Utility functions. """ diff --git a/src/artkit/util/_image.py b/src/artkit/util/_image.py index 4342fd8..bc22622 100644 --- a/src/artkit/util/_image.py +++ b/src/artkit/util/_image.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of the Image class for I/O handling of Vision and Diffusion models. """ diff --git a/src/artkit/util/_log_throttling.py b/src/artkit/util/_log_throttling.py index cd08a0a..8eefb32 100644 --- a/src/artkit/util/_log_throttling.py +++ b/src/artkit/util/_log_throttling.py @@ -1,3 +1,19 @@ +# ----------------------------------------------------------------------------- +# © 2024 Boston Consulting Group. All rights reserved. +# +# 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. +# ----------------------------------------------------------------------------- + """ Implementation of ``LogThrottlingHandler``. """