whosellm 0.1.1a1__tar.gz

whosellm-0.1.1a1/.bumpversion.toml

@@ -0,0 +1,58 @@
+# bump-my-version 配置文件 / bump-my-version configuration file
+# 文档 / Documentation: https://callowayproject.github.io/bump-my-version/
+
+[tool.bumpversion]
+current_version = "0.1.1a1"
+parse = """(?x)
+    (?P<major>0|[1-9]\\d*)\\.
+    (?P<minor>0|[1-9]\\d*)\\.
+    (?P<patch>0|[1-9]\\d*)
+    (?:(?P<pre_l>a|b|rc)(?P<pre_n>0|[1-9]\\d*))?   # Pre-release
+    (?:\\.post(?P<post>0|[1-9]\\d*))?              # Post-release
+    (?:\\.dev(?P<dev>0|[1-9]\\d*))?                # Dev release
+"""
+serialize = [
+    "{major}.{minor}.{patch}{pre_l}{pre_n}.post{post}.dev{dev}",
+    "{major}.{minor}.{patch}{pre_l}{pre_n}.post{post}",
+    "{major}.{minor}.{patch}{pre_l}{pre_n}.dev{dev}",
+    "{major}.{minor}.{patch}{pre_l}{pre_n}",
+    "{major}.{minor}.{patch}.post{post}.dev{dev}",
+    "{major}.{minor}.{patch}.post{post}",
+    "{major}.{minor}.{patch}.dev{dev}",
+    "{major}.{minor}.{patch}",
+]
+search = "{current_version}"
+replace = "{new_version}"
+regex = false
+ignore_missing_version = false
+ignore_missing_files = false
+tag = true
+sign_tags = false
+tag_name = "v{new_version}"
+tag_message = "Bump version: {current_version} → {new_version}"
+allow_dirty = false
+commit = true
+message = "Bump version: {current_version} → {new_version}"
+commit_args = ""
+
+[[tool.bumpversion.files]]
+filename = "pyproject.toml"
+search = "version = \"{current_version}\""
+replace = "version = \"{new_version}\""
+
+[[tool.bumpversion.files]]
+filename = "whosellm/__init__.py"
+search = "__version__ = \"{current_version}\""
+replace = "__version__ = \"{new_version}\""
+
+[tool.bumpversion.parts.pre_l]
+values = ["a", "b", "rc"]
+
+[tool.bumpversion.parts.pre_n]
+first_value = 0
+
+[tool.bumpversion.parts.post]
+first_value = 0
+
+[tool.bumpversion.parts.dev]
+first_value = 0

whosellm-0.1.1a1/.coveragerc

