LLM-Bridge 1.14.0a1__tar.gz → 1.15.0__tar.gz

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.14.0a1
+Version: 1.15.0
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
@@ -37,21 +37,21 @@ PyPI: [https://pypi.org/project/LLM-Bridge/](https://pypi.org/project/LLM-Bridge
     1. **Model Message Converter**: converts general messages to model messages
         1. **Media Processor**: converts general media (Image, Audio, Video, PDF) to model compatible formats.
 3. **Chat Client**: generate stream or non-stream responses
-    - **Model Thoughts**: captures and formats the model's thinking process
-    - **Code Execution**: auto generate and execute Python code
-    - **Web Search + Citations**: extracts and formats citations from search results
+    - **Model Thoughts**: captures the model's thinking process
+    - **Code Execution**: generates and executes Python code
+    - **Web Search**: generates response from search results
     - **Token Counter**: tracks and reports input and output token usage
 
 ### Supported Features for API Types
 
 The features listed represent the maximum capabilities of each API type supported by LLM Bridge.
 
-| API Type | Input Format                   | Capabilities                                                        | Output Format     |
-|----------|--------------------------------|---------------------------------------------------------------------|-------------------|
-| OpenAI   | Text, Image, PDF               | Thinking, Web Search, Code Execution                                | Text              |
-| Gemini   | Text, Image, Video, Audio, PDF | Thinking, Web Search + Citations, Code Execution, Structured Output | Text, Image, File |
-| Claude   | Text, Image, PDF               | Thinking, Web Search, Code Execution                                | Text              |
-| Grok     | Text, Image                    |                                                                     | Text              |
+| API Type | Input Format                   | Capabilities                                            | Output Format     |
+|----------|--------------------------------|---------------------------------------------------------|-------------------|
+| OpenAI   | Text, Image, PDF               | Thinking, Web Search, Code Execution                    | Text, Image       |
+| Gemini   | Text, Image, Video, Audio, PDF | Thinking, Web Search, Code Execution, Structured Output | Text, Image, File |
+| Claude   | Text, Image, PDF               | Thinking, Web Search, Code Execution                    | Text              |
+| Grok     | Text, Image                    |                                                         | Text              |
 
 #### Planned Features
 
@@ -59,12 +59,6 @@ The features listed represent the maximum capabilities of each API type supporte
 - More features for API Types
 - Native support for Grok
 
-## Installation
-
-```bash
-pip install --upgrade llm_bridge
-```
-
 ## Development
 
 ### Python uv
@@ -73,16 +67,15 @@ pip install --upgrade llm_bridge
 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
+  uv sync --refresh
   ```
 
 ### Pycharm
 
-Add New Configuration >> uv run
-   - script: `./usage/main.py`
-   - Paths to ".env" files: `./usage/.env`
-
-If uv interpreter is not found, create a new project with uv.
+1. Add New Interpreter >> Add Local Interpreter
+  - Environment: Select existing
+  - Type: uv
+2. Add New Configuration >> uv run >> script: `./usage/main.py`
 
 ### Usage
 

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/README.md

@@ -13,21 +13,21 @@ PyPI: [https://pypi.org/project/LLM-Bridge/](https://pypi.org/project/LLM-Bridge
     1. **Model Message Converter**: converts general messages to model messages
         1. **Media Processor**: converts general media (Image, Audio, Video, PDF) to model compatible formats.
 3. **Chat Client**: generate stream or non-stream responses
-    - **Model Thoughts**: captures and formats the model's thinking process
-    - **Code Execution**: auto generate and execute Python code
-    - **Web Search + Citations**: extracts and formats citations from search results
+    - **Model Thoughts**: captures the model's thinking process
+    - **Code Execution**: generates and executes Python code
+    - **Web Search**: generates response from search results
     - **Token Counter**: tracks and reports input and output token usage
 
 ### Supported Features for API Types
 
 The features listed represent the maximum capabilities of each API type supported by LLM Bridge.
 
-| API Type | Input Format                   | Capabilities                                                        | Output Format     |
-|----------|--------------------------------|---------------------------------------------------------------------|-------------------|
-| OpenAI   | Text, Image, PDF               | Thinking, Web Search, Code Execution                                | Text              |
-| Gemini   | Text, Image, Video, Audio, PDF | Thinking, Web Search + Citations, Code Execution, Structured Output | Text, Image, File |
-| Claude   | Text, Image, PDF               | Thinking, Web Search, Code Execution                                | Text              |
-| Grok     | Text, Image                    |                                                                     | Text              |
+| API Type | Input Format                   | Capabilities                                            | Output Format     |
+|----------|--------------------------------|---------------------------------------------------------|-------------------|
+| OpenAI   | Text, Image, PDF               | Thinking, Web Search, Code Execution                    | Text, Image       |
+| Gemini   | Text, Image, Video, Audio, PDF | Thinking, Web Search, Code Execution, Structured Output | Text, Image, File |
+| Claude   | Text, Image, PDF               | Thinking, Web Search, Code Execution                    | Text              |
+| Grok     | Text, Image                    |                                                         | Text              |
 
 #### Planned Features
 
@@ -35,12 +35,6 @@ The features listed represent the maximum capabilities of each API type supporte
 - More features for API Types
 - Native support for Grok
 
-## Installation
-
-```bash
-pip install --upgrade llm_bridge
-```
-
 ## Development
 
 ### Python uv
@@ -49,16 +43,15 @@ pip install --upgrade llm_bridge
 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
+  uv sync --refresh
   ```
 
 ### Pycharm
 
-Add New Configuration >> uv run
-   - script: `./usage/main.py`
-   - Paths to ".env" files: `./usage/.env`
-
-If uv interpreter is not found, create a new project with uv.
+1. Add New Interpreter >> Add Local Interpreter
+  - Environment: Select existing
+  - Type: uv
+2. Add New Configuration >> uv run >> script: `./usage/main.py`
 
 ### Usage
 

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/llm_bridge/__init__.py

@@ -2,6 +2,6 @@ from .logic.chat_generate.chat_client_factory import create_chat_client
 from .logic.chat_generate.chat_message_converter import *
 from .logic.message_preprocess.message_preprocessor import preprocess_messages
 from .logic.model_prices import ModelPrice, get_model_prices, find_model_prices, calculate_chat_cost
-from .type.chat_response import Citation, ChatResponse
+from .type.chat_response import ChatResponse
 from .type.message import Role, Message, Content, ContentType
 from .type.serializer import serialize

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/llm_bridge/client/implementations/gemini/gemini_response_handler.py

@@ -7,7 +7,7 @@ from google.genai.types import Part
 
 from llm_bridge.client.implementations.gemini.gemini_token_counter import count_gemini_tokens
 from llm_bridge.client.implementations.printing_status import PrintingStatus
-from llm_bridge.type.chat_response import Citation, ChatResponse, File
+from llm_bridge.type.chat_response import ChatResponse, File
 
 
 class GeminiResponseHandler:
@@ -26,7 +26,6 @@ class GeminiResponseHandler:
         code_output: str = ""
         files: list[File] = []
         display: Optional[str] = None
-        citations: list[Citation] = extract_citations(response)
         input_tokens, stage_output_tokens = await count_gemini_tokens(response)
 
         parts: list[Part] = []
@@ -89,19 +88,6 @@ class GeminiResponseHandler:
             code_output=code_output,
             files=files,
             display=display,
-            citations=citations,
             input_tokens=input_tokens,
             output_tokens=output_tokens,
         )
