language-model-common 3.0.7__tar.gz → 3.0.8__tar.gz

{language_model_common-3.0.7/language_model_common.egg-info → language_model_common-3.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 3.0.7
+Version: 3.0.8
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

language_model_common-3.0.8/VERSION

@@ -0,0 +1 @@
+3.0.8

{language_model_common-3.0.7 → language_model_common-3.0.8/language_model_common.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 3.0.7
+Version: 3.0.8
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

{language_model_common-3.0.7 → language_model_common-3.0.8}/languagemodelcommon/converters/tool_event_handlers.py

@@ -19,6 +19,10 @@ from languagemodelcommon.converters.streaming_formatters import (
     convert_message_content_into_string,
     make_tool_key,
 )
+from languagemodelcommon.utilities.chat_message_helpers import (
+    extract_image_output_parts,
+    iter_message_content_text_chunks,
+)
 from languagemodelcommon.file_managers.file_writer import (
     DebugFileWriteResult,
     FileWriter,
@@ -202,6 +206,29 @@ class ToolEventHandler(StreamContextMixin):
             if tool_end_event:
                 yield tool_end_event
 
+            # A tool's own returned image (e.g. create_health_link's QR code) lives on
+            # tool_message.content as an ImageContent-derived block, not on `artifact`.
+            # create_tool_end_sse_event above only ever carries `tool_message_content`,
+            # which redacts image blocks to a "[image: mime]" placeholder
+            # (convert_message_content_into_string -> _summarize_content_block) -- that
+            # string is the only thing BAI-806's streaming fix (_handle_on_chat_model_stream)
+            # doesn't cover, since that hook only sees the model's own AIMessage chunks, never
+            # a ToolMessage. Emit the same atomic output_image event here so a tool-returned
+            # image reaches the client without depending on the model echoing it back.
+            image_parts = extract_image_output_parts(
+                non_text_blocks=iter_message_content_text_chunks(
+                    content=tool_message.content,
+                    include_non_text_placeholders=False,
+                ).non_text_blocks
+            )
+            for image_part in image_parts:
+                image_event = chat_request_wrapper.create_image_output_sse_event(
+                    request_id=request_information.request_id,
+                    image_part=image_part,
+                )
+                if image_event:
+                    yield image_event
+
             logger.debug(
                 "Tool %s has artifact of type %s: %s",
                 tool_name,

{language_model_common-3.0.7 → language_model_common-3.0.8}/languagemodelcommon/utilities/openai/responses_api_converter.py

@@ -146,6 +146,14 @@ def convert_responses_api_to_single_message(
     combined_content = ""
     combined_kwargs: Dict[str, Any] = {}
     function_calls_list: List[Dict[str, Any]] = []
+    # Tracks which LangChain message class to return. Defaults to "assistant" to
+    # preserve pre-existing behavior for Responses API *output* items (assistant
+    # message/function_call items), which never carried a non-assistant role.
+    # Only the `else` branch below -- which handles plain input items such as
+    # EasyInputMessageParam (role + content, no "type": "message" wrapper) -- can
+    # override this to "user"/"system", since only *input* items legitimately
+    # carry those roles here.
+    target_role: str = "assistant"
 
     if output_type == "message" and response.get("role") == "assistant":
         content_list = response.get("content", [])
@@ -185,6 +193,10 @@ def convert_responses_api_to_single_message(
                 elif isinstance(item, str):
                     combined_content += item
 
+        role = response.get("role")
+        if role in ("user", "system"):
+            target_role = role
+
     if function_calls_list:
         combined_kwargs["tool_calls"] = function_calls_list
         if not combined_content:
@@ -193,6 +205,10 @@ def convert_responses_api_to_single_message(
     if not combined_content and not combined_kwargs:
         raise ValueError("No valid message content found in output_item")
 
+    if target_role == "user":
+        return HumanMessage(content=combined_content)
+    if target_role == "system":
+        return SystemMessage(content=combined_content)
     return AIMessage(content=combined_content, additional_kwargs=combined_kwargs)
 
 

language_model_common-3.0.7/VERSION

@@ -1 +0,0 @@
-3.0.7