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

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.3.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.3.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.3.dev0}/opal_tools_sdk/decorators.py

@@ -1,7 +1,7 @@
 import inspect
 import re
 import logging
-from typing import Callable, Any, List, Dict, Type, get_type_hints, Optional, Union
+from typing import Callable, Any, List, Dict, Type, get_origin, get_type_hints, Optional, Union
 from fastapi import APIRouter, Depends, Header, HTTPException
 from pydantic import BaseModel
 
@@ -69,13 +69,17 @@ def tool(name: str, description: str, auth_requirements: Optional[List[Dict[str,
                     param_type = ParameterType.number
                 elif field_type == bool:
                     param_type = ParameterType.boolean
-                elif field_type == list or field_type == List:
+                elif get_origin(field_type) is list:
                     param_type = ParameterType.list
-                elif field_type == dict or field_type == Dict:
+                elif get_origin(field_type) is dict:
                     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.3.dev0}/opal_tools_sdk/models.py

@@ -8,7 +8,7 @@ class ParameterType(str, Enum):
     integer = "integer"
     number = "number"
     boolean = "boolean"
-    list = "list"  # Changed to match main service expectation
+    list = "array"  # Changed to match main service expectation
     dictionary = "object"  # Standard JSON schema type
 
 @dataclass
@@ -18,7 +18,7 @@ class Parameter:
     param_type: ParameterType
     description: str
     required: bool
-    
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for the discovery endpoint."""
         return {
@@ -34,7 +34,7 @@ class AuthRequirement:
     provider: str  # e.g., "google", "microsoft"
     scope_bundle: str  # e.g., "calendar", "drive"
     required: bool = True
-    
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for the discovery endpoint."""
         return {
@@ -52,7 +52,7 @@ class Function:
     endpoint: str
     auth_requirements: Optional[List[AuthRequirement]] = None
     http_method: str = "POST"
-    
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for the discovery endpoint."""
         result = {
@@ -62,8 +62,8 @@ class Function:
             "endpoint": self.endpoint,
             "http_method": self.http_method
         }
-        
+
         if self.auth_requirements:
             result["auth_requirements"] = [auth.to_dict() for auth in self.auth_requirements]
-            
+
         return result

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.3.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.3.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.3.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.3.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.3-dev"
 description = "SDK for creating Opal-compatible tools services"
 authors = [
     {name = "Optimizely", email = "opal-team@optimizely.com"}

{optimizely_opal_opal_tools_sdk-0.1.1.dev0 → optimizely_opal_opal_tools_sdk-0.1.3.dev0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="optimizely-opal.opal-tools-sdk",
-    version="0.1.1-dev",
+    version="0.1.3-dev",
     packages=find_packages(),
     install_requires=[
         "fastapi>=0.100.0",