@@ -0,0 +1,20 @@
+# Coverage 配置文件 / Coverage configuration file
+
+[run]
+source = llmeta
+omit = 
+    */tests/*
+    */__pycache__/*
+    */site-packages/*
+    .venv/*
+
+[report]
+exclude_lines =
+    pragma: no cover
+    def __repr__
+    raise AssertionError
+    raise NotImplementedError
+    if __name__ == .__main__.:
+    if TYPE_CHECKING:
+    @abstractmethod
+    @abc.abstractmethod

whosellm-0.1.1a1/.github/workflows/publish.yml

@@ -0,0 +1,124 @@
+# 文件名: publish.yml / File name: publish.yml
+# 作者: 自动生成 / Author: Generated by Cascade
+# 创建日期: 2025/11/09 / Created: 2025/11/09
+# 描述: 基于 Release 触发的 whosellm 包构建与发布流程 / Release-triggered publish workflow for whosellm
+
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:
+
+permissions:
+  contents: read
+  id-token: write
+
+concurrency:
+  group: publish-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  build:
+    name: Build distributions
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Setup Python 3.11
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install uv
+        run: |
+          curl -LsSf https://astral.sh/uv/install.sh | sh
+          echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+
+      - name: Validate tag matches project version
+        env:
+          TAG_NAME: ${{ github.event.release.tag_name }}
+        run: |
+          python - <<'PY'
+          import os, sys, tomllib
+          tag = os.environ.get("TAG_NAME", "").lstrip("v")
+          if not tag:
+              print("No release tag provided", file=sys.stderr)
+              sys.exit(1)
+          with open("pyproject.toml", "rb") as f:
+              version = tomllib.load(f)["project"]["version"]
+          if version != tag:
+              print(f"Version mismatch: pyproject {version} != tag {tag}", file=sys.stderr)
+              sys.exit(2)
+          print(f"Version OK: {version}")
+          PY
+
+      - name: Install dependencies (runtime only)
+        run: |
+          uv sync --no-dev
+
+      - name: Build sdist and wheel
+        run: |
+          uv build
+
+      - name: Upload dist artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: whosellm-dist
+          path: dist/*
+
+  publish-testpypi:
+    name: Publish to TestPyPI
+    needs: [build]
+    runs-on: ubuntu-latest
+    if: ${{ github.event.release.prerelease == true }}
+    environment:
+      name: testpypi
+      url: https://test.pypi.org/project/whosellm/
+
+    steps:
+      - name: Download artifacts
+        uses: actions/download-artifact@v4
+        with:
+          name: whosellm-dist
+          path: dist
+
+      - name: List distributions
+        run: ls -lh dist/
+
+      - name: Publish to TestPyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/
+          skip-existing: true
+          verbose: true
+
+  publish-pypi:
+    name: Publish to PyPI
+    needs: [build]
+    runs-on: ubuntu-latest
+    if: ${{ github.event.release.prerelease == false && github.event.release.target_commitish == 'main' }}
+    environment:
+      name: pypi
+      url: https://pypi.org/project/whosellm/
+
+    permissions:
+      id-token: write
+
+    steps:
+      - name: Download artifacts
+        uses: actions/download-artifact@v4
+        with:
+          name: whosellm-dist
+          path: dist
+
+      - name: List distributions
+        run: ls -lh dist/
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          skip-existing: true
+          verbose: true

whosellm-0.1.1a1/.github/workflows/tests.yml

@@ -0,0 +1,95 @@
+# 文件名: tests.yml / File name: tests.yml
+# 作者: 自动生成 / Author: Generated by Cascade
+# 创建日期: 2025/11/09 / Created: 2025/11/09
+# 描述: 针对 whosellm 项目的常规质量检查工作流 / Regular QA workflow for whosellm project
+
+name: Tests
+
+on:
+  push:
+    branches: ["main", "develop"]
+  pull_request:
+    branches: ["main", "develop"]
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+concurrency:
+  group: tests-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  qa:
+    name: Run quality checks (Python ${{ matrix.python-version }})
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.11", "3.12", "3.13", "3.14"]
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Setup Python ${{ matrix.python-version }}
+        id: setup-python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          allow-prereleases: ${{ matrix.python-version == '3.14' }}
+
+      - name: Install system dependencies (ripgrep)
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y ripgrep
+
+      - name: Install uv
+        run: |
+          curl -LsSf https://astral.sh/uv/install.sh | sh
+          echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+
+      - name: Cache uv cache
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/uv
+          key: ${{ runner.os }}-uv-${{ hashFiles('**/uv.lock', '**/pyproject.toml') }}-py${{ matrix.python-version }}-v1
+          restore-keys: |
+            ${{ runner.os }}-uv-
+
+      - name: Install dependencies (dev + test extras)
+        run: |
+          uv --version
+          uv sync --extra dev --extra test
+
+      - name: Check code formatting (ruff)
+        if: matrix.python-version == '3.11'
+        run: |
+          uv run ruff format --check .
+
+      - name: Run linting (ruff)
+        if: matrix.python-version == '3.11'
+        run: |
+          uv run ruff check .
+
+      - name: Run type checking (mypy)
+        if: matrix.python-version == '3.11'
+        run: |
+          uv run mypy whosellm
+
+      - name: Run tests (pytest)
+        run: |
+          uv run pytest tests --tb=short
+
+      # 可选：覆盖率报告示例
+      # - name: Generate coverage report
+      #   if: always()
+      #   run: |
+      #     uv run pytest tests --cov=whosellm --cov-report=xml
+      #
+      # - name: Upload coverage report
+      #   if: always()
+      #   uses: actions/upload-artifact@v4
+      #   with:
+      #     name: coverage-report
+      #     path: coverage.xml

