LLM-Bridge 1.9.0a0__py3-none-any.whl → 1.9.0a1__py3-none-any.whl

llm_bridge/client/implementations/gemini/gemini_response_handler.py

@@ -18,13 +18,13 @@ class GeminiResponseHandler:
             self,
             response: types.GenerateContentResponse,
     ) -> ChatResponse:
-        text = ""
-        thought = ""
-        code = ""
-        code_output = ""
-        display = None
-        image_base64 = None
-        citations = extract_citations(response)
+        text: str = ""
+        thought: str = ""
+        code: str = ""
+        code_output: str = ""
+        image: Optional[str] = None
+        display: Optional[str] = None
+        citations: list[Citation] = extract_citations(response)
         input_tokens, stage_output_tokens = await count_gemini_tokens(response)
 
         printing_status = None
@@ -48,7 +48,7 @@ class GeminiResponseHandler:
                         code_output += part.code_execution_result.output
                     # Image
                     if part.inline_data is not None:
-                        image_base64 = base64.b64encode(part.inline_data.data).decode('utf-8')
+                        image = base64.b64encode(part.inline_data.data).decode('utf-8')
 
         # Grounding Sources
         if candidates := response.candidates:
@@ -74,7 +74,7 @@ class GeminiResponseHandler:
             thought=thought,
             code=code,
             code_output=code_output,
-            image=image_base64,
+            image=image,
             display=display,
             citations=citations,
             input_tokens=input_tokens,

llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py

@@ -1,6 +1,7 @@
 import logging
 import re
 from pprint import pprint
+from typing import Optional
 
 import httpx
 import openai
@@ -22,8 +23,8 @@ def process_openai_responses_non_stream_response(
 
     output_list = response.output
 
-    text = ""
-    image = None
+    text: str = ""
+    image: Optional[str] = None
     citations: list[Citation] = []
 
     for output in output_list:
@@ -40,7 +41,7 @@ def process_openai_responses_non_stream_response(
                 #                 url=annotation.url
                 #             )
                 #         )
-        # Unable to test due to organization verification requirement
+        # Image Generation untestable due to organization verification requirement
         # if output.type == "image_generation_call":
         #     image = output.result
 

llm_bridge/client/implementations/openai/steam_openai_responses_client.py

@@ -1,7 +1,7 @@
 import logging
 import re
 from pprint import pprint
-from typing import AsyncGenerator
+from typing import AsyncGenerator, Optional
 
 import httpx
 import openai
@@ -17,7 +17,8 @@ from llm_bridge.type.serializer import serialize
 
 
 def process_delta(event: ResponseStreamEvent) -> ChatResponse:
-    text = ""
+    text: str = ""
+    image: Optional[str] = None
     citations: list[Citation] = []
 
     if event.type == "response.output_text.delta":
@@ -25,9 +26,14 @@ def process_delta(event: ResponseStreamEvent) -> ChatResponse:
     # Citation is unavailable in OpenAI Responses API
     if event.type == "response.output_text.annotation.added":
         pass
+    # Image Generation untestable due to organization verification requirement
+    # if event.type == "response.image_generation_call.partial_image":
+    #     image = event.partial_image_b64
 
     chat_response = ChatResponse(
         text=text,
+        image=image,
+        citations=citations,
     )
     return chat_response
 
@@ -42,6 +48,8 @@ async def generate_chunk(
             output_tokens = count_openai_output_tokens(chat_response)
             yield ChatResponse(
                 text=chat_response.text,
+                image=chat_response.image,
+                citations=chat_response.citations,
                 input_tokens=input_tokens,
                 output_tokens=output_tokens,
             )

{llm_bridge-1.9.0a0.dist-info → llm_bridge-1.9.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.9.0a0
+Version: 1.9.0a1
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT
@@ -13,10 +13,10 @@ License-File: LICENSE
 Requires-Dist: fastapi
 Requires-Dist: httpx
 Requires-Dist: tenacity
-Requires-Dist: openai==1.99.6
+Requires-Dist: openai==1.106.1
 Requires-Dist: tiktoken==0.11.0
 Requires-Dist: google-genai==1.28.0
-Requires-Dist: anthropic==0.62.0
+Requires-Dist: anthropic==0.66.0
 Requires-Dist: PyMuPDF
 Requires-Dist: docxlatex>=1.1.1
 Requires-Dist: openpyxl
@@ -58,6 +58,12 @@ The features listed represent the maximum capabilities of each API type supporte
 | Claude   | Text, Image, PDF               | Thinking, Web Search                                       | Text          |
 | Grok     | Text, Image                    |                                                            | Text          |
 
+#### Planned Features
+
+- OpenAI: Web Search: Citations, Image Output
+- Gemini: Code Execution: Code, Code Output
+- Claude: Code Execution, File Output
+
 ## Installation
 
 ```bash

{llm_bridge-1.9.0a0.dist-info → llm_bridge-1.9.0a1.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=2fnvtNecRlkwxwht93k93r-0uY1Zz86ffGbRAFNnyjk,3989
+llm_bridge/client/implementations/gemini/gemini_response_handler.py,sha256=eqXpIx1xJK5VZtuUlye5kIVjLr1YWBU9koq2HEEUX9s,4034
 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=xUQqeDTwsf61IV5V0MFN0C-q1KGK5vlNJFetoWehfEk,3534
+llm_bridge/client/implementations/openai/non_stream_openai_responses_client.py,sha256=Ecd0dLLDXMqNEBut_t74VPDZlLObbEOoyt3vkttJEug,3595
 llm_bridge/client/implementations/openai/openai_token_couter.py,sha256=pWsuaUjoqXjnptVlRma-dItczEo9DMw2o_9uF7FPVAk,1449
-llm_bridge/client/implementations/openai/steam_openai_responses_client.py,sha256=SbfRNE03JacBCD2_u0pGt4JSXqGchDLAxbKvP1KIVhU,3177
+llm_bridge/client/implementations/openai/steam_openai_responses_client.py,sha256=52re50oU1ArIwWuocDSUN6TNDtZIP348qt4wjO3qj30,3560
 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
@@ -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.9.0a0.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
-llm_bridge-1.9.0a0.dist-info/METADATA,sha256=J2G_cF3GpKT83q7MTyUYkA-2Fi2Bgq1iKh0zc43Ydaw,7697
-llm_bridge-1.9.0a0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-llm_bridge-1.9.0a0.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
-llm_bridge-1.9.0a0.dist-info/RECORD,,
+llm_bridge-1.9.0a1.dist-info/licenses/LICENSE,sha256=m6uon-6P_CaiqcBfApMfjG9YRtDxcr40Z52JcqUCEAE,1069
+llm_bridge-1.9.0a1.dist-info/METADATA,sha256=Dw18PNKM4wo3O2xeRTG_edhsi0GNd2jP1ss-rOxAtL4,7850
+llm_bridge-1.9.0a1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+llm_bridge-1.9.0a1.dist-info/top_level.txt,sha256=PtxyrgNX1lSa1Ab_qswg0sekSXejG5zrS6b_v3Po05g,11
+llm_bridge-1.9.0a1.dist-info/RECORD,,