beswarm 0.1.63__tar.gz → 0.1.64__tar.gz

{beswarm-0.1.63 → beswarm-0.1.64}/PKG-INFO

@@ -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 → beswarm-0.1.64}/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-0.1.63 → beswarm-0.1.64}/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 → beswarm-0.1.64}/beswarm.egg-info/PKG-INFO

@@ -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 → beswarm-0.1.64}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.1.63"
+version = "0.1.64"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"