fix(ToolGuard): updates

altk/core/llm/base.py

@@ -175,6 +175,12 @@ def provider_class(cls) -> Type[Any]:
         Underlying SDK client class, e.g. openai.OpenAI or litellm.LiteLLM.
         """
 
+    @abstractmethod
+    def get_model_id(self) -> str|None:
+        """
+        returns the id of the model. eg: "gpt-5.1", "meta-llama/llama-4-maverick-17b-128e-instruct-fp8", and "claude-4-sonnet"
+        """
+
     @abstractmethod
     def _register_methods(self) -> None:
         """

altk/core/llm/providers/auto_from_env/auto_from_env.py

@@ -46,6 +46,10 @@ def __init__(self) -> None:
     def provider_class(cls) -> Type[Any]:
         raise NotImplementedError
 
+    def get_model_id(self) -> str|None:
+        if self._chosen_provider:
+            return self._chosen_provider.get_model_id()
+
     def _register_methods(self) -> None:
         if self._chosen_provider:
             self._chosen_provider._register_methods()

altk/core/llm/providers/ibm_watsonx_ai/ibm_watsonx_ai.py

@@ -127,6 +127,9 @@ def provider_class(cls) -> Type[Any]:
         """
         return ModelInference  # type: ignore
 
+    def get_model_id(self) -> str:
+        return self.model_name # type: ignore
+
     def _register_methods(self) -> None:
         """
         Register how to call watsonx methods:
@@ -471,6 +474,9 @@ def provider_class(cls) -> Type[Any]:
         """
         return ModelInference  # type: ignore
 
+    def get_model_id(self) -> str:
+        return self.model_name # type: ignore
+
     def _register_methods(self) -> None:
         """
         Register how to call watsonx methods for validation:

altk/core/llm/providers/litellm/litellm.py

@@ -35,6 +35,9 @@ def __init__(
     @classmethod
     def provider_class(cls) -> type:
         return litellm  # type: ignore
+
+    def get_model_id(self) -> str:
+        return self.model_path
 
     def _register_methods(self) -> None:
         """Register LiteLLM methods - only chat and chat_async are supported"""
@@ -302,6 +305,9 @@ def provider_class(cls) -> Type[Any]:
         Must be callable with no arguments (per LLMClient __init__ logic).
         """
         return litellm  # type: ignore
+
+    def get_model_id(self) -> str:
+        return self.model_path
 
     def _register_methods(self) -> None:
         """

altk/core/llm/providers/openai/openai.py

@@ -77,6 +77,9 @@ def transform_min_tokens(value: Any, mode: Any) -> dict[str, Any]:
         )
         self._parameter_mapper.set_custom_transform("min_tokens", transform_min_tokens)
 
+    def get_model_id(self) -> str|None:
+        if self._other_kwargs:
+            return self._other_kwargs.get("model")
 
 class BaseValidatingOpenAIClient(ValidatingLLMClient):
     """Base class for validating OpenAI and Azure OpenAI clients with shared parameter mapping"""
@@ -146,6 +149,10 @@ def transform_min_tokens(value: Any, mode: Any) -> dict[str, Any]:
         )
         self._parameter_mapper.set_custom_transform("min_tokens", transform_min_tokens)
 
+    def get_model_id(self) -> str|None:
+        if self._other_kwargs:
+            return self._other_kwargs.get("model")
+
 
 @register_llm("openai.sync")
 class SyncOpenAIClient(BaseOpenAIClient, BaseLLMClient):

altk/pre_llm/core/types.py

@@ -125,4 +125,4 @@ def get_topics(
         n_results: int = 10,
         query_kwargs: Dict[str, Any] | None = None,
         distance_threshold: float | None = None,
-    ) -> List[RetrievedTopic]: ...
+    ) -> List[RetrievedTopic]: ...

altk/pre_tool/core/types.py

@@ -90,44 +90,4 @@ class SPARCReflectionRunOutput(PreToolReflectionRunOutput):
 
     output: SPARCReflectionRunOutputSchema = Field(
         default_factory=lambda: SPARCReflectionRunOutputSchema()
-    )
-
-
-class ToolGuardBuildInputMetaData(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-    policy_text: str = Field(description="Text of the policy document file")
-    short1: bool = Field(default=True, description="Run build short or long version. ")
-    validating_llm_client: LLMClient = Field(
-        description="ValidatingLLMClient for build time"
-    )
-
-
-class ToolGuardBuildInput(ComponentInput):
-    metadata: ToolGuardBuildInputMetaData = Field(
-        default_factory=lambda: ToolGuardBuildInputMetaData()
-    )
-
-
-class ToolGuardRunInputMetaData(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-    tool_name: str = Field(description="Tool name")
-    tool_parms: dict = Field(default={}, description="Tool parameters")
-    llm_client: LLMClient = Field(description="LLMClient for build time")
-
-
-class ToolGuardRunInput(ComponentInput):
-    metadata: ToolGuardRunInputMetaData = Field(
-        default_factory=lambda: ToolGuardRunInputMetaData()
-    )
-
-
-class ToolGuardRunOutputMetaData(BaseModel):
-    error_message: Union[str, bool] = Field(
-        description="Error string or False if no error occurred"
-    )
-
-
-class ToolGuardRunOutput(ComponentOutput):
-    output: ToolGuardRunOutputMetaData = Field(
-        default_factory=lambda: ToolGuardRunOutputMetaData()
-    )
+    )