apply sync hack

Author: CTY-git

patchwork/common/multiturn_strategy/agentic_strategy_v2.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 import logging
 import sys
 
@@ -30,14 +31,14 @@ class Config:
 
 class AgenticStrategyV2:
  def __init__(
- self,
- api_key: str,
- template_data: dict[str, str],
- system_prompt_template: str,
- user_prompt_template: str,
- agent_configs: list[AgentConfig],
- example_json: Union[str, dict[str, Any]] = '{"output":"output text"}',
- limit: Optional[int] = None,
+  self,
+  api_key: str,
+  template_data: dict[str, str],
+  system_prompt_template: str,
+  user_prompt_template: str,
+  agent_configs: list[AgentConfig],
+  example_json: Union[str, dict[str, Any]] = '{"output":"output text"}',
+  limit: Optional[int] = None,
  ):
  self.__limit = limit
  self.__template_data = template_data
@@ -67,13 +68,14 @@ def __init__(
 
  def execute(self, limit: Optional[int] = None) -> dict:
  agents_result = dict()
+ loop = asyncio.new_event_loop()
  try:
  for index, agent in enumerate(self.__agents):
  user_message = mustache_render(self.__user_prompt_template, self.__template_data)
  message_history = None
  agent_output = None
  for i in range(limit or self.__limit or sys.maxsize):
- agent_output = agent.run_sync(user_message, message_history=message_history)
+ agent_output = loop.run_until_complete(agent.run(user_message, message_history=message_history))
  message_history = agent_output.all_messages()
  if getattr(agent_output.data, _COMPLETION_FLAG_ATTRIBUTE, False):
  break
@@ -86,26 +88,27 @@ def execute(self, limit: Optional[int] = None) -> dict:
  return dict()
 
  if len(agents_result) == 1:
- final_result = self.__summariser.run_sync(
+ final_result = loop.run_until_complete(self.__summariser.run(
  "From the actions taken by the assistant. Please give me the result.",
  message_history=next(v for _, v in agents_result.items()).all_messages(),
- )
+ ))
  else:
  agent_summaries = []
  for agent_result in agents_result.values():
- agent_summary_result = self.__summariser.run_sync(
+ agent_summary_result = loop.run_until_complete(self.__summariser.run(
  "From the actions taken by the assistant. Please give me the result.",
  message_history=agent_result.all_messages(),
- )
+ ))
  agent_summary = getattr(agent_summary_result.data, _MESSAGE_ATTRIBUTE, None)
  if agent_summary is None:
  continue
 
  agent_summaries.append(agent_summary)
  agent_summary_list = "\n* " + "\n* ".join(agent_summaries)
- final_result = self.__summariser.run_sync(
+ final_result = loop.run_until_complete(self.__summariser.run(
  "Please give me the result from the following summary of what the assistants have done."
  + agent_summary_list,
- )
+ ))
 
+ loop.close()
  return final_result.data.dict()

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.100.dev1"
+version = "0.0.100.dev2"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"