qol stuff

Author: CTY-git

patchwork/common/multiturn_strategy/planning_strategy.py

@@ -137,13 +137,15 @@ def usage(self):
         }
 
     def __agent_run(self, agent: Agent, prompt: str, **kwargs) -> AgentRunResult[Any]:
-        planner_response = agent.run_sync(prompt, **kwargs)
+        loop = asyncio.new_event_loop()
+        planner_response = loop.run_until_complete(agent.run(prompt, **kwargs))
+        loop.close()
         self.__request_tokens += planner_response.usage().request_tokens
         self.__response_tokens += planner_response.usage().response_tokens
+
         return planner_response
 
     def run(self, task: str, conversation_limit: int = 10) -> dict:
-        loop = asyncio.new_event_loop()
 
         planner_response = self.__agent_run(self.planner, f"Produce the initial plan for {task}")
         planner_history = planner_response.all_messages()
@@ -199,5 +201,4 @@ def run(self, task: str, conversation_limit: int = 10) -> dict:
             message_history=planner_history,
         )
 
-        loop.close()
         return final_result.data.dict()

patchwork/common/tools/db_query_tool.py

@@ -8,13 +8,14 @@ class DatabaseQueryTool(Tool, tool_name="db_query_tool"):
     def __init__(self, inputs: dict[str, Any]):
         super().__init__()
         self.db_settings = inputs.copy()
+        self.db_dialect = inputs.get("db_dialect", "SQL")
 
     @property
     def json_schema(self) -> dict:
         return {
             "name": "db_query_tool",
-            "description": """\
-Run SQL Query on current database.
+            "description": f"""\
+Run SQL Query on current {self.db_dialect} database.
 """,
             "input_schema": {
                 "type": "object",