meshagent-tools 0.21.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

meshagent/tools/database.py

@@ -130,16 +130,17 @@ class UpdateTool(Tool):
         for k, v in schema.items():
             columns += f"column {k} => {v.to_json()}"
 
-        values_schema = {
-            "type": "object",
-            "required": [],
-            "additionalProperties": False,
-            "properties": {},
-        }
+        anyOf = []
 
         for k, v in schema.items():
-            values_schema["required"].append(k)
-            values_schema["properties"][k] = v.to_json_schema()
+            anyOf.append(
+                {
+                    "type": "object",
+                    "additionalProperties": False,
+                    "required": [k],
+                    "properties": {k: v.to_json_schema()},
+                }
+            )
 
         input_schema = {
             "type": "object",
@@ -153,7 +154,11 @@ class UpdateTool(Tool):
                     "type": "string",
                     "description": f"a lance db compatible filter, columns are: {columns}",
                 },
-                "values": values_schema,
+                "values": {
+                    "type": "array",
+                    "description": "a list of columns to update",
+                    "items": {"anyOf": anyOf},
+                },
             },
         }
 
@@ -164,9 +169,14 @@ class UpdateTool(Tool):
             input_schema=input_schema,
         )
 
-    async def execute(self, context: ToolContext, *, where: str, values: dict):
+    async def execute(self, context: ToolContext, *, where: str, values: list[dict]):
+        set = {}
+        for value in values:
+            for k, v in value.items():
+                set[k] = v
+
         await context.room.database.update(
-            table=self.table, where=where, values=values, namespace=self.namespace
+            table=self.table, where=where, values=set, namespace=self.namespace
         )
 
         return {"ok": True}
@@ -245,7 +255,7 @@ class SearchTool(Tool):
         }
 
 
