fix: make it discoverable on ui

Author: Abhijit L

litellm/proxy/guardrails/guardrail_hooks/javelin/javelin.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Dict, List, Literal, Optional, Union
+from typing import TYPE_CHECKING, Dict, List, Literal, Optional, Union, Type
 
 import litellm
 from litellm._logging import verbose_proxy_logger
@@ -18,6 +18,9 @@
 )
 from fastapi import HTTPException
 
+if TYPE_CHECKING:
+    from litellm.types.proxy.guardrails.guardrail_hooks.base import GuardrailConfigModel
+
 
 class JavelinGuardrail(CustomGuardrail):
     def __init__(
@@ -284,3 +287,15 @@ async def async_pre_call_hook(
         )
 
         return data
+
+    @staticmethod
+    def get_config_model() -> Optional[Type["GuardrailConfigModel"]]:
+        """
+        Get the config model for the Javelin guardrail.
+        """
+        from litellm.types.proxy.guardrails.guardrail_hooks.javelin import (
+            JavelinGuardrailConfigModel,
+        )
+        return JavelinGuardrailConfigModel
+
+

litellm/types/proxy/guardrails/guardrail_hooks/javelin.py

@@ -1,7 +1,10 @@
 from typing import Dict, List, Optional
 
+from pydantic import Field
 from typing_extensions import TypedDict
 
+from .base import GuardrailConfigModel
+
 
 class JavelinGuardInput(TypedDict):
     text: str
@@ -81,3 +84,27 @@ class JavelinGuardResponse(TypedDict):
             | JavelinLanguageDetectionAssessment,
         ]
     ]
+
+
+class JavelinGuardrailConfigModel(GuardrailConfigModel):
+    """Configuration parameters for the Javelin guardrail"""
+
+    guard_name: Optional[str] = Field(
+        default=None, description="Name of the Javelin guard to use"
+    )
+    api_version: Optional[str] = Field(
+        default="v1", description="API version for Javelin service"
+    )
+    metadata: Optional[Dict] = Field(
+        default=None, description="Additional metadata to send with requests"
+    )
+    application: Optional[str] = Field(
+        default=None, description="Application name for Javelin service"
+    )
+    config: Optional[Dict] = Field(
+        default=None, description="Configuration parameters for Javelin service"
+    )
+
+    @staticmethod
+    def ui_friendly_name() -> str:
+        return "Javelin Guardrails"