LLM-Bridge 1.9.0a1__tar.gz → 1.10.0a1__tar.gz

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1/LLM_Bridge.egg-info}/PKG-INFO

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

{llm_bridge-1.9.0a1/LLM_Bridge.egg-info → llm_bridge-1.10.0a1}/PKG-INFO

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

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/llm_bridge/client/implementations/gemini/gemini_response_handler.py

@@ -22,7 +22,7 @@ class GeminiResponseHandler:
         thought: str = ""
         code: str = ""
         code_output: str = ""
-        image: Optional[str] = None
+        files: list[str] = []
         display: Optional[str] = None
         citations: list[Citation] = extract_citations(response)
         input_tokens, stage_output_tokens = await count_gemini_tokens(response)
@@ -46,9 +46,10 @@ class GeminiResponseHandler:
                     # Code Output
                     if part.code_execution_result is not None:
                         code_output += part.code_execution_result.output
-                    # Image
+                    # File
                     if part.inline_data is not None:
-                        image = base64.b64encode(part.inline_data.data).decode('utf-8')
+                        file = base64.b64encode(part.inline_data.data).decode('utf-8')
+                        files.append(file)
 
         # Grounding Sources
         if candidates := response.candidates:
@@ -74,7 +75,7 @@ class GeminiResponseHandler:
             thought=thought,
             code=code,
             code_output=code_output,
-            image=image,
+            files=files,
             display=display,
             citations=citations,
             input_tokens=input_tokens,

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py

@@ -24,7 +24,7 @@ def process_openai_responses_non_stream_response(
     output_list = response.output
 
     text: str = ""
-    image: Optional[str] = None
+    files: list[str] = []
     citations: list[Citation] = []
 
     for output in output_list:
@@ -43,13 +43,13 @@ def process_openai_responses_non_stream_response(
                 #         )
         # Image Generation untestable due to organization verification requirement
         # if output.type == "image_generation_call":
-        #     image = output.result
+        #     files.append(output.result)
 
-    chat_response = ChatResponse(text=text, image=image)
+    chat_response = ChatResponse(text=text, files=files)
     output_tokens = count_openai_output_tokens(chat_response)
     return ChatResponse(
         text=text,
-        image=image,
+        files=files,
         citations=citations,
         input_tokens=input_tokens,
         output_tokens=output_tokens,

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/llm_bridge/client/implementations/openai/openai_token_couter.py

@@ -35,7 +35,7 @@ def count_openai_responses_input_tokens(messages: list[OpenAIResponsesMessage])
 
 def count_openai_output_tokens(chat_response: ChatResponse) -> int:
     text = chat_response.text
-    file_count = 1 if chat_response.image else 0
+    file_count = len(chat_response.files)
 
     return num_tokens_from_text(text) + file_count * 1000
 

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/llm_bridge/client/implementations/openai/steam_openai_responses_client.py

@@ -18,7 +18,7 @@ from llm_bridge.type.serializer import serialize
 
 def process_delta(event: ResponseStreamEvent) -> ChatResponse:
     text: str = ""
-    image: Optional[str] = None
+    files: list[str] = []
     citations: list[Citation] = []
 
     if event.type == "response.output_text.delta":
@@ -28,11 +28,11 @@ def process_delta(event: ResponseStreamEvent) -> ChatResponse:
         pass
     # Image Generation untestable due to organization verification requirement
     # if event.type == "response.image_generation_call.partial_image":
-    #     image = event.partial_image_b64
+    #     files.append(event.partial_image_b64)
 
     chat_response = ChatResponse(
         text=text,
-        image=image,
+        files=files,
         citations=citations,
     )
     return chat_response
@@ -48,7 +48,7 @@ async def generate_chunk(
             output_tokens = count_openai_output_tokens(chat_response)
             yield ChatResponse(
                 text=chat_response.text,
-                image=chat_response.image,
+                files=chat_response.files,
                 citations=chat_response.citations,
                 input_tokens=input_tokens,
                 output_tokens=output_tokens,

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/llm_bridge/type/chat_response.py

@@ -22,7 +22,7 @@ class ChatResponse:
     thought: Optional[str] = None
     code: Optional[str] = None
     code_output: Optional[str] = None
-    image: Optional[str] = None
+    files: Optional[list[str]] = None
     display: Optional[str] = None
     citations: Optional[list[Citation]] = None
     error: Optional[str] = None

{llm_bridge-1.9.0a1 → llm_bridge-1.10.0a1}/pyproject.toml

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