whosellm-0.1.1a1/.gitignore

@@ -0,0 +1,58 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+.venv/
+venv/
+ENV/
+env/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+.hypothesis/
+
+# MyPy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Ruff
+.ruff_cache/
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Project specific
+*.log
+.env
+.env.local

whosellm-0.1.1a1/.windsurf/workflows/addmodel.md

@@ -0,0 +1,78 @@
+---
+description: 添加与修正模型配置
+---
+
+目前的模型配置定义参考：
+
+from dataclasses import dataclass, field
+
+import parse  # type: ignore[import-untyped]
+
+from llmeta.capabilities import ModelCapabilities
+from llmeta.models.base import ModelFamily
+from llmeta.provider import Provider
+
+
+@dataclass
+class SpecificModelConfig:
+    """
+    特定模型配置 / Specific model configuration
+
+    用于预注册特定模型的详细配置
+    Used for pre-registering detailed configuration for specific models
+    """
+
+    # 模型版本 / Model version
+    version: str
+
+    # 模型变体 / Model variant
+    variant: str
+
+    # 自定义能力（可选） / Custom capabilities (optional)
+    capabilities: ModelCapabilities | None = None
+
+    # 子命名模式（可选） / Sub-patterns (optional)
+    # 必须是父 patterns 的子集 / Must be a subset of parent patterns
+    patterns: list[str] = field(default_factory=list)
+
+
+@dataclass
+class ModelFamilyConfig:
+    """
+    模型家族配置 / Model family configuration
+
+    集中管理一个模型家族的所有配置信息
+    Centrally manage all configuration for a model family
+    """
+
+    # 基本信息 / Basic information
+    family: ModelFamily
+    provider: Provider
+
+    # 命名模式 / Naming patterns
+    # 按优先级排序，更具体的在前 / Ordered by priority, more specific first
+    patterns: list[str]
+
+    # 默认值 / Defaults
+    version_default: str = "1.0"
+
+    # 默认能力 / Default capabilities
+    capabilities: ModelCapabilities = field(default_factory=ModelCapabilities)
+
+    # 预注册的特定模型（可选） / Pre-registered specific models (optional)
+    # 格式: {model_name: SpecificModelConfig}
+    # Format: {model_name: SpecificModelConfig}
+    specific_models: dict[str, SpecificModelConfig] = field(default_factory=dict)
+
+---
+
+目前已经有相应的添加模式在 llmeta/models/families
+
+但目前的配置并不是非常正常，目前的配置数据很多是Mock数据。我会复制给你官网相应的真实配置文档，你根据这个配置文档，来修改当前的Mock数据或者添加新的配置数据。
+
+我给出的文档中会尽可能多出现一些示例模型名称，你需要总结其模式。另外，模型的能力我会给出描述，你需要配置Capabilities。
+
+1. 我每次会提供一个（有时候可能有多个）从官网复制的模型说明，你需要根据提供内容来判断在当前哪个Provider提供丰富对应的配置，如果我提供的模型已经在配置中存在，则检查其capabilities是否正确。需要注意，你可以新建SpecialModels，或者新建模型家族，模型家族的分类有时候不一定非得按品牌来分，比如GPT模型众多，每个版本都形成一个独立家族，而有些品牌模型量比较少，可以一个品牌一个家族，不同版本区分。
+2. 完成配置后你需要自检，Patterns不要有冲突，因为匹配模式的时候有顺序，优先SpecialModels，然后按主配置来，SpecialsModels又按顺序尝试，因此要注意避免提前被匹配的情况。
+3. 测试用例需要在 tests/ 目录下，映射源代码目录结构。可以参考 @tests/models/families/test_gpt5.py。注意的时候注意写一些空间被错误捕获的Case来验证，因为目前的经验来看，非常容易因为Parsed模板提前捕获导致错误匹配。
+4. 每次添加或者修改配置后，会有些之前的测试用例无法通过，因为毕竟修改了配置。你需要在每次修改完配置后运行 uv run poe test。查看无法通过的用例，如果是因为测试数据过时的原因，则修复测试数据，如果是因为配置逻辑问题（比如Patterns顺序，或者SpcialModels顺序造成的），则修复配置