-
-
-def extract_citations(response: types.GenerateContentResponse) -> list[Citation]:
-    citations = []
-    if candidates := response.candidates:
-        if grounding_metadata := candidates[0].grounding_metadata:
-            if grounding_supports := grounding_metadata.grounding_supports:
-                for grounding_support in grounding_supports:
-                    citation_indices = [index + 1 for index in grounding_support.grounding_chunk_indices]
-                    citation_text = grounding_support.segment.text
-                    citations.append(Citation(text=citation_text, indices=citation_indices))
-    return citations

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py

@@ -1,19 +1,17 @@
 import logging
 import re
-from pprint import pprint
-from typing import Optional
 
 import httpx
 import openai
 from fastapi import HTTPException
 from openai import APIStatusError
-from openai.types.responses import WebSearchToolParam, Response, ResponseOutputItem, ResponseOutputMessage, \
+from openai.types.responses import Response, ResponseOutputItem, ResponseOutputMessage, \
     ResponseOutputText, ResponseReasoningItem
 
 from llm_bridge.client.implementations.openai.openai_token_couter import count_openai_responses_input_tokens, \
     count_openai_output_tokens
 from llm_bridge.client.model_client.openai_client import OpenAIClient
-from llm_bridge.type.chat_response import ChatResponse, Citation, File
+from llm_bridge.type.chat_response import ChatResponse, File
 from llm_bridge.type.serializer import serialize
 
 