-class LLMßSearchTool(Tool):
+class LLMSearchTool(Tool):
     def __init__(
         self,
         *,
@@ -490,7 +500,7 @@ class AdvancedSearchTool(Tool):
         columns = ""
 
         for k, v in schema.items():
-            columns += f"column {k} => {v.to_json()}"
+            columns += f"column {k} => {v.to_json()}\n"
 
         input_schema = {
             "type": "object",
@@ -580,9 +590,9 @@ class DatabaseToolkit(RemoteToolkit):
                 tools.append(
                     UpdateTool(table=table, schema=schema, namespace=namespace)
                 )
-                tools.append(
-                    DeleteRowsTool(table=table, schema=schema, namespace=namespace)
-                )
+                # tools.append(
+                #    DeleteRowsTool(table=table, schema=schema, namespace=namespace)
+                # )
                 tools.append(
                     AdvancedDeleteRowsTool(
                         table=table, schema=schema, namespace=namespace
@@ -590,7 +600,7 @@ class DatabaseToolkit(RemoteToolkit):
                 )
 
             tools.append(CountTool(table=table, schema=schema, namespace=namespace))
-            tools.append(SearchTool(table=table, schema=schema, namespace=namespace))
+            # tools.append(SearchTool(table=table, schema=schema, namespace=namespace))
             tools.append(
                 AdvancedSearchTool(table=table, schema=schema, namespace=namespace)
             )

meshagent/tools/version.py

@@ -1 +1 @@
-__version__ = "0.21.0"
+__version__ = "0.22.0"

{meshagent_tools-0.21.0.dist-info → meshagent_tools-0.22.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: meshagent-tools
-Version: 0.21.0
+Version: 0.22.0
 Summary: Tools for Meshagent
 License-Expression: Apache-2.0
 Project-URL: Documentation, https://docs.meshagent.com
@@ -12,7 +12,7 @@ License-File: LICENSE
 Requires-Dist: pyjwt~=2.10
 Requires-Dist: pytest~=8.4
 Requires-Dist: pytest-asyncio~=0.26
-Requires-Dist: meshagent-api~=0.21.0
+Requires-Dist: meshagent-api~=0.22.0
 Requires-Dist: aiohttp~=3.10
 Requires-Dist: opentelemetry-distro~=0.54b1
 Dynamic: license-file

{meshagent_tools-0.21.0.dist-info → meshagent_tools-0.22.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 meshagent/tools/__init__.py,sha256=1zgD5OvAJP10eERoE72VbDu9hFVfAqCWUXw3SiCYFTE,1285
 meshagent/tools/blob.py,sha256=aDW_z8R4HrmrYzAWoWm13Ypqcxcl4rL1Dc0ESnQETLM,1473
 meshagent/tools/config.py,sha256=zH2xGxg28K7Tg-aYor6LXdzf0LRxS9iE0679H1FuWhE,79
-meshagent/tools/database.py,sha256=YpCuLsLRcaDec1KAN9ax1-3dun-iSbVIUHVTM9B-3lg,18116
+meshagent/tools/database.py,sha256=Jd_YJLPIBsypSK1SziwaYb8xKnG4oafoCDs6vzAC2Oc,18396
 meshagent/tools/datetime.py,sha256=2pOUOWopYIsc5y4EoFo_1PdBaBcTSkeOOs_EqdqYTk0,17503
 meshagent/tools/discovery.py,sha256=GZcC4XCgk0ftcYCBxuWlaIsLV5vU4-gXiu0HhlDUlwY,1861
 meshagent/tools/document_tools.py,sha256=LMULXOSBjsvhKjqzxUxe8586t0Vol0v1Btu5v6ofm7A,11755
@@ -13,10 +13,10 @@ meshagent/tools/strict_schema.py,sha256=IytdAANa6lsfrsg5FsJuqYrxH9D_fayl-Lc9EwgL
 meshagent/tools/tool.py,sha256=9OAlbfaHqfgJnCDBSW-8PS0Z1K1KjWGD3JBUyiHOxAk,3131
 meshagent/tools/toolkit.py,sha256=rCCkpQBoSkmmhjnRGA4jx0QP-ds6WTJ0PkQVnf1Ls7s,3843
 meshagent/tools/uuid.py,sha256=mzRwDmXy39U5lHhd9wqV4r-ZdS8jPfDTTs4UfW4KHJQ,1342
-meshagent/tools/version.py,sha256=-reNiXr25nUU7em7_IKJzimCI10W4nA8ouxAiOr4FaQ,23
+meshagent/tools/version.py,sha256=0kk8efeJF41FZIYweGTJylbizaWrp9W3qN78RClCWIU,23
 meshagent/tools/web_toolkit.py,sha256=IoOYjOBmcbQsqWT14xYg02jjWpWmGOkDSxt2U-LQoaA,1258
-meshagent_tools-0.21.0.dist-info/licenses/LICENSE,sha256=eTt0SPW-sVNdkZe9PS_S8WfCIyLjRXRl7sUBWdlteFg,10254
-meshagent_tools-0.21.0.dist-info/METADATA,sha256=cYkX9lEH8RFBWv55ocTRJ20op7gNyiUdZTmd0whvZAw,2878
-meshagent_tools-0.21.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-meshagent_tools-0.21.0.dist-info/top_level.txt,sha256=GlcXnHtRP6m7zlG3Df04M35OsHtNXy_DY09oFwWrH74,10
-meshagent_tools-0.21.0.dist-info/RECORD,,
+meshagent_tools-0.22.0.dist-info/licenses/LICENSE,sha256=eTt0SPW-sVNdkZe9PS_S8WfCIyLjRXRl7sUBWdlteFg,10254
+meshagent_tools-0.22.0.dist-info/METADATA,sha256=3XOS3WMtZFuhAYiF7IpwiYnwtX9WgL8qr-GweleH108,2878
+meshagent_tools-0.22.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+meshagent_tools-0.22.0.dist-info/top_level.txt,sha256=GlcXnHtRP6m7zlG3Df04M35OsHtNXy_DY09oFwWrH74,10
+meshagent_tools-0.22.0.dist-info/RECORD,,