LLM-Bridge 1.7.17__tar.gz → 1.7.18__tar.gz

{llm_bridge-1.7.17 → llm_bridge-1.7.18/LLM_Bridge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.7.17
+Version: 1.7.18
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
@@ -72,10 +72,15 @@ pytest
 
 ## Quick Start
 
-See `./usage/`
+### Setup
+
+1. Copy `./.env.example` and rename it to `./.env`, then fill in the environment variables.
+2. Install requirements: `pip install -r requirements.txt`
 
 ### Workflow
 
+See `./usage/`
+
 ```python
 from typing import AsyncGenerator
 

{llm_bridge-1.7.17/LLM_Bridge.egg-info → llm_bridge-1.7.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.7.17
+Version: 1.7.18
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
@@ -72,10 +72,15 @@ pytest
 
 ## Quick Start
 
-See `./usage/`
+### Setup
+
+1. Copy `./.env.example` and rename it to `./.env`, then fill in the environment variables.
+2. Install requirements: `pip install -r requirements.txt`
 
 ### Workflow
 
+See `./usage/`
+
 ```python
 from typing import AsyncGenerator
 

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/README.md

@@ -42,10 +42,15 @@ pytest
 
 ## Quick Start
 
-See `./usage/`
+### Setup
+
+1. Copy `./.env.example` and rename it to `./.env`, then fill in the environment variables.
+2. Install requirements: `pip install -r requirements.txt`
 
 ### Workflow
 
+See `./usage/`
+
 ```python
 from typing import AsyncGenerator
 

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/llm_bridge/logic/chat_generate/media_processor.py

@@ -8,10 +8,11 @@ async def get_raw_content_from_url(req_url: str) -> tuple[bytes, str]:
     return file_data, media_type
 
 
+# Base64 Encoded
 async def get_encoded_content_from_url(req_url: str) -> tuple[str, str]:
-    img_data, media_type = await get_raw_content_from_url(req_url)
-    base64_image = base64.b64encode(img_data).decode('utf-8')
-    return base64_image, media_type
+    media_data, media_type = await get_raw_content_from_url(req_url)
+    base64_media = base64.b64encode(media_data).decode('utf-8')
+    return base64_media, media_type
 
 
 async def get_openai_image_content_from_url(req_img_url: str) -> str:

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/llm_bridge/logic/chat_generate/model_client_factory/openai_client_factory.py

@@ -44,7 +44,7 @@ async def create_openai_client(
     else:
         raise HTTPException(status_code=500, detail="API Type not matched")
 
-    if api_type in ("OpenAI", "OpenAI-Azure"):
+    if api_type in ("OpenAI", "OpenAI-Azure", "Grok"):
         use_responses_api = True
     else:
         use_responses_api = False

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/llm_bridge/logic/chat_generate/model_message_converter/openai_responses_message_converter.py

@@ -1,8 +1,8 @@
 from openai.types.responses import ResponseInputTextParam, ResponseInputImageParam, ResponseOutputTextParam, \
-    ResponseInputContentParam, EasyInputMessageParam, ResponseOutputMessageParam
+    ResponseInputContentParam, EasyInputMessageParam, ResponseOutputMessageParam, ResponseInputFileParam
 
 from llm_bridge.logic.chat_generate import media_processor
-from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type
+from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_file_name
 from llm_bridge.type.message import Message, ContentType
 from llm_bridge.type.model_message.openai_responses_message import OpenAIResponsesMessage
 
@@ -29,6 +29,14 @@ async def convert_message_to_openai_responses(message: Message) -> OpenAIRespons
                     detail="auto"
                 )
                 content.append(image_content)
+            elif sub_type == "pdf":
+                file_data, _ = await media_processor.get_encoded_content_from_url(file_url)
+                pdf_content = ResponseInputFileParam(
+                    type="input_file",
+                    filename=get_file_name(file_url),
+                    file_data=f"data:application/pdf;base64,{file_data}",
+                )
+                content.append(pdf_content)
             # TODO: Responses API is currently unsupported for audio input
             # elif file_type == "audio":
             #     encoded_string = await media_processor.get_gpt_audio_content_from_url(file_url)

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/llm_bridge/logic/message_preprocess/file_type_checker.py

@@ -41,7 +41,8 @@ async def get_file_type(file_url: str) -> tuple[str, str]:
     return 'unknown', 'unknown'
 
 
-def get_file_name(file_url) -> str:
+# Without Timestamp
+def get_file_name(file_url: str) -> str:
     base_name = os.path.basename(file_url)
     match = re.search(r'-(.+)', base_name)
     if match:

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/llm_bridge/logic/message_preprocess/message_preprocessor.py

@@ -21,7 +21,7 @@ async def extract_text_files_to_message(message: Message, api_type: str) -> None
         if file_type != "text" and file_type != "application":
             continue
 
-        if sub_type == "pdf" and api_type in ("Gemini-Free", "Gemini-Paid", "Claude"):
+        if sub_type == "pdf" and api_type in ("OpenAI", "OpenAI-Azure", "Gemini-Free", "Gemini-Paid", "Claude"):
             continue
             
         filename = get_file_name(file_url)

{llm_bridge-1.7.17 → llm_bridge-1.7.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "LLM-Bridge"
-version = "1.7.17"
+version = "1.7.18"
 authors = [
     {name = "windsnow1025", email = "windsnow1025@gmail.com"}
 ]