LLM-Bridge 1.7.17__py3-none-any.whl → 1.7.18__py3-none-any.whl

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/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/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/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/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.dist-info → llm_bridge-1.7.18.dist-info}/METADATA

@@ -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.dist-info → llm_bridge-1.7.18.dist-info}/RECORD

@@ -29,21 +29,21 @@ llm_bridge/logic/model_prices.py,sha256=hiXVbki3004Rrm5LQrmVfdm0lLABeygxtFB-Qn9_
 llm_bridge/logic/chat_generate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/logic/chat_generate/chat_client_factory.py,sha256=huZO5pqRNFDUK9KpCk3-pLTgxAjbVGFRu3Y56w2QPxI,2814
 llm_bridge/logic/chat_generate/chat_message_converter.py,sha256=40VTBOPXg_ocrEZMdt1ObYlm-mhRL35zWzzxv8m2xRc,1538
-llm_bridge/logic/chat_generate/media_processor.py,sha256=1Rwhr7DsEB8HY5_li_hcuHzOglTncrz885Y2dwfWluc,681
+llm_bridge/logic/chat_generate/media_processor.py,sha256=ZR8G24EHwZZr2T9iFDRmScDGyJ_kvThApABzSzK0CL0,702
 llm_bridge/logic/chat_generate/model_client_factory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/logic/chat_generate/model_client_factory/claude_client_factory.py,sha256=_iX36A4WJx2-QeX4oAurOaRuCjvm8UDhaKL8lO9iP9o,2186
 llm_bridge/logic/chat_generate/model_client_factory/gemini_client_factory.py,sha256=d_DnTZ2Rkbmo0La7uKnrVY9NO9wa6QZ8eK0hf2K1ZII,3041
-llm_bridge/logic/chat_generate/model_client_factory/openai_client_factory.py,sha256=kj4JLHiRRDEYaYU0Ac6wZ5dlBAVCX0HNN2sbWDWKPKs,3523
+llm_bridge/logic/chat_generate/model_client_factory/openai_client_factory.py,sha256=KG5QF4D6ouPssD7g2512CW1M8Iv_xA2gmjxh8FQ8z84,3531
 llm_bridge/logic/chat_generate/model_message_converter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/logic/chat_generate/model_message_converter/claude_message_converter.py,sha256=SfDhQXR7L5nCPHS4MIjwgzK_wER7qOUCc8gh-K77kKY,2441
 llm_bridge/logic/chat_generate/model_message_converter/gemini_message_converter.py,sha256=UjhzRX7sBa3-Zv1flMJd8bc8uRWMMav4UdJFhE6nVq4,1527
 llm_bridge/logic/chat_generate/model_message_converter/openai_message_converter.py,sha256=MdGcOJ1MMOD7E8RxjMk7HhobZ7lIMT118eWh9cpORgA,2180
-llm_bridge/logic/chat_generate/model_message_converter/openai_responses_message_converter.py,sha256=p3xYixa-E8lCf88qXnvgux6NICPtm6HuaGXErS8kPI8,2528
+llm_bridge/logic/chat_generate/model_message_converter/openai_responses_message_converter.py,sha256=V8eIvtS1utOd-HOOMOuR880cqONebjoAhxxdgHFg3_0,2978
 llm_bridge/logic/message_preprocess/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/logic/message_preprocess/code_file_extensions.py,sha256=5bsnSKC9PGbl6ZMy80sXfagAbz77pGjt6Z2-qwzUw48,9306
 llm_bridge/logic/message_preprocess/document_processor.py,sha256=IsVqoFgWNa9i8cRsDAfmCynJMdlvBqiCKIT9kbx96kg,2861
-llm_bridge/logic/message_preprocess/file_type_checker.py,sha256=iIQNn5DTpLJM5l9aOWjCK1ppgcrOzp22kWE45kWXuZs,1578
-llm_bridge/logic/message_preprocess/message_preprocessor.py,sha256=pWFAMo7IZP6_7Kl_B50rDABd455AEh5E6KDchfSa56A,1881
+llm_bridge/logic/message_preprocess/file_type_checker.py,sha256=nkrVki1a2udCeVqUnfIVi7Wxx8OMKbBuHw3FOlm17uo,1603
+llm_bridge/logic/message_preprocess/message_preprocessor.py,sha256=ERws57Dsu-f5LpWKqJ_SEP7omNWXeGoJaocX91P6QDQ,1907
 llm_bridge/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/resources/model_prices.json,sha256=j54U4U14USiRYEasMjmoOQxNeaHqXmMfSJEG-vmfRdY,2733
 llm_bridge/type/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -55,8 +55,8 @@ llm_bridge/type/model_message/claude_message.py,sha256=gYJUTbLUeifQMva3Axarc-VFe
 llm_bridge/type/model_message/gemini_message.py,sha256=mh8pf929g7_NkBzSOwnLXyrwSzTT4yt2FmyX7NZn0sM,4302
 llm_bridge/type/model_message/openai_message.py,sha256=xFaLY-cZoSwNd7E9BSWQjBNcRfCVH11X9s2yxXlctR0,453
 llm_bridge/type/model_message/openai_responses_message.py,sha256=be1q2euA0ybjj4NO6NxOGIRB9eJuXSb4ssUm_bM4Ocs,1529
-llm_bridge-1.7.17.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
-llm_bridge-1.7.17.dist-info/METADATA,sha256=6UlBbYsYAIbvQ5TUfJbHGedlCfvjmFD_Vl856ObmSuw,7226
-llm_bridge-1.7.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-llm_bridge-1.7.17.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
-llm_bridge-1.7.17.dist-info/RECORD,,
+llm_bridge-1.7.18.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
+llm_bridge-1.7.18.dist-info/METADATA,sha256=zNphNVFcqpPog9I1x3jrA6aETlO21EZaHE2od3y8FRs,7389
+llm_bridge-1.7.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+llm_bridge-1.7.18.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
+llm_bridge-1.7.18.dist-info/RECORD,,