├── .github ├── assets │ └── badges │ │ └── .gitkeep ├── dependabot.yml ├── pr-labeler.yml ├── release-drafter.yml └── workflows │ ├── draft.yml │ ├── pr_labeler.yml │ ├── pre_commit_auto_update.yml │ └── pypi-deploy.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── pyproject.toml ├── requirements-dev.txt ├── skllm ├── __init__.py ├── classification.py ├── config.py ├── datasets │ ├── __init__.py │ ├── multi_class.py │ ├── multi_label.py │ ├── summarization.py │ └── translation.py ├── llm │ ├── anthropic │ │ ├── completion.py │ │ ├── credentials.py │ │ └── mixin.py │ ├── base.py │ ├── gpt │ │ ├── clients │ │ │ ├── llama_cpp │ │ │ │ ├── completion.py │ │ │ │ └── handler.py │ │ │ └── openai │ │ │ │ ├── completion.py │ │ │ │ ├── credentials.py │ │ │ │ ├── embedding.py │ │ │ │ └── tuning.py │ │ ├── completion.py │ │ ├── embedding.py │ │ ├── mixin.py │ │ └── utils.py │ └── vertex │ │ ├── completion.py │ │ ├── mixin.py │ │ └── tuning.py ├── memory │ ├── __init__.py │ ├── _annoy.py │ ├── _sklearn_nn.py │ └── base.py ├── model_constants.py ├── models │ ├── _base │ │ ├── classifier.py │ │ ├── tagger.py │ │ ├── text2text.py │ │ └── vectorizer.py │ ├── anthropic │ │ ├── classification │ │ │ ├── few_shot.py │ │ │ └── zero_shot.py │ │ ├── tagging │ │ │ └── ner.py │ │ └── text2text │ │ │ ├── __init__.py │ │ │ ├── summarization.py │ │ │ └── translation.py │ ├── gpt │ │ ├── classification │ │ │ ├── few_shot.py │ │ │ ├── tunable.py │ │ │ └── zero_shot.py │ │ ├── tagging │ │ │ └── ner.py │ │ ├── text2text │ │ │ ├── __init__.py │ │ │ ├── summarization.py │ │ │ ├── translation.py │ │ │ └── tunable.py │ │ └── vectorization.py │ └── vertex │ │ ├── classification │ │ ├── tunable.py │ │ └── zero_shot.py │ │ └── text2text │ │ ├── __init__.py │ │ └── tunable.py ├── prompts │ ├── builders.py │ └── templates.py ├── text2text.py ├── utils │ ├── __init__.py │ ├── rendering.py │ └── xml.py └── vectorization.py └── tests ├── llm ├── __init__.py ├── anthropic │ ├── __init__.py │ └── test_anthropic_mixins.py ├── gpt │ ├── __init__.py │ └── test_gpt_mixins.py └── vertex │ ├── __init__.py │ └── test_vertex_mixins.py └── test_utils.py /.github/assets/badges/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/pr-labeler.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/pr-labeler.yml -------------------------------------------------------------------------------- /.github/release-drafter.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/release-drafter.yml -------------------------------------------------------------------------------- /.github/workflows/draft.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/workflows/draft.yml -------------------------------------------------------------------------------- /.github/workflows/pr_labeler.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/workflows/pr_labeler.yml -------------------------------------------------------------------------------- /.github/workflows/pre_commit_auto_update.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/workflows/pre_commit_auto_update.yml -------------------------------------------------------------------------------- /.github/workflows/pypi-deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.github/workflows/pypi-deploy.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /skllm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/__init__.py -------------------------------------------------------------------------------- /skllm/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/classification.py -------------------------------------------------------------------------------- /skllm/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/config.py -------------------------------------------------------------------------------- /skllm/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/datasets/__init__.py -------------------------------------------------------------------------------- /skllm/datasets/multi_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/datasets/multi_class.py -------------------------------------------------------------------------------- /skllm/datasets/multi_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/datasets/multi_label.py -------------------------------------------------------------------------------- /skllm/datasets/summarization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/datasets/summarization.py -------------------------------------------------------------------------------- /skllm/datasets/translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/datasets/translation.py -------------------------------------------------------------------------------- /skllm/llm/anthropic/completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/anthropic/completion.py -------------------------------------------------------------------------------- /skllm/llm/anthropic/credentials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/anthropic/credentials.py -------------------------------------------------------------------------------- /skllm/llm/anthropic/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/anthropic/mixin.py -------------------------------------------------------------------------------- /skllm/llm/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/base.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/llama_cpp/completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/llama_cpp/completion.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/llama_cpp/handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/llama_cpp/handler.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/openai/completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/openai/completion.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/openai/credentials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/openai/credentials.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/openai/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/openai/embedding.py -------------------------------------------------------------------------------- /skllm/llm/gpt/clients/openai/tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/clients/openai/tuning.py -------------------------------------------------------------------------------- /skllm/llm/gpt/completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/completion.py -------------------------------------------------------------------------------- /skllm/llm/gpt/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/embedding.py -------------------------------------------------------------------------------- /skllm/llm/gpt/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/mixin.py -------------------------------------------------------------------------------- /skllm/llm/gpt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/gpt/utils.py -------------------------------------------------------------------------------- /skllm/llm/vertex/completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/vertex/completion.py -------------------------------------------------------------------------------- /skllm/llm/vertex/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/vertex/mixin.py -------------------------------------------------------------------------------- /skllm/llm/vertex/tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/llm/vertex/tuning.py -------------------------------------------------------------------------------- /skllm/memory/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/memory/__init__.py -------------------------------------------------------------------------------- /skllm/memory/_annoy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/memory/_annoy.py -------------------------------------------------------------------------------- /skllm/memory/_sklearn_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/memory/_sklearn_nn.py -------------------------------------------------------------------------------- /skllm/memory/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/memory/base.py -------------------------------------------------------------------------------- /skllm/model_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/model_constants.py -------------------------------------------------------------------------------- /skllm/models/_base/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/_base/classifier.py -------------------------------------------------------------------------------- /skllm/models/_base/tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/_base/tagger.py -------------------------------------------------------------------------------- /skllm/models/_base/text2text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/_base/text2text.py -------------------------------------------------------------------------------- /skllm/models/_base/vectorizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/_base/vectorizer.py -------------------------------------------------------------------------------- /skllm/models/anthropic/classification/few_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/anthropic/classification/few_shot.py -------------------------------------------------------------------------------- /skllm/models/anthropic/classification/zero_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/anthropic/classification/zero_shot.py -------------------------------------------------------------------------------- /skllm/models/anthropic/tagging/ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/anthropic/tagging/ner.py -------------------------------------------------------------------------------- /skllm/models/anthropic/text2text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /skllm/models/anthropic/text2text/summarization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/anthropic/text2text/summarization.py -------------------------------------------------------------------------------- /skllm/models/anthropic/text2text/translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/anthropic/text2text/translation.py -------------------------------------------------------------------------------- /skllm/models/gpt/classification/few_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/classification/few_shot.py -------------------------------------------------------------------------------- /skllm/models/gpt/classification/tunable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/classification/tunable.py -------------------------------------------------------------------------------- /skllm/models/gpt/classification/zero_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/classification/zero_shot.py -------------------------------------------------------------------------------- /skllm/models/gpt/tagging/ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/tagging/ner.py -------------------------------------------------------------------------------- /skllm/models/gpt/text2text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /skllm/models/gpt/text2text/summarization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/text2text/summarization.py -------------------------------------------------------------------------------- /skllm/models/gpt/text2text/translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/text2text/translation.py -------------------------------------------------------------------------------- /skllm/models/gpt/text2text/tunable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/text2text/tunable.py -------------------------------------------------------------------------------- /skllm/models/gpt/vectorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/gpt/vectorization.py -------------------------------------------------------------------------------- /skllm/models/vertex/classification/tunable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/vertex/classification/tunable.py -------------------------------------------------------------------------------- /skllm/models/vertex/classification/zero_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/vertex/classification/zero_shot.py -------------------------------------------------------------------------------- /skllm/models/vertex/text2text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /skllm/models/vertex/text2text/tunable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/models/vertex/text2text/tunable.py -------------------------------------------------------------------------------- /skllm/prompts/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/prompts/builders.py -------------------------------------------------------------------------------- /skllm/prompts/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/prompts/templates.py -------------------------------------------------------------------------------- /skllm/text2text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/text2text.py -------------------------------------------------------------------------------- /skllm/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/utils/__init__.py -------------------------------------------------------------------------------- /skllm/utils/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/utils/rendering.py -------------------------------------------------------------------------------- /skllm/utils/xml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/utils/xml.py -------------------------------------------------------------------------------- /skllm/vectorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/skllm/vectorization.py -------------------------------------------------------------------------------- /tests/llm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/llm/anthropic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/llm/anthropic/test_anthropic_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/tests/llm/anthropic/test_anthropic_mixins.py -------------------------------------------------------------------------------- /tests/llm/gpt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/llm/gpt/test_gpt_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/tests/llm/gpt/test_gpt_mixins.py -------------------------------------------------------------------------------- /tests/llm/vertex/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/llm/vertex/test_vertex_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/tests/llm/vertex/test_vertex_mixins.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BeastByteAI/scikit-llm/HEAD/tests/test_utils.py --------------------------------------------------------------------------------