optimizely-opal.opal-tools-sdk 0.1.5.dev0__tar.gz → 0.1.6.dev0__tar.gz

{optimizely_opal_opal_tools_sdk-0.1.5.dev0 → optimizely_opal_opal_tools_sdk-0.1.6.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimizely-opal.opal-tools-sdk
-Version: 0.1.5.dev0
+Version: 0.1.6.dev0
 Summary: SDK for creating Opal-compatible tools services
 Home-page: https://github.com/optimizely/opal-tools-sdk
 Author: Optimizely

{optimizely_opal_opal_tools_sdk-0.1.5.dev0 → optimizely_opal_opal_tools_sdk-0.1.6.dev0}/opal_tools_sdk/service.py

@@ -1,15 +1,16 @@
-from typing import Dict, List, Any, Callable, Type, Optional, get_type_hints
+from typing import Dict, List, Any, Callable, Optional, get_type_hints
 import inspect
 import logging
-from fastapi import FastAPI, APIRouter, Depends, Header, HTTPException, Request
+from fastapi import FastAPI, APIRouter, HTTPException, Request
 from fastapi.routing import APIRoute
-from pydantic import BaseModel, create_model
+from pydantic import BaseModel, ValidationError
 
-from .models import Function, Parameter, ParameterType, AuthRequirement
+from .models import Function, Parameter, AuthRequirement
 from . import _registry
 
 logger = logging.getLogger(__name__)
 
+
 class ToolsService:
     """Main class for managing Opal tools."""
 
@@ -33,15 +34,12 @@ class ToolsService:
             routes = []
             for route in app.routes:
                 if isinstance(route, APIRoute):
-                    routes.append({
-                        "path": route.path,
-                        "name": route.name,
-                        "methods": route.methods
-                    })
+                    routes.append({"path": route.path, "name": route.name, "methods": route.methods})
             return {"routes": routes}
 
     def _init_routes(self) -> None:
         """Initialize the discovery endpoint."""
+
         @self.router.get("/discovery")
         async def discovery() -> Dict[str, Any]:
             """Return the discovery information for this tools service."""
@@ -66,21 +64,23 @@ class ToolsService:
             # Auth requirements should always be a list
             if isinstance(handler.__auth_requirements__, list):
                 for req in handler.__auth_requirements__:
-                    auth_requirements.append(AuthRequirement(
-                        provider=req.get("provider", ""),
-                        scope_bundle=req.get("scope_bundle", ""),
-                        required=req.get("required", True)
-                    ))
+                    auth_requirements.append(
+                        AuthRequirement(
+                            provider=req.get("provider", ""), scope_bundle=req.get("scope_bundle", ""), required=req.get("required", True)
+                        )
+                    )
 
         return auth_requirements
 
-    def register_tool(self,
-                     name: str,
-                     description: str,
-                     handler: Callable,
-                     parameters: List[Parameter],
-                     endpoint: str,
-                     auth_requirements: Optional[List[AuthRequirement]] = None) -> None:
+    def register_tool(
+        self,
+        name: str,
+        description: str,
+        handler: Callable,
+        parameters: List[Parameter],
+        endpoint: str,
+        auth_requirements: Optional[List[AuthRequirement]] = None,
+    ) -> None:
         """Register a tool function.
 
         Args:
@@ -101,11 +101,7 @@ class ToolsService:
         final_auth_requirements = auth_requirements if auth_requirements else handler_auth_requirements
 
         function = Function(
-            name=name,
-            description=description,
-            parameters=parameters,
-            endpoint=endpoint,
-            auth_requirements=final_auth_requirements
+            name=name, description=description, parameters=parameters, endpoint=endpoint, auth_requirements=final_auth_requirements
         )
 
         self.functions.append(function)
@@ -144,7 +140,6 @@ class ToolsService:
                 param_name = list(sig.parameters.keys())[0]
                 param_type = get_type_hints(handler).get(param_name)
 
-
                 args = []
                 kwargs = {}
                 if param_type:
@@ -168,8 +163,12 @@ class ToolsService:
 
                 logger.info(f"Tool {name} returned: {result}")
                 return result
+            except ValidationError as e:
+                logger.warning(f"Invalid parameters predicted by LLM for tool {name}: {str(e)}")
+                raise HTTPException(status_code=400, detail=str(e))
             except Exception as e:
                 import traceback
+
                 logger.error(f"Error in tool {name}: {str(e)}")
                 logger.error(traceback.format_exc())
                 raise HTTPException(status_code=500, detail=str(e))

{optimizely_opal_opal_tools_sdk-0.1.5.dev0 → optimizely_opal_opal_tools_sdk-0.1.6.dev0}/optimizely_opal.opal_tools_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimizely-opal.opal-tools-sdk
-Version: 0.1.5.dev0
+Version: 0.1.6.dev0
 Summary: SDK for creating Opal-compatible tools services
 Home-page: https://github.com/optimizely/opal-tools-sdk
 Author: Optimizely

{optimizely_opal_opal_tools_sdk-0.1.5.dev0 → optimizely_opal_opal_tools_sdk-0.1.6.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "optimizely-opal.opal-tools-sdk"
-version = "0.1.5-dev"
+version = "0.1.6-dev"
 description = "SDK for creating Opal-compatible tools services"
 authors = [
     {name = "Optimizely", email = "opal-team@optimizely.com"}