LLM-Bridge 1.15.6__tar.gz → 1.15.8__tar.gz

llm_bridge-1.15.8/.run/pytest.run.xml

@@ -0,0 +1,18 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="pytest" type="UvRunConfigurationType" factoryName="UvRunConfigurationType">
+    <option name="args">
+      <list />
+    </option>
+    <option name="checkSync" value="true" />
+    <option name="env">
+      <map />
+    </option>
+    <option name="runType" value="MODULE" />
+    <option name="scriptOrModule" value="pytest" />
+    <option name="uvArgs">
+      <list />
+    </option>
+    <option name="uvSdkKey" value="uv (LLM-Bridge)" />
+    <method v="2" />
+  </configuration>
+</component>

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.15.6
+Version: 1.15.8
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
@@ -24,7 +24,7 @@ Description-Content-Type: text/markdown
 
 # LLM Bridge
 
-LLM Bridge is a unified Python interface for interacting with LLMs, including OpenAI (Native / Azure / GitHub), Gemini (AI Studio / Vertex), Claude, and Grok.
+LLM Bridge is a unified API wrapper for native interactions with various LLM providers.
 
 GitHub: [https://github.com/windsnow1025/LLM-Bridge](https://github.com/windsnow1025/LLM-Bridge)
 
@@ -81,12 +81,6 @@ The features listed represent the maximum capabilities of each API type supporte
 
 Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
 
-### Test
-
-```bash
-uv run pytest
-```
-
 ### Build
 
 ```bash

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/README.md

@@ -1,6 +1,6 @@
 # LLM Bridge
 
-LLM Bridge is a unified Python interface for interacting with LLMs, including OpenAI (Native / Azure / GitHub), Gemini (AI Studio / Vertex), Claude, and Grok.
+LLM Bridge is a unified API wrapper for native interactions with various LLM providers.
 
 GitHub: [https://github.com/windsnow1025/LLM-Bridge](https://github.com/windsnow1025/LLM-Bridge)
 
@@ -57,12 +57,6 @@ The features listed represent the maximum capabilities of each API type supporte
 
 Copy `./usage/.env.example` and rename it to `./usage/.env`, then fill in the environment variables.
 
-### Test
-
-```bash
-uv run pytest
-```
-
 ### Build
 
 ```bash

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/llm_bridge/logic/chat_generate/model_message_converter/openai_responses_message_converter.py

@@ -3,18 +3,19 @@ from openai.types.responses import ResponseInputTextParam, ResponseInputImagePar
 
 from llm_bridge.logic.chat_generate import media_processor
 from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_filename_without_timestamp
-from llm_bridge.type.message import Message, ContentType
+from llm_bridge.type.message import Message, ContentType, Role
 from llm_bridge.type.model_message.openai_responses_message import OpenAIResponsesMessage
 
 
 async def convert_message_to_openai_responses(message: Message) -> OpenAIResponsesMessage:
-    role = message.role.value
+    role = message.role
     content: list[ResponseInputContentParam | ResponseOutputTextParam] = []
+    contains_pdf = False
 
     for content_item in message.contents:
         if content_item.type == ContentType.Text:
-            if role == "assistant":
-                text_content = ResponseOutputTextParam(type="output_text", text=content_item.data)
+            if role == Role.Assistant:
+                text_content = ResponseOutputTextParam(type="output_text", text=content_item.data, annotations=[])
             else:
                 text_content = ResponseInputTextParam(type="input_text", text=content_item.data)
             content.append(text_content)
@@ -30,6 +31,7 @@ async def convert_message_to_openai_responses(message: Message) -> OpenAIRespons
                 )
                 content.append(image_content)
             elif sub_type == "pdf":
+                contains_pdf = True
                 file_data, _ = await media_processor.get_base64_content_from_url(file_url)
                 pdf_content = ResponseInputFileParam(
                     type="input_file",
@@ -52,7 +54,11 @@ async def convert_message_to_openai_responses(message: Message) -> OpenAIRespons
                 )
                 content.append(text_content)
 