@@ -27,7 +25,6 @@ def process_openai_responses_non_stream_response(
     text: str = ""
     thought: str = ""
     files: list[File] = []
-    citations: list[Citation] = []
 
     for output in output_list:
         if output.type == "message":
@@ -36,27 +33,17 @@ def process_openai_responses_non_stream_response(
                 if content.type == "output_text":
                     output_text: ResponseOutputText = content
                     text += output_text.text
-                # Citation is unavailable in OpenAI Responses API
-                # elif annotations := content.annotations:
-                #     for annotation in annotations:
-                #         citations.append(
-                #             Citation(
-                #                 text=content.text[annotation.start_index:annotation.end_index],
-                #                 url=annotation.url
-                #             )
-                #         )
         elif output.type == "reasoning":
             reasoning_item: ResponseReasoningItem = output
             for summary_delta in reasoning_item.summary:
                 thought += summary_delta.text
-        # Image Generation to be tested
-        # if output.type == "image_generation_call":
-        #     file = File(
-        #         name="generated_image.png",
-        #         data=output.result,
-        #         type="image/png",
-        #     )
-        #     files.append(file)
+        if output.type == "image_generation_call":
+            file = File(
+                name="generated_image.png",
+                data=output.result,
+                type="image/png",
+            )
+            files.append(file)
 
     chat_response = ChatResponse(text=text, files=files)
     output_tokens = count_openai_output_tokens(chat_response)
@@ -64,7 +51,6 @@ def process_openai_responses_non_stream_response(
         text=text,
         thought=thought,
         files=files,
-        citations=citations,
         input_tokens=input_tokens,
         output_tokens=output_tokens,
     )

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/llm_bridge/client/implementations/openai/steam_openai_responses_client.py

@@ -1,7 +1,6 @@
 import logging
 import re
-from pprint import pprint
-from typing import AsyncGenerator, Optional
+from typing import AsyncGenerator
 
 import httpx
 import openai
@@ -12,7 +11,7 @@ from openai.types.responses import ResponseStreamEvent, ResponseReasoningSummary
 from llm_bridge.client.implementations.openai.openai_token_couter import count_openai_responses_input_tokens, \
     count_openai_output_tokens
 from llm_bridge.client.model_client.openai_client import OpenAIClient
-from llm_bridge.type.chat_response import ChatResponse, Citation, File
+from llm_bridge.type.chat_response import ChatResponse, File
 from llm_bridge.type.serializer import serialize
 
 
@@ -20,7 +19,6 @@ def process_delta(event: ResponseStreamEvent) -> ChatResponse:
     text: str = ""
     thought: str = ""
     files: list[File] = []
-    citations: list[Citation] = []
 
     if event.type == "response.output_text.delta":
         text_delta_event: ResponseTextDeltaEvent = event
@@ -28,23 +26,18 @@ def process_delta(event: ResponseStreamEvent) -> ChatResponse:
     elif event.type == "response.reasoning_summary_text.delta":
         reasoning_summary_text_delta_event: ResponseReasoningSummaryTextDeltaEvent = event
         thought = reasoning_summary_text_delta_event.delta
-    # Citation is unavailable in OpenAI Responses API
-    elif event.type == "response.output_text.annotation.added":
-        pass
-    # Image Generation to be tested
-    # if event.type == "response.image_generation_call.partial_image":
-    #     file = File(
-    #         name="generated_image.png",
-    #         data=event.partial_image_b64,
-    #         type="image/png",
-    #     )
-    #     files.append(file)
+    if event.type == "response.image_generation_call.partial_image":
+        file = File(
+            name="generated_image.png",
+            data=event.partial_image_b64,
+            type="image/png",
+        )
+        files.append(file)
 
     chat_response = ChatResponse(
         text=text,
         thought=thought,
         files=files,
-        citations=citations,
     )
     return chat_response
 
@@ -61,7 +54,6 @@ async def generate_chunk(
                 text=chat_response.text,
                 thought=chat_response.thought,
                 files=chat_response.files,
-                citations=chat_response.citations,
                 input_tokens=input_tokens,
                 output_tokens=output_tokens,
             )

{llm_bridge-1.14.0a1 → llm_bridge-1.15.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.14.0a1 → llm_bridge-1.15.0}/llm_bridge/resources/model_prices.json

@@ -1,75 +1,57 @@
 [
   {
-    "apiType": "Gemini-Vertex",
+    "apiType": "Gemini-Paid",
     "model": "gemini-3-pro-preview",
     "input": 4,
     "output": 18
   },
   {
-    "apiType": "Gemini-Vertex",
+    "apiType": "Gemini-Paid",
+    "model": "gemini-3-flash-preview",
+    "input": 1,
+    "output": 3
+  },
+  {
+    "apiType": "Gemini-Paid",
     "model": "gemini-3-pro-image-preview",
     "input": 2,
     "output": 120
   },
   {
-    "apiType": "Gemini-Vertex",
-    "model": "gemini-2.5-flash",
+    "apiType": "Gemini-Paid",
+    "model": "gemini-flash-latest",
     "input": 1,
     "output": 2.5
   },
   {
     "apiType": "Gemini-Vertex",
-    "model": "gemini-2.5-pro",
-    "input": 2.5,
-    "output": 15
-  },
-  {
-    "apiType": "Gemini-Free",
-    "model": "gemini-flash-latest",
-    "input": 0,
-    "output": 0
-  },
-  {
-    "apiType": "Gemini-Free",
-    "model": "gemini-2.5-flash",
-    "input": 0,
-    "output": 0
-  },
-  {
-    "apiType": "Gemini-Free",
-    "model": "gemini-2.5-pro",
-    "input": 0,
-    "output": 0
-  },
-  {
-    "apiType": "Gemini-Paid",
     "model": "gemini-3-pro-preview",
     "input": 4,
     "output": 18
   },
   {
-    "apiType": "Gemini-Paid",
+    "apiType": "Gemini-Vertex",
+    "model": "gemini-3-flash-preview",
+    "input": 1,
+    "output": 3
+  },
+  {
+    "apiType": "Gemini-Vertex",
     "model": "gemini-3-pro-image-preview",
     "input": 2,
     "output": 120
   },
   {
-    "apiType": "Gemini-Paid",
-    "model": "gemini-flash-latest",
-    "input": 1,
-    "output": 2.5
-  },
-  {
-    "apiType": "Gemini-Paid",
-    "model": "gemini-2.5-flash",
-    "input": 1,
-    "output": 2.5
+    "apiType": "Gemini-Free",
+    "model": "gemini-3-flash-preview",
+    "input": 0,
+    "output": 0
   },
   {
-    "apiType": "Gemini-Paid",
-    "model": "gemini-2.5-pro",
-    "input": 2.5,
-    "output": 15
+    "apiType": "OpenAI",
+    "model": "gpt-5.2",
+    "input": 1.75,
+    "output": 14
   },
   {
     "apiType": "OpenAI",
@@ -89,6 +71,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.14.0a1 → llm_bridge-1.15.0}/llm_bridge/type/chat_response.py

@@ -9,20 +9,6 @@ class File:
     type: str
 
 
-@dataclass
-class Citation:
-    text: str
-    indices: list[int]
-
-
-# TODO: adapt to different Citation formats
-# @dataclass
-# class Citation:
-#     text: str
-#     indices: Optional[list[int]] = None
-#     url: Optional[str] = None
-
-
 @dataclass
 class ChatResponse:
     text: Optional[str] = None
@@ -31,7 +17,6 @@ class ChatResponse:
     code_output: Optional[str] = None
     files: Optional[list[File]] = None
     display: Optional[str] = None
-    citations: Optional[list[Citation]] = None
     error: Optional[str] = None
     input_tokens: Optional[int] = 0
     output_tokens: Optional[int] = 0

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "LLM-Bridge"
-version = "1.14.0-alpha.1"
+version = "1.15.0"
 dependencies = [
     "fastapi",
     "httpx",

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/usage/main.py

@@ -98,7 +98,7 @@ messages = [
             # 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?"),
+            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"),
@@ -118,7 +118,7 @@ messages = [
             # Content(type=ContentType.Text, data="Please implement a minimum example of Neural Network in `script.py`"),
 
             # Structured Output
-            Content(type=ContentType.Text, data="Please generate a product."),
+            # Content(type=ContentType.Text, data="Please generate a product."),
         ]
     ),
     # Message(
@@ -134,27 +134,28 @@ messages = [
     # ),
 ]
 # 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 = "gemini-3-pro-preview"
+model = "gemini-3-pro-preview"
 # model = "gemini-3-pro-image-preview"
-# model = "gemini-flash-latest"
+# model = "gemini-3-flash-preview"
 # model = "grok-4-1-fast-reasoning"
-model = "claude-sonnet-4-5"
+# model = "claude-sonnet-4-5"
 # model = "claude-opus-4-5"
-# api_type = "Gemini-Vertex"
+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 = "Claude"
 # api_type = "Grok"
 temperature = 0
-stream = True
-# stream = False
+# stream = True
+stream = False
 thought = True
 # thought = False
 code_execution = True

{llm_bridge-1.14.0a1 → llm_bridge-1.15.0}/uv.lock

@@ -357,7 +357,7 @@ wheels = [
 
 [[package]]
 name = "llm-bridge"
-version = "1.14.0a1"
+version = "1.15.0a0"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },
@@ -726,6 +726,8 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/72/74/448b6172927c829c6a3fba80078d7b0a016ebbe2c9ee528821f5ea21677a/pymupdf-1.26.7-cp310-abi3-macosx_11_0_arm64.whl", hash = "sha256:31aa9c8377ea1eea02934b92f4dcf79fb2abba0bf41f8a46d64c3e31546a3c02", size = 22470101, upload-time = "2025-12-11T21:47:37.105Z" },
     { url = "https://files.pythonhosted.org/packages/65/e7/47af26f3ac76be7ac3dd4d6cc7ee105948a8355d774e5ca39857bf91c11c/pymupdf-1.26.7-cp310-abi3-manylinux_2_28_aarch64.whl", hash = "sha256:e419b609996434a14a80fa060adec72c434a1cca6a511ec54db9841bc5d51b3c", size = 23502486, upload-time = "2025-12-12T09:51:25.824Z" },
     { url = "https://files.pythonhosted.org/packages/2a/6b/3de1714d734ff949be1e90a22375d0598d3540b22ae73eb85c2d7d1f36a9/pymupdf-1.26.7-cp310-abi3-manylinux_2_28_x86_64.whl", hash = "sha256:69dfc78f206a96e5b3ac22741263ebab945fdf51f0dbe7c5757c3511b23d9d72", size = 24115727, upload-time = "2025-12-11T21:47:51.274Z" },
+    { url = "https://files.pythonhosted.org/packages/62/9b/f86224847949577a523be2207315ae0fd3155b5d909cd66c274d095349a3/pymupdf-1.26.7-cp310-abi3-musllinux_1_2_x86_64.whl", hash = "sha256:1d5106f46e1ca0d64d46bd51892372a4f82076bdc14a9678d33d630702abca36", size = 24324386, upload-time = "2025-12-12T14:58:45.483Z" },
+    { url = "https://files.pythonhosted.org/packages/85/8e/a117d39092ca645fde8b903f4a941d9aa75b370a67b4f1f435f56393dc5a/pymupdf-1.26.7-cp310-abi3-win32.whl", hash = "sha256:7c9645b6f5452629c747690190350213d3e5bbdb6b2eca227d82702b327f6eee", size = 17203888, upload-time = "2025-12-12T13:59:57.613Z" },
     { url = "https://files.pythonhosted.org/packages/dd/c3/d0047678146c294469c33bae167c8ace337deafb736b0bf97b9bc481aa65/pymupdf-1.26.7-cp310-abi3-win_amd64.whl", hash = "sha256:425b1befe40d41b72eb0fe211711c7ae334db5eb60307e9dd09066ed060cceba", size = 18405952, upload-time = "2025-12-11T21:48:02.947Z" },
 ]