whosellm-0.1.1a1/.windsurf/workflows/arch.md

@@ -0,0 +1,50 @@
+---
+description: 当前项目的架构设计说明
+---
+
+背景：
+
+当前大模型的厂商（provider）/版本(3.5/4.0/4)/型号(flash/mini/o)混乱不堪
+不同厂商，不同版本，不同型号其能力又不尽相同
+
+比如以智谱AI的GLM-4V来讲，它支持视觉多模态，但其接口中传入的视频有如下要求：
+GLM-4V-Plus视频大小限制为20M以内，视频时长不超过 30s。
+GLM-4V-Plus-0111视频大小限制为 200M 以内。
+图片有如下要求：
+图片url或者base64编码。
+图像大小上传限制为每张图像 5M以下，且像素不超过 6000*6000。
+支持jpg、png、jpeg格式。
+说明： GLM-4V-Flash 不支持base64编码
+
+这些要求非常细碎，开发者将非常难以处理这些“变态”的要求。一个不小心就会触发警告，同时如果开发者有适配多个模型的需求，这些不同厂商API之间的要求将是恶梦。
+
+当前的项目目标是完成以下要求：
+
+1. 仅需要一个字符串即可完成这个模型配置的初始化。
+2. 初始化完成后，可以自动匹配到厂商/版本/型号，同时可以基于同一厂商下的不同版本/型号进行大小比较，支持运算符比较。
+3. 初始化完成后，提供其能力范围说明，初期能力范围主要着重「是否支持thinking（reasoning）模式, 是否支持图片，是否支持音频，是否支持视频，是否支持PDF」
+4. 针对具体的模型，提供一个对请求参数进行validate的可选实现，用于解决像GLM-4V-Plus这样的变态参数要求自动整改。这个实现会基于我们vrl-python库，使用VRL脚本语言实现。（使用 uv add vrl-python即可添加）
+
+---
+
+当前项目使用uv管理依赖
+使用ruff进行格式检查
+使用mypy进行类型检查
+使用bump-my-version管理版本
+
+测试的时候注意使用 uv run pytest 激活虚拟环境进行测试
+
+---
+
+已经实现的功能：
+
+1. 相同provider提供的相同版本的模型，会有型号的不同，比如GPT系列有 base turbo omni等，一般来讲，我们需要添加 gpt-4-turbo < gpt-4 < gpt-4o的判断能力。也就是不同的型号也有高低之分，因为这一般意味着每百万Token的价格区别。因此元数据需要有一个类似的字段，而且这个字段需要参数大小比较，同时每个模型家族这个字段的大小比较方法需要能自定义，建议使用tuple从左向右越来越大
+2. 当前只有Provider字段的时候我发现一个问题，同一个模型，比如Deepseek，可能是原厂提供，也可能由Tencent提供，不同的Provider其能力可能不一样。因此我们需要添加一个「模型家族」的字段和Provider区分开，同一个模型家族，可能有多个Provider，同时其Capibility可能不同。某个模型家族有默认的Provider，但同时，可以在模型名称前通过 Provider::ModelName 的形式来指定Provider。
+3. 模型家族，模型版本，模型型号一般不会轻易改变，但需要注意，多数模型会有一个日期，代表其小更新的发布，日期也需要参数比较。因此添加日期相关能力，参与比较时在版本，型号相同的情况下比较日期
+
+需要继续实现的功能：
+
+4. 因为不同公司的模型多如牛毛，我们无法一一添加，我们需要提供一个auto_register的方式，实现以下能力：
+    a. 用户初始化一个ModelVersion如果使用一个不存在的名称时，触发自动注册。
+    b. 仅仅需要提供一个模型名（可以选择带上Provider，不带则使用默认）。然后根据其归属的模型家族来初始化能力
+    c.  如果当前家族无任何可用的参考，用户又未指定能力，则抛出异常，告知用户添加失败，无法自动补充能力，需要用户手动注册添加。

