LLM-Bridge 1.13.0__tar.gz → 1.14.0__tar.gz

llm_bridge-1.14.0/.gitattributes

@@ -0,0 +1,2 @@
+# Auto detect text files and perform LF normalization
+* text=auto

llm_bridge-1.14.0/.github/workflows/python-publish.yml

@@ -0,0 +1,32 @@
+name: "Publish"
+
+on:
+  push:
+    tags:
+      # Publish on any tag starting with a `v`, e.g., v0.1.0
+      - v*
+
+jobs:
+  run:
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
+      - name: Install Python 3.12
+        run: uv python install 3.12
+      - name: Build
+        run: uv build
+      # Check that basic features work and we didn't miss to include crucial files
+#      - name: Smoke test (wheel)
+#        run: uv run --isolated --no-project --with dist/*.whl tests/smoke_test.py
+#      - name: Smoke test (source distribution)
+#        run: uv run --isolated --no-project --with dist/*.tar.gz tests/smoke_test.py
+      - name: Publish
+        run: uv publish

llm_bridge-1.14.0/.gitignore

@@ -0,0 +1,160 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+.idea/

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/PKG-INFO

@@ -1,32 +1,26 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.13.0
+Version: 1.14.0
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
-Keywords: llm,ai
+License-File: LICENSE
+Keywords: ai,llm
 Classifier: Framework :: FastAPI
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.12
-Description-Content-Type: text/markdown
-License-File: LICENSE
+Requires-Dist: anthropic==0.75.0
+Requires-Dist: docxlatex>=1.1.1
 Requires-Dist: fastapi
+Requires-Dist: google-genai==1.46.0
 Requires-Dist: httpx
-Requires-Dist: tenacity
 Requires-Dist: openai==2.9.0
-Requires-Dist: tiktoken==0.11.0
-Requires-Dist: google-genai==1.46.0
-Requires-Dist: anthropic==0.75.0
-Requires-Dist: PyMuPDF
-Requires-Dist: docxlatex>=1.1.1
 Requires-Dist: openpyxl
+Requires-Dist: pymupdf
 Requires-Dist: python-pptx
-Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Requires-Dist: pytest-asyncio; extra == "test"
-Requires-Dist: python-dotenv; extra == "test"
-Requires-Dist: protobuf; extra == "test"
-Dynamic: license-file
+Requires-Dist: tenacity
+Requires-Dist: tiktoken==0.11.0
+Description-Content-Type: text/markdown
 
 # LLM Bridge
 
@@ -71,18 +65,36 @@ The features listed represent the maximum capabilities of each API type supporte
 pip install --upgrade llm_bridge
 ```
 
-## Test
+## Development
+
+### Python uv
+
+1. Install uv: `powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex"`
+2. Install Python in uv: `uv python install 3.12`; upgrade Python in uv: `uv python install 3.12`
+3. Configure requirements:
+  ```bash
+  uv sync
+  ```
+
+### Pycharm Professional
+
+1. Add New Interpreter >> Add Local Interpreter
+  - Environment: Select existing
+  - Type: uv
+2. Add New Configuration >> uv run >> script: `./usage/main.py`
+
+### Usage
+
+Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
+
+### Test
 
 ```bash
-pytest
+uv run pytest
 ```
 
-## Quick Start
+### Build
 
-### Setup
-
-1. Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
-2. Install requirements: `pip install -r requirements.txt`
-3. In PyCharm, add a new Python configuration:
-   - script: `./usage/main.py`
-   - Paths to ".env" files: `./usage/.env`
+```bash
+uv build
+```

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/README.md

@@ -41,18 +41,36 @@ The features listed represent the maximum capabilities of each API type supporte
 pip install --upgrade llm_bridge
 ```
 
-## Test
+## Development
+
+### Python uv
+
+1. Install uv: `powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex"`
+2. Install Python in uv: `uv python install 3.12`; upgrade Python in uv: `uv python install 3.12`
+3. Configure requirements:
+  ```bash
+  uv sync
+  ```
+
+### Pycharm Professional
+
+1. Add New Interpreter >> Add Local Interpreter
+  - Environment: Select existing
+  - Type: uv
+2. Add New Configuration >> uv run >> script: `./usage/main.py`
+
+### Usage
+
+Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
+
+### Test
 
 ```bash
-pytest
+uv run pytest
 ```
 
-## Quick Start
-
-### Setup
+### Build
 
-1. Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
-2. Install requirements: `pip install -r requirements.txt`
-3. In PyCharm, add a new Python configuration:
-   - script: `./usage/main.py`
-   - Paths to ".env" files: `./usage/.env`
+```bash
+uv build
+```

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/llm_bridge/logic/chat_generate/model_client_factory/gemini_client_factory.py

