beswarm 0.1.63__py3-none-any.whl → 0.1.64__py3-none-any.whl

beswarm/aient/src/aient/core/response.py

@@ -54,8 +54,6 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
                 if line and '\"finishReason\": \"' in line:
                     if "stop" not in line.lower():
                         logger.error(f"finishReason: {line}")
-                    sse_string = await generate_sse_response(timestamp, model, stop="stop")
-                    yield sse_string
                     is_finish = True
                 if is_finish and '\"promptTokenCount\": ' in line:
                     json_data = parse_json_safely( "{" + line + "}")
@@ -117,6 +115,9 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
             sse_string = await generate_sse_response(timestamp, model, content=None, tools_id="chatcmpl-9inWv0yEtgn873CxMBzHeCeiHctTV", function_call_name=None, function_call_content=function_full_response)
             yield sse_string
 
+        sse_string = await generate_sse_response(timestamp, model, stop="stop")
+        yield sse_string
+
         sse_string = await generate_sse_response(timestamp, model, None, None, None, None, None, totalTokenCount, promptTokenCount, candidatesTokenCount)
         yield sse_string
 

beswarm/tools/worker.py

@@ -99,8 +99,10 @@ async def worker(goal, tools, work_dir, cache_messages=None):
                     extracted_content = f"<latest_file_content>{match.group(1)}</latest_file_content>\n\n"
                 else:
                     extracted_content = ""
-
-                conversation_history[0]["content"] = extracted_content + conversation_history[0]["content"]
+                if isinstance(conversation_history[0]["content"], str):
+                    conversation_history[0]["content"] = extracted_content + conversation_history[0]["content"]
+                elif isinstance(conversation_history[0]["content"], list) and extracted_content:
+                    conversation_history[0]["content"].append({"type": "text", "text": extracted_content})
 
             instruction_agent.conversation["default"][1:] = conversation_history
             if "find_and_click_element" in str(tools_json):
@@ -221,7 +223,10 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
                 else:
                     extracted_content = ""
 
-                conversation_history[0]["content"] = extracted_content + conversation_history[0]["content"]
+                if isinstance(conversation_history[0]["content"], str):
+                    conversation_history[0]["content"] = extracted_content + conversation_history[0]["content"]
+                elif isinstance(conversation_history[0]["content"], list) and extracted_content:
+                    conversation_history[0]["content"].append({"type": "text", "text": extracted_content})
 
             instruction_agent.conversation["default"][1:] = conversation_history
             if "find_and_click_element" in str(tools_json):

{beswarm-0.1.63.dist-info → beswarm-0.1.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.63
+Version: 0.1.64
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.63.dist-info → beswarm-0.1.64.dist-info}/RECORD

@@ -7,7 +7,7 @@ beswarm/aient/src/aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeA
 beswarm/aient/src/aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 beswarm/aient/src/aient/core/models.py,sha256=kF-HLi1I2k_G5r153ZHuiGH8_NmpTlFMfK0_myB28YQ,7366
 beswarm/aient/src/aient/core/request.py,sha256=nvF_V71svezQ0-UbnC9RB_pXo_wV6QC7WE_SANwQzxE,66195
-beswarm/aient/src/aient/core/response.py,sha256=3OASBI-Y_8VlAusY-ch3oPwBHhn1rMDhhMyDBsK1FcY,33224
+beswarm/aient/src/aient/core/response.py,sha256=YphzhA9jtQKzWb3L4XGTp9xJZ2FOzHr1aAMTsi896FQ,33201
 beswarm/aient/src/aient/core/utils.py,sha256=VQ9uutGRR_JOvECOrjeoRBO2aA6w-pGwoXnnS2UvfPU,27263
 beswarm/aient/src/aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
 beswarm/aient/src/aient/core/test/test_geminimask.py,sha256=HFX8jDbNg_FjjgPNxfYaR-0-roUrOO-ND-FVsuxSoiw,13254
@@ -127,8 +127,8 @@ beswarm/tools/repomap.py,sha256=N09K0UgwjCN7Zjg_5TYlVsulp3n2fztYlS8twalChU8,4500
 beswarm/tools/search_arxiv.py,sha256=9slwBemXjEqrd7-YgVmyMijPXlkhZCybEDRVhWVQ9B0,7937
 beswarm/tools/search_web.py,sha256=B24amOnGHnmdV_6S8bw8O2PdhZRRIDtJjg-wXcfP7dQ,11859
 beswarm/tools/think.py,sha256=WLw-7jNIsnS6n8MMSYUin_f-BGLENFmnKM2LISEp0co,1760
-beswarm/tools/worker.py,sha256=z1DmJ1TxU43tJe3K58JeBM3vNEFSuUbunRFBYoROk4w,12120
-beswarm-0.1.63.dist-info/METADATA,sha256=4-0ctzL9yxdo2QbQvHKF6o0NQN2TOvSzYX_fnMQ3s1M,3553
-beswarm-0.1.63.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.1.63.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.1.63.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=bZwMGUS4wt4yOleb2kKu0rCLHIJgPcokbCmh9bGMzXA,12679
+beswarm-0.1.64.dist-info/METADATA,sha256=KnWiUh0orLTMVypQqxxo-enfguZLdIWhVvJWCRj0tfE,3553
+beswarm-0.1.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.1.64.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.1.64.dist-info/RECORD,,