Toolguard comply with Python 3.10

altk/pre_tool/toolguard/README.md

@@ -24,7 +24,7 @@ uv pip install "agent-lifecycle-toolkit[toolguard]"
 ```
 
 ## ToolGuardSpecComponent
-This component gets a set of tools and a policy document and generates multiple ToolGuard specifications, known as `ToolGuardSpec`s. Each specification is attached to a tool, and it declares a precondition that must apply before invoking the tool. The specification has a `name`, `description`, list of `references` to the original policy document, a set of declarative `compliance_examples`, describing test cases that the toolGuard should allow the tool invocation, and `violation_examples`, where the toolGuard should raise an exception.
+This component gets a set of tools and a policy document and generates multiple ToolGuard specifications, known as `ToolGuardSpec`s. Each specification is attached to a tool, and it declares a precondition that must apply before invoking the tool. The specification has a `name`, `description`, list of `references` to the original policy document, a set of declarative `compliance_examples`, describing test cases that the toolGuard should allow the tool invocation, and `violation_examples`, where the toolGuard should raise a PolicyViolationException.
 
 This component supports only a `build` phase. The generated specifications are returned as output, and are also saved to a specified file system directory.
 The specifications are aimed to be used as input into our next component - the `ToolGuardCodeComponent` described below.

altk/pre_tool/toolguard/llm_client.py

@@ -1,8 +1,10 @@
 from typing import Union, cast
-from toolguard.buildtime.llm.tg_litellm import LanguageModelBase
+from toolguard.buildtime.llm import LanguageModelBase
 from altk.core.llm.types import GenerationArgs
 from altk.core.llm import ValidatingLLMClient, LLMClient
 
+MAX_TOKENS = 16000  # min of claude4, oss, gpt 4o, ...
+
 
 class TG_LLMClient(LanguageModelBase):
     def __init__(self, llm_client: Union[LLMClient, ValidatingLLMClient]):
@@ -14,9 +16,9 @@ async def generate(self, messages: list[dict]) -> str:
             return await llm_client.generate_async(
                 prompt=messages,
                 schema=str,
-                generation_args=GenerationArgs(max_tokens=10000),
+                generation_args=GenerationArgs(max_tokens=MAX_TOKENS),
             )
 
         return await self.llm_client.generate_async(
-            prompt=messages, generation_args=GenerationArgs(max_tokens=10000)
+            prompt=messages, generation_args=GenerationArgs(max_tokens=MAX_TOKENS)
         )  # type: ignore

altk/pre_tool/toolguard/toolguard_spec_component.py

@@ -78,5 +78,4 @@ async def _abuild(self, data: ToolGuardSpecBuildInput) -> ToolGuardSpecs:
             tools=to_tools(data.tools),
             work_dir=data.out_dir,
             llm=llm,
-            short=True,
         )

pyproject.toml

@@ -129,7 +129,7 @@ spotlight = [
 ]
 
 toolguard = [
-    "toolguard>=0.2.7",
+    "toolguard>=0.2.9",
 ]
 
 refraction = [

tests/pre_tool/toolguard/inputs/tool_functions.py

@@ -3,13 +3,11 @@ def divide_tool(g: float, h: float) -> float:
 
     Args:
         g (float): The dividend.
-        h (float): The divisor (must not be zero).
+        h (float): The divisor.
 
     Returns:
         float: The result of g divided by h.
 
-    Raises:
-        ZeroDivisionError: If h is zero.
     """
     return g / h