whosellm-0.1.1a1/.windsurf/workflows/testllmeta.md

@@ -0,0 +1,11 @@
+---
+description: 测试LLMeta
+---
+
+我会从网上找到各模型官方网站的信息，包括名称及其能力
+
+你需要在  @tests/test_llmeta.py 中写真实的测试用例，你根据我提供的信息（截图）来写断言。然后直接使用 LLMeta("xxx")初始化模型，断言能力是匹配的。
+
+断言能力尽可能全面。
+
+如果测试不通过，你需要分析是配置内容问题，还是配置执行问题（比如pattern执行顺序，优先级判断，或者parsed匹配异常等），根据具体问题进行修复。

whosellm-0.1.1a1/CHANGELOG.md

@@ -0,0 +1,44 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+## [Unreleased]
+
+### Added
+- **模型家族（ModelFamily）概念** - 区分模型家族和提供商，同一模型家族可能由多个Provider提供
+- **型号优先级比较系统** - 支持同一家族下不同型号的智能比较
+  - GPT-4 系列: `gpt-4o-mini < gpt-4 < gpt-4-turbo < gpt-4o`
+  - GLM-4 系列: `glm-4-flash < glm-4 < glm-4-plus`
+  - GLM-4V 系列: `glm-4v-flash < glm-4v < glm-4v-plus < glm-4v-plus-0111`
+  - O1 系列: `o1-mini < o1-preview < o1`
+- **Provider指定语法** - 支持 `Provider::ModelName` 语法来显式指定Provider
+- **variant_priority字段** - 在 `ModelInfo` 中添加型号优先级元组，用于同版本不同型号的比较
+- **发布日期支持** - 支持从模型名称中自动解析发布日期
+  - 支持 `YYYY-MM-DD` 格式（如 `gpt-4-turbo-2024-04-09`）
+  - 支持 `MMDD` 格式（如 `gpt-4-0125-preview`）
+  - 无日期的模型被认为是最新版本（指向 latest）
+- **日期比较功能** - 在版本和型号相同时，通过日期进行精确比较
+- **高级示例文件** - 添加 `examples/advanced_usage.py` 展示新功能的使用
+
+### Changed
+- **比较逻辑优化** - 模型比较从基于Provider改为基于ModelFamily，更符合实际使用场景
+- **版本比较增强** - 比较顺序：版本 > 型号优先级 > 发布日期
+- **错误提示改进** - 不同家族模型比较时的错误提示更加清晰
+- **测试框架兼容** - 测试类现在同时兼容 pytest 和 unittest，方便 PyCharm 运行
+
+### Updated
+- 所有现有模型注册都添加了 `family` 和 `variant_priority` 字段
+- OpenAI 模型定义（gpt-4, gpt-4-turbo, gpt-4o, gpt-4o-mini, gpt-3.5-turbo, o1, o1-mini, o1-preview）
+  - 新增带日期的模型：`gpt-4-turbo-2024-04-09`, `gpt-4-0125-preview`
+- 智谱 AI 模型定义（glm-4, glm-4-plus, glm-4-flash, glm-4v, glm-4v-plus, glm-4v-plus-0111, glm-4v-flash, glm-3-turbo）
+- 测试用例更新以验证新功能（新增3个日期相关测试）
+- README 文档更新，添加新功能说明和使用示例
+
+## [0.1.0] - 2025-11-07
+
+### Added
+- 初始版本发布
+- 基础模型版本管理功能
+- Provider 自动检测
+- 模型能力描述系统
+- 基础版本比较功能

whosellm-0.1.1a1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 JQQ
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.