optimizely-opal.opal-tools-sdk 0.1.1.dev0__tar.gz → 0.1.2.dev0__tar.gz

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimizely-opal.opal-tools-sdk
-Version: 0.1.1.dev0
+Version: 0.1.2.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.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.2.dev0}/opal_tools_sdk/decorators.py

@@ -75,7 +75,11 @@ def tool(name: str, description: str, auth_requirements: Optional[List[Dict[str,
                     param_type = ParameterType.dictionary
 
                 # Determine if required
-                required = field_info.default is ... if hasattr(field_info, 'default') else True
+                field_info_extra = getattr(field_info, "json_schema_extra") or {}
+                if "required" in field_info_extra:
+                    required = field_info_extra["required"]
+                else:
+                    required = field_info.default is ... if hasattr(field_info, 'default') else True
 
                 # Get description
                 description_text = ""

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.2.dev0}/opal_tools_sdk/service.py

@@ -132,6 +132,11 @@ class ToolsService:
                 if auth_data:
                     print(f"Auth data provided for provider: {auth_data.get('provider', 'unknown')}")
 
+                # Extract environment data if available
+                environment = body.get("environment", {})
+                if environment:
+                    print(f"Environment data provided: {environment}")
+
                 print(f"Extracted parameters: {params}")
 
                 # Get the parameter model from handler's signature
@@ -139,24 +144,27 @@ class ToolsService:
                 param_name = list(sig.parameters.keys())[0]
                 param_type = get_type_hints(handler).get(param_name)
 
-                # Check signature to see if it accepts auth data
-                accepts_auth = len(sig.parameters) > 1
 
+                args = []
+                kwargs = {}
                 if param_type:
                     # Create instance of param model
                     model_instance = param_type(**params)
-                    if accepts_auth:
-                        # Call with auth data if the handler accepts it
-                        result = await handler(model_instance, auth_data)
-                    else:
-                        # Call without auth data
-                        result = await handler(model_instance)
+                    args.append(model_instance)
                 else:
                     # Fall back if type hints not available
-                    if accepts_auth:
-                        result = await handler(BaseModel(**params), auth_data)
-                    else:
-                        result = await handler(BaseModel(**params))
+                    args.append(BaseModel(**params))
+
+                # Check signature to see if it accepts other values
+
+                # TODO: Change this to "auth"
+                if auth_param := sig.parameters.get("auth_data"):
+                    kwargs[auth_param.name] = auth_data
+
+                if environment_param := sig.parameters.get("environment"):
+                    kwargs[environment_param.name] = environment
+
+                result = await handler(*args, **kwargs)
 
                 print(f"Tool {name} returned: {result}")
                 return result

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.2.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.1.dev0
+Version: 0.1.2.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.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.2.dev0}/pyproject.toml

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