-    if role in ("user", "system"):
-        return EasyInputMessageParam(role=role, content=content)
+    # Force system role to user if the message contains a PDF
+    if role == Role.System and contains_pdf:
+        role = Role.User
+
+    if role in (Role.User, Role.System):
+        return EasyInputMessageParam(role=role.value, content=content)
     else:
-        return ResponseOutputMessageParam(role=role, content=content)
+        return ResponseOutputMessageParam(role=role.value, content=content)

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/llm_bridge/logic/message_preprocess/file_type_checker.py

@@ -15,11 +15,11 @@ def is_file_type_supported(file_name: str) -> bool:
 
 
 async def get_file_type(file_url: str) -> tuple[str, str]:
-    file_name = get_filename_without_timestamp(file_url)
+    file_name: str = get_filename_without_timestamp(file_url)
 
     # Treat filenames without an extension as their own extension
-    suffix = PurePosixPath(file_name).suffix.lower()
-    file_extension = suffix if suffix else '.' + file_name.lower()
+    suffix: str = PurePosixPath(file_name).suffix.lower()
+    file_extension: str = suffix if suffix else '.' + file_name.lower()
 
     if file_extension in code_file_extensions:
         return 'text', 'code'

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "LLM-Bridge"
-version = "1.15.6"
+version = "1.15.8"
 dependencies = [
     "fastapi",
     "httpx",
@@ -33,9 +33,9 @@ classifiers = [
 
 [dependency-groups]
 dev = [
-    "pytest",
-    "pytest-asyncio",
-    "python-dotenv", #dotenv
+    "pytest==9.0.2",
+    "pytest-asyncio==1.3.0",
+    "python-dotenv==1.2.1", #dotenv
 ]
 
 [tool.pytest.ini_options]

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/tests/file_type_checker_test.py

@@ -5,13 +5,20 @@ from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type
 
 @pytest.mark.asyncio
 async def test_get_file_type_with_extension():
+    file_type, sub_type = await get_file_type("https://example.com/1767243600000-markdown.md")
+    assert file_type == "text"
+    assert sub_type == "code"
+
+
+@pytest.mark.asyncio
+async def test_get_file_type_pdf():
     file_type, sub_type = await get_file_type("https://example.com/1767243600000-document.pdf")
     assert file_type == "text"
     assert sub_type == "pdf"
 
 
 @pytest.mark.asyncio
-async def test_get_file_type_without_extension():
+async def test_get_file_type_dockerfile():
     file_type, sub_type = await get_file_type("https://example.com/1767243600000-Dockerfile")
     assert file_type == "text"
     assert sub_type == "code"

llm_bridge-1.15.8/usage/config.py

@@ -0,0 +1,132 @@
+from llm_bridge import *
+
+structured_output_schema = {
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "$id": "https://example.com/product.schema.json",
+  "title": "Product",
+  "description": "A product from Acme's catalog",
+  "type": "object",
+  "properties": {
+    "productId": {
+      "description": "The unique identifier for a product",
+      "type": "integer"
+    },
+    "productName": {
+      "description": "Name of the product",
+      "type": "string"
+    },
+    "price": {
+      "description": "The price of the product",
+      "type": "number",
+      "exclusiveMinimum": 0
+    },
+    "tags": {
+      "description": "Tags for the product",
+      "type": "array",
+      "items": {
+        "type": "string"
+      },
+      "minItems": 1,
+      "uniqueItems": True
+    }
+  },
+  "required": [
+    "productId",
+    "productName",
+    "price"
+  ]
+}
+structured_output_schema = None
+
+messages = [
+    Message(
+        role=Role.System,
+        contents=[
+            # Content(type=ContentType.Text, data="You are a helpful assistant."),
+            Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746209841847-A%20Tutorial%20on%20Spectral%20Clustering.pdf")
+        ]
+    ),
+    # Message(
+    #     role=Role.User,
+    #     contents=[
+    #         Content(type=ContentType.Text, data="Hello")
+    #     ]
+    # ),
+    # Message(
+    #     role=Role.Assistant,
+    #     contents=[
+    #         Content(type=ContentType.Text, data="Hello! How can I assist you today?")
+    #     ]
+    # ),
+    Message(
+        role=Role.User,
+        contents=[
+            # Simple Question
+            Content(type=ContentType.Text, data="What's this?"),
+
+            # Thinking
+            # Content(type=ContentType.Text, data="Explain the concept of Occam's Razor and provide a simple, everyday example."),
+
+            # Web Search
+            # Content(type=ContentType.Text, data="What's the weather in NYC today?"),
+
+            # Image Understanding
+            # Content(type=ContentType.File, data="https://www.gstatic.com/webp/gallery3/1.png"),
+            # Content(type=ContentType.Text, data="What is in this image?"),
+
+            # Image Generation
+            # Content(type=ContentType.Text, data="Please generate an image of a cat."),
+
+            # URL Context
+            # Content(type=ContentType.Text, data="What is in https://www.windsnow1025.com/"),
+
+            # Code Execution
+            # Content(type=ContentType.Text, data="What is the sum of the first 50 prime numbers? Generate and run code for the calculation, and make sure you get all 50."),
+
+            # File Output
+            # Content(type=ContentType.Text, data="Create a matplotlib visualization and save it as output.png"),
+
+            # Structured Output
+            # Content(type=ContentType.Text, data="Please generate a product."),
+        ]
+    ),
+    # Message(
+    #     role=Role.User,
+    #     contents=[
+    #         Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1769429581512-Test.txt"),
+    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746208707489-image.png"),
+    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746209841847-A%20Tutorial%20on%20Spectral%20Clustering.pdf"),
+    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212253473-file_example_MP3_700KB.mp3"),
+    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212980820-file_example_MP4_480_1_5MG.mp4"),
+    #         Content(type=ContentType.Text, data="What's this?"),
+    #     ]
+    # ),
+]
+# See /llm_bridge/resources/model_prices.json for available models
+model = "gpt-5.2"
+# model = "gpt-5.1"
+# model = "gpt-5-pro"
+# model = "gpt-5"
+# model = "gpt-4.1"
+# model = "gpt-5-codex"
+# model = "gemini-3-pro-preview"
+# model = "gemini-3-pro-image-preview"
+# model = "gemini-3-flash-preview"
+# model = "grok-4-1-fast-reasoning"
+# model = "claude-sonnet-4-5"
+# model = "claude-opus-4-5"
+# api_type = "Gemini-Vertex"
+# api_type = "Gemini-Free"
+# api_type = "Gemini-Paid"
+api_type = "OpenAI"
+# api_type = "OpenAI-Azure"
+# api_type = "OpenAI-GitHub"
+# api_type = "Claude"
+# api_type = "Grok"
+temperature = 0
+stream = True
+# stream = False
+thought = True
+# thought = False
+# code_execution = True
+code_execution = False

{llm_bridge-1.15.6 → llm_bridge-1.15.8}/uv.lock

@@ -357,7 +357,7 @@ wheels = [
 
 [[package]]
 name = "llm-bridge"
-version = "1.15.6"
+version = "1.15.8"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },
@@ -397,9 +397,9 @@ requires-dist = [
 
 [package.metadata.requires-dev]
 dev = [
-    { name = "pytest" },
-    { name = "pytest-asyncio" },
-    { name = "python-dotenv" },
+    { name = "pytest", specifier = "==9.0.2" },
+    { name = "pytest-asyncio", specifier = "==1.3.0" },
+    { name = "python-dotenv", specifier = "==1.2.1" },
 ]
 
 [[package]]

llm_bridge-1.15.6/usage/config.py

@@ -1,129 +0,0 @@
-from llm_bridge import *
-
-structured_output_schema = {
-  "$schema": "https://json-schema.org/draft/2020-12/schema",
-  "$id": "https://example.com/product.schema.json",
-  "title": "Product",
-  "description": "A product from Acme's catalog",
-  "type": "object",
-  "properties": {
-    "productId": {
-      "description": "The unique identifier for a product",
-      "type": "integer"
-    },
-    "productName": {
-      "description": "Name of the product",
-      "type": "string"
-    },
-    "price": {
-      "description": "The price of the product",
-      "type": "number",
-      "exclusiveMinimum": 0
-    },
-    "tags": {
-      "description": "Tags for the product",
-      "type": "array",
-      "items": {
-        "type": "string"
-      },
-      "minItems": 1,
-      "uniqueItems": True
-    }
-  },
-  "required": [
-    "productId",
-    "productName",
-    "price"
-  ]
-}
-structured_output_schema = None
-
-messages = [
-    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")
-        ]
-    ),
-    Message(
-        role=Role.Assistant,
-        contents=[
-            Content(type=ContentType.Text, data="Hello! How can I assist you today?")
-        ]
-    ),
-    # Message(
-    #     role=Role.User,
-    #     contents=[
-    #         # Thinking
-    #         # Content(type=ContentType.Text, data="Explain the concept of Occam's Razor and provide a simple, everyday example."),
-    #
-    #         # Web Search
-    #         # Content(type=ContentType.Text, data="What's the weather in NYC today?"),
-    #
-    #         # Image Understanding
-    #         # Content(type=ContentType.File, data="https://www.gstatic.com/webp/gallery3/1.png"),
-    #         # Content(type=ContentType.Text, data="What is in this image?"),
-    #
-    #         # Image Generation
-    #         # Content(type=ContentType.Text, data="Please generate an image of a cat."),
-    #
-    #         # URL Context
-    #         # Content(type=ContentType.Text, data="What is in https://www.windsnow1025.com/"),
-    #
-    #         # Code Execution
-    #         # Content(type=ContentType.Text, data="What is the sum of the first 50 prime numbers? Generate and run code for the calculation, and make sure you get all 50."),
-    #
-    #         # File Output
-    #         # Content(type=ContentType.Text, data="Create a matplotlib visualization and save it as output.png"),
-    #
-    #         # Structured Output
-    #         # Content(type=ContentType.Text, data="Please generate a product."),
-    #     ]
-    # ),
-    Message(
-        role=Role.User,
-        contents=[
-            Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1769429581512-Test.txt"),
-            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746208707489-image.png"),
-            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746209841847-A%20Tutorial%20on%20Spectral%20Clustering.pdf"),
-            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212253473-file_example_MP3_700KB.mp3"),
-            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212980820-file_example_MP4_480_1_5MG.mp4"),
-            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1753804900037-Calculus.docx"),
-            Content(type=ContentType.Text, data="What's this?"),
-        ]
-    ),
-]
-# See /llm_bridge/resources/model_prices.json for available models
-# model = "gpt-5.2"
-# model = "gpt-5.1"
-# model = "gpt-5-pro"
-# model = "gpt-5"
-# model = "gpt-4.1"
-# model = "gpt-5-codex"
-# model = "gemini-3-pro-preview"
-# model = "gemini-3-pro-image-preview"
-model = "gemini-3-flash-preview"
-# model = "grok-4-1-fast-reasoning"
-# model = "claude-sonnet-4-5"
-# model = "claude-opus-4-5"
-api_type = "Gemini-Vertex"
-# api_type = "Gemini-Free"
-# api_type = "Gemini-Paid"
-# api_type = "OpenAI"
-# api_type = "OpenAI-Azure"
-# api_type = "OpenAI-GitHub"
-# api_type = "Claude"
-# api_type = "Grok"
-temperature = 0
-stream = True
-# stream = False
-thought = True
-# thought = False
-code_execution = True
-# code_execution = False