optimizely-opal.opal-tools-sdk 0.1.5.dev0__py3-none-any.whl → 0.1.6.dev0__py3-none-any.whl

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.dist-info → optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info}/METADATA

@@ -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.dist-info → optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info}/RECORD

@@ -4,8 +4,8 @@ opal_tools_sdk/auth.py,sha256=9aMiZv6n6_iu7hQA0sKg4hgNr5DzYFFuP0SWUoZf_Vw,1520
 opal_tools_sdk/decorators.py,sha256=iHal7RElOSCWjqKHIK-Hc4GKOsjuGyUg0_lsQKVQShg,5315
 opal_tools_sdk/logging.py,sha256=krvkhdql2GWh0aCWj5gfqyn5plsyLJ773WzIirK7tkU,1325
 opal_tools_sdk/models.py,sha256=YESFXdIPIFoVqP6OXtswXzHQ1tYMFVSnb8rFVe9cFj4,3403
-opal_tools_sdk/service.py,sha256=wVYe40zEZc1ynIkvowD9APU8BFHQ5fdy3j8qlsdZB1U,6965
-optimizely_opal_opal_tools_sdk-0.1.5.dev0.dist-info/METADATA,sha256=KS830xd4wFliLcbfJBuWSxj49pIH_AfajJJOrm_UG0Q,8085
-optimizely_opal_opal_tools_sdk-0.1.5.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-optimizely_opal_opal_tools_sdk-0.1.5.dev0.dist-info/top_level.txt,sha256=nCJ5PxF0rgoV6yNJvvuUaZPx4D3EWkl7gpu-6xafH1E,15
-optimizely_opal_opal_tools_sdk-0.1.5.dev0.dist-info/RECORD,,
+opal_tools_sdk/service.py,sha256=a5ZC-J1GF5y1-FdWCXdilCy7p5hG63MTTn6frBBCo4Y,6939
+optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info/METADATA,sha256=dY_LreIH84jGIXPLzpdL3MSUnzn3L5b4fRRTfKd5jvM,8085
+optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info/top_level.txt,sha256=nCJ5PxF0rgoV6yNJvvuUaZPx4D3EWkl7gpu-6xafH1E,15
+optimizely_opal_opal_tools_sdk-0.1.6.dev0.dist-info/RECORD,,