freeplay 0.3.4__tar.gz → 0.3.6__tar.gz

{freeplay-0.3.4 → freeplay-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: freeplay
-Version: 0.3.4
+Version: 0.3.6
 Summary: 
 License: MIT
 Author: FreePlay Engineering

{freeplay-0.3.4 → freeplay-0.3.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "freeplay"
-version = "0.3.4"
+version = "0.3.6"
 description = ""
 authors = ["FreePlay Engineering <engineering@freeplay.ai>"]
 license = "MIT"

{freeplay-0.3.4 → freeplay-0.3.6}/src/freeplay/model.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import List, Union, Any, Dict, Mapping, TypedDict
 
-InputValue = Union[str, int, bool, Dict[str, Any], List[Any]]
+InputValue = Union[str, int, bool, float, Dict[str, Any], List[Any]]
 InputVariables = Mapping[str, InputValue]
 TestRunInput = Mapping[str, InputValue]
 FeedbackValue = Union[bool, str, int, float]

{freeplay-0.3.4 → freeplay-0.3.6}/src/freeplay/resources/prompts.py

@@ -256,13 +256,13 @@ class FilesystemTemplateResolver(TemplateResolver):
     def __render_into_v2(json_dom: Dict[str, Any]) -> PromptTemplate:
         format_version = json_dom.get('format_version')
 
-        if format_version == 2:
+        if format_version and format_version >= 2:
             metadata = json_dom['metadata']
             flavor_name = metadata.get('flavor')
             model = metadata.get('model')
 
             return PromptTemplate(
-                format_version=2,
+                format_version=format_version,
                 prompt_template_id=json_dom.get('prompt_template_id'),  # type: ignore
                 prompt_template_version_id=json_dom.get('prompt_template_version_id'),  # type: ignore
                 prompt_template_name=json_dom.get('prompt_template_name'),  # type: ignore

{freeplay-0.3.4 → freeplay-0.3.6}/src/freeplay/utils.py

@@ -10,7 +10,7 @@ from .model import InputVariables
 
 # Validate that the variables are of the correct type, and do not include functions, dates, classes or None values.
 def all_valid(obj: Any) -> bool:
-    if isinstance(obj, (int, str, bool)):
+    if isinstance(obj, (int, str, bool, float)):
         return True
     elif isinstance(obj, list):
         return all(all_valid(item) for item in obj)