@@ -61,7 +61,6 @@ async def create_gemini_client(
     config = types.GenerateContentConfig(
         system_instruction=system_instruction,
         temperature=temperature,
-        media_resolution=MediaResolution.MEDIA_RESOLUTION_HIGH,
         safety_settings=[
             types.SafetySetting(
                 category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
@@ -89,6 +88,9 @@ async def create_gemini_client(
         response_modalities=response_modalities,
     )
 
+    if vertexai:
+        config.media_resolution=MediaResolution.MEDIA_RESOLUTION_HIGH
+
     if structured_output_schema:
         config.response_mime_type = "application/json"
         config.response_json_schema = structured_output_schema

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/llm_bridge/logic/chat_generate/model_client_factory/openai_client_factory.py

@@ -65,7 +65,7 @@ async def create_openai_client(
     tools = []
     reasoning = None
 
-    if model not in ["gpt-5-chat-latest", "gpt-5-pro"]:
+    if model not in ["gpt-5-pro", "gpt-5.2-pro"]:
         if code_execution:
             tools.append(
                 CodeInterpreter(
@@ -73,16 +73,15 @@ async def create_openai_client(
                     container=CodeInterpreterContainerCodeInterpreterToolAuto(type="auto")
                 )
             )
-    if model not in ["gpt-5-chat-latest"]:
-        tools.append(
-            WebSearchToolParam(
-                type="web_search",
-                search_context_size="high",
-            )
+    tools.append(
+        WebSearchToolParam(
+            type="web_search",
+            search_context_size="high",
         )
-    if re.match(r"gpt-5.*", model) and model != "gpt-5-chat-latest":
+    )
+    if re.match(r"gpt-5.*", model):
         temperature = 1
-    if re.match(r"gpt-5.*", model) and model != "gpt-5-chat-latest":
+    if re.match(r"gpt-5.*", model):
         if thought:
             reasoning = Reasoning(
                 effort="high",

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/llm_bridge/resources/model_prices.json

@@ -71,6 +71,12 @@
     "input": 2.5,
     "output": 15
   },
+  {
+    "apiType": "OpenAI",
+    "model": "gpt-5.2",
+    "input": 1.75,
+    "output": 14
+  },
   {
     "apiType": "OpenAI",
     "model": "gpt-5.1",
@@ -89,6 +95,12 @@
     "input": 0.25,
     "output": 2
   },
+  {
+    "apiType": "OpenAI",
+    "model": "gpt-5.2-pro",
+    "input": 21,
+    "output": 168
+  },
   {
     "apiType": "OpenAI",
     "model": "gpt-5-pro",

{llm_bridge-1.13.0 → llm_bridge-1.14.0}/pyproject.toml

@@ -1,47 +1,42 @@
 [build-system]
-requires = ["setuptools"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
 [project]
 name = "LLM-Bridge"
-version = "1.13.0"
-authors = [
-    {name = "windsnow1025", email = "windsnow1025@gmail.com"}
-]
-description = "A Bridge for LLMs"
-readme = "README.md"
-requires-python = ">=3.12"
-keywords = ["llm", "ai"]
-license = "MIT"
-classifiers = [
-    "Framework :: FastAPI",
-    "Programming Language :: Python :: 3",
-]
+version = "1.14.0"
 dependencies = [
     "fastapi",
     "httpx",
     "tenacity",
     "openai==2.9.0",
     "tiktoken==0.11.0",
-    "google-genai==1.46.0",
+    "google-genai==1.46.0", # google.genai
     "anthropic==0.75.0",
-    "PyMuPDF",
+    "PyMuPDF", # fitz
     "docxlatex>=1.1.1",
     "openpyxl",
-    "python-pptx",
+    "python-pptx", # pptx
+]
+requires-python = ">=3.12"
+authors = [
+    {name = "windsnow1025", email = "windsnow1025@gmail.com"}
+]
+description = "A Bridge for LLMs"
+readme = "README.md"
+license = "MIT"
+keywords = ["llm", "ai"]
+classifiers = [
+    "Framework :: FastAPI",
+    "Programming Language :: Python :: 3",
 ]
 
-[project.optional-dependencies]
-test = [
+[dependency-groups]
+dev = [
     "pytest",
     "pytest-asyncio",
-    "python-dotenv",
-    "protobuf"
+    "python-dotenv", #dotenv
 ]
 
-[tool.setuptools.packages.find]
-where = ["."]
-include = ["llm_bridge*"]
-
 [tool.pytest.ini_options]
-asyncio_default_fixture_loop_scope = "function"
+asyncio_default_fixture_loop_scope = "function"

llm_bridge-1.14.0/tests/chat_client_factory_test.py

@@ -0,0 +1,20 @@
+import pytest
+
+from llm_bridge.type.message import Message, Role, Content, ContentType
+
+
+@pytest.fixture
+def sample_messages():
+    return [
+        Message(role=Role.System, contents=[
+            Content(type=ContentType.Text, data="You are a helpful assistant.")
+        ]),
+        Message(role=Role.User, contents=[
+            Content(type=ContentType.Text, data="Hello")
+        ])
+    ]
+
+
+@pytest.mark.asyncio
+async def test_placeholder():
+    assert True

llm_bridge-1.14.0/tests/message_preprocessor_test.py

@@ -0,0 +1,26 @@
+import pytest
+
+from llm_bridge.logic.message_preprocess.message_preprocessor import extract_system_messages
+from llm_bridge.type.message import Message, Role, Content, ContentType
+
+
+@pytest.fixture
+def sample_messages():
+    return [
+        Message(role=Role.System, contents=[
+            Content(type=ContentType.Text, data="You are a helpful assistant.")
+        ]),
+        Message(role=Role.User, contents=[
+            Content(type=ContentType.Text, data="Hello")
+        ])
+    ]
+
+def test_extract_system_messages(sample_messages):
+    extracted_text = extract_system_messages(sample_messages)
+
+    assert extracted_text == "You are a helpful assistant.\n"
+
+    assert len(sample_messages) == 1
+    assert sample_messages[0].role == Role.User
+    assert sample_messages[0].contents[0].type == ContentType.Text
+    assert sample_messages[0].contents[0].data == "Hello"

llm_bridge-1.14.0/usage/.env.example

@@ -0,0 +1,9 @@
+OPENAI_API_KEY=
+GEMINI_FREE_API_KEY=
+GEMINI_PAID_API_KEY=
+GEMINI_VERTEX_API_KEY=
+ANTHROPIC_API_KEY=
+AZURE_API_KEY=
+AZURE_API_BASE=
+GITHUB_API_KEY=
+XAI_API_KEY=