LLM-Bridge 1.10.0a1__py3-none-any.whl → 1.10.0a3__py3-none-any.whl

llm_bridge/client/implementations/gemini/gemini_response_handler.py

@@ -1,11 +1,12 @@
 import base64
+import mimetypes
 from typing import Optional
 
 from google.genai import types
 
 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
+from llm_bridge.type.chat_response import Citation, ChatResponse, File
 
 
 class GeminiResponseHandler:
@@ -22,7 +23,7 @@ class GeminiResponseHandler:
         thought: str = ""
         code: str = ""
         code_output: str = ""
-        files: list[str] = []
+        files: list[File] = []
         display: Optional[str] = None
         citations: list[Citation] = extract_citations(response)
         input_tokens, stage_output_tokens = await count_gemini_tokens(response)
@@ -48,7 +49,13 @@ class GeminiResponseHandler:
                         code_output += part.code_execution_result.output
                     # File
                     if part.inline_data is not None:
-                        file = base64.b64encode(part.inline_data.data).decode('utf-8')
+                        mime_type = part.inline_data.mime_type
+                        extension = mimetypes.guess_extension(mime_type) or ""
+                        file = File(
+                            name=f"generated_file{extension}",
+                            data=base64.b64encode(part.inline_data.data).decode('utf-8'),
+                            type=mime_type,
+                        )
                         files.append(file)
 
         # Grounding Sources

llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py

@@ -12,7 +12,7 @@ from openai.types.responses import WebSearchToolParam, Response
 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
+from llm_bridge.type.chat_response import ChatResponse, Citation, File
 from llm_bridge.type.serializer import serialize
 
 
@@ -24,7 +24,7 @@ def process_openai_responses_non_stream_response(
     output_list = response.output
 
     text: str = ""
-    files: list[str] = []
+    files: list[File] = []
     citations: list[Citation] = []
 
     for output in output_list:
@@ -43,7 +43,12 @@ def process_openai_responses_non_stream_response(
                 #         )
         # Image Generation untestable due to organization verification requirement
         # if output.type == "image_generation_call":
-        #     files.append(output.result)
+        #     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)

llm_bridge/client/implementations/openai/steam_openai_responses_client.py

@@ -12,13 +12,13 @@ from openai.types.responses import ResponseStreamEvent
 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
+from llm_bridge.type.chat_response import ChatResponse, Citation, File
 from llm_bridge.type.serializer import serialize
 
 
 def process_delta(event: ResponseStreamEvent) -> ChatResponse:
     text: str = ""
-    files: list[str] = []
+    files: list[File] = []
     citations: list[Citation] = []
 
     if event.type == "response.output_text.delta":
@@ -28,7 +28,12 @@ def process_delta(event: ResponseStreamEvent) -> ChatResponse:
         pass
     # Image Generation untestable due to organization verification requirement
     # if event.type == "response.image_generation_call.partial_image":
-    #     files.append(event.partial_image_b64)
+    #     file = File(
+    #         name="generated_image.png",
+    #         data=event.partial_image_b64,
+    #         type="image/png",
+    #     )
+    #     files.append(file)
 
     chat_response = ChatResponse(
         text=text,

llm_bridge/type/chat_response.py

@@ -2,6 +2,13 @@ from dataclasses import dataclass
 from typing import Optional
 
 
+@dataclass
+class File:
+    name: str
+    data: str
+    type: str
+
+
 @dataclass
 class Citation:
     text: str
@@ -22,7 +29,7 @@ class ChatResponse:
     thought: Optional[str] = None
     code: Optional[str] = None
     code_output: Optional[str] = None
-    files: Optional[list[str]] = None
+    files: Optional[list[File]] = None
     display: Optional[str] = None
     citations: Optional[list[Citation]] = None
     error: Optional[str] = None

{llm_bridge-1.10.0a1.dist-info → llm_bridge-1.10.0a3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.10.0a1
+Version: 1.10.0a3
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT

{llm_bridge-1.10.0a1.dist-info → llm_bridge-1.10.0a3.dist-info}/RECORD

@@ -9,15 +9,15 @@ llm_bridge/client/implementations/claude/claude_token_counter.py,sha256=g8M7BFY2
 llm_bridge/client/implementations/claude/non_stream_claude_client.py,sha256=xnge1J-j_Er4K4L1UxhjuxAs_Pl6vralxTKk9yItwjI,2500
 llm_bridge/client/implementations/claude/stream_claude_client.py,sha256=q4w1UYc1yZJw5UFOtnxCoeg8MFp5soc1d57YiCTCCGE,2109
 llm_bridge/client/implementations/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llm_bridge/client/implementations/gemini/gemini_response_handler.py,sha256=TQCQ1lr0lQjnXjsTWBKRVIU9_Uf-cuv5unh3U5yYZBY,4069
+llm_bridge/client/implementations/gemini/gemini_response_handler.py,sha256=ecefEywnBwJmblAHC53PyVY8QnxyyZAG7EuIzjfQiKQ,4408
 llm_bridge/client/implementations/gemini/gemini_token_counter.py,sha256=M_mlrtu_dZTgEG9JgRaPDVyXqFtHSSVAIhsknhOaVrs,504
 llm_bridge/client/implementations/gemini/non_stream_gemini_client.py,sha256=JGNNpeln42SoXg2vGIC9xG5GGlBh6dIhz4BzYIkgraA,1302
 llm_bridge/client/implementations/gemini/stream_gemini_client.py,sha256=vqPhQdr-jaHXzn-_1PSZfpo96zM-_89XOEXIx7UBBIw,1545
 llm_bridge/client/implementations/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/client/implementations/openai/non_stream_openai_client.py,sha256=aceJm6FF6VdzVRECzJyTY8-aQjCekhhbrMPEcUN24fo,2171
-llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py,sha256=tPLLDUO7MiEVwQrlFkCSsEUcgF3RbjIiX6JH7wP9ADA,3595
+llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py,sha256=mP4N6UuyWKrJPwh4GzzAZYzIaaSPWNeULFKpVCqgpt4,3756
 llm_bridge/client/implementations/openai/openai_token_couter.py,sha256=twGl3-VvbvyrfgJBYJxL-g6OjvQvJMRp03mfOK_0bW0,1442
-llm_bridge/client/implementations/openai/steam_openai_responses_client.py,sha256=mEwMOW0gI0JaDs4juHAtWgHiUiiAqnmpqxyi5VXU_Tc,3560
+llm_bridge/client/implementations/openai/steam_openai_responses_client.py,sha256=z7r35pUzMXuM1n9B5WjlKQCX-jtZL8jbiTe0t1P1Xco,3701
 llm_bridge/client/implementations/openai/stream_openai_client.py,sha256=Izq4xH9EuLjUCBJsuSr6U4Kj6FN5c7w_oHf9wmQatXE,2988
 llm_bridge/client/model_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/client/model_client/claude_client.py,sha256=cuYORseQY8HVt-COh2J0C_mhqPehDB3A4G4vrunoSFA,1352
@@ -47,7 +47,7 @@ llm_bridge/logic/message_preprocess/message_preprocessor.py,sha256=ERws57Dsu-f5L
 llm_bridge/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_bridge/resources/model_prices.json,sha256=_2ZXKjnMDa6YSKfnWEPR_vUtuMw3cEi1d2L3IZ2kVxs,2707
 llm_bridge/type/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llm_bridge/type/chat_response.py,sha256=HL3j9sRA1LoZkeVeR8zjfOaqs8THm0aayR2OAF6vVXM,686
+llm_bridge/type/chat_response.py,sha256=zEw-my_I0-7msmlTySdBGE2vWUIPILex0UrUPqTJiYY,754
 llm_bridge/type/message.py,sha256=NyWmSSrciFfvF81aBwAH8qFpo5IpRhh8QXMselbYen8,370
 llm_bridge/type/serializer.py,sha256=moCL9y_HTO2CFg2w_jc5MljDxKgHiCo_qiz-o4l2jYU,515
 llm_bridge/type/model_message/__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.10.0a1.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
-llm_bridge-1.10.0a1.dist-info/METADATA,sha256=XFAaqs_5rXLzGKvP5L8L0Zi1QMIlgHsr4qTjlCUmWHQ,7851
-llm_bridge-1.10.0a1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-llm_bridge-1.10.0a1.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
-llm_bridge-1.10.0a1.dist-info/RECORD,,
+llm_bridge-1.10.0a3.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
+llm_bridge-1.10.0a3.dist-info/METADATA,sha256=VxdLpIocZU9fY8H_UacHxvrRD5xbA0UrxtDyamM4En8,7851
+llm_bridge-1.10.0a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+llm_bridge-1.10.0a3.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
+llm_bridge-1.10.0a3.dist-info/RECORD,,