freeplay 0.2.35__tar.gz → 0.2.37__tar.gz

{freeplay-0.2.35 → freeplay-0.2.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: freeplay
-Version: 0.2.35
+Version: 0.2.37
 Summary: 
 License: MIT
 Author: FreePlay Engineering
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: anthropic (>=0.7.7,<0.8.0)
 Requires-Dist: click (==8.1.7)
 Requires-Dist: dacite (>=1.8.0,<2.0.0)

{freeplay-0.2.35 → freeplay-0.2.37}/pyproject.toml

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

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/flavors.py

@@ -50,7 +50,7 @@ class Flavor(ABC):
         pass
 
     @abstractmethod
-    def to_llm_syntax(self, messages: List[ChatMessage]) -> str | List[ChatMessage]:
+    def to_llm_syntax(self, messages: List[ChatMessage]) -> Union[str, List[ChatMessage]]:
         raise NotImplementedError()
 
     @abstractmethod

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/model.py

@@ -3,7 +3,7 @@ from typing import List, Union, Any, Dict, Mapping
 
 from pydantic import RootModel
 
-InputValue = Union[str, int, bool, dict[str, Any], list[Any]]
+InputValue = Union[str, int, bool, Dict[str, Any], List[Any]]
 InputVariable = RootModel[Union[Dict[str, "InputVariable"], List["InputVariable"], str, int, bool, float]]
 InputVariable.model_rebuild()
 

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/support.py

@@ -28,7 +28,7 @@ class TestRunResponse:
     def __init__(
             self,
             test_run_id: str,
-            test_cases: list[JsonDom]
+            test_cases: List[JsonDom]
     ):
         self.test_cases = [
             TestCaseTestRunResponse(test_case)

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/thin/freeplay_thin.py

@@ -1,8 +1,10 @@
+from typing import Optional
+
 from freeplay.errors import FreeplayConfigurationError
 from freeplay.record import DefaultRecordProcessor
 from freeplay.support import CallSupport
 from freeplay.thin.resources.customer_feedback import CustomerFeedback
-from freeplay.thin.resources.prompts import Prompts
+from freeplay.thin.resources.prompts import Prompts, APITemplateResolver, TemplateResolver
 from freeplay.thin.resources.recordings import Recordings
 from freeplay.thin.resources.sessions import Sessions
 from freeplay.thin.resources.test_runs import TestRuns
@@ -12,7 +14,8 @@ class Freeplay:
     def __init__(
             self,
             freeplay_api_key: str,
-            api_base: str
+            api_base: str,
+            template_resolver: Optional[TemplateResolver] = None
     ) -> None:
         if not freeplay_api_key or not freeplay_api_key.strip():
             raise FreeplayConfigurationError("Freeplay API key not set. It must be set to the Freeplay API.")
@@ -22,9 +25,15 @@ class Freeplay:
         self.freeplay_api_key = freeplay_api_key
         self.api_base = api_base
 
+        resolver: TemplateResolver
+        if template_resolver is None:
+            resolver = APITemplateResolver(self.call_support)
+        else:
+            resolver = template_resolver
+
         # Resources ========
         self.customer_feedback = CustomerFeedback(self.call_support)
-        self.prompts = Prompts(self.call_support)
+        self.prompts = Prompts(self.call_support, resolver)
         self.recordings = Recordings(self.call_support)
         self.sessions = Sessions()
         self.test_runs = TestRuns(self.call_support)

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/thin/resources/prompts.py

@@ -1,8 +1,10 @@
 import json
+from abc import ABC, abstractmethod
 from dataclasses import dataclass
-from typing import Optional, List, Union, cast
+from pathlib import Path
+from typing import Dict, Optional, List, Union, cast
 
-from freeplay.completions import PromptTemplates, ChatMessage
+from freeplay.completions import PromptTemplates, ChatMessage, PromptTemplateWithMetadata
 from freeplay.errors import FreeplayConfigurationError
 from freeplay.flavors import Flavor
 from freeplay.llm_parameters import LLMParameters
@@ -27,8 +29,8 @@ class FormattedPrompt:
     def __init__(
             self,
             prompt_info: PromptInfo,
-            messages: List[dict[str, str]],
-            formatted_prompt: Union[str, List[dict[str, str]]]
+            messages: List[Dict[str, str]],
+            formatted_prompt: Union[str, List[Dict[str, str]]]
     ):
         self.prompt_info = prompt_info
         self.messages = messages
@@ -36,8 +38,8 @@ class FormattedPrompt:
 
     def all_messages(
             self,
-            new_message: dict[str, str]
-    ) -> List[dict[str, str]]:
+            new_message: Dict[str, str]
+    ) -> List[Dict[str, str]]:
         return self.messages + [new_message]
 
 
@@ -45,7 +47,7 @@ class BoundPrompt:
     def __init__(
             self,
             prompt_info: PromptInfo,
-            messages: List[dict[str, str]]
+            messages: List[Dict[str, str]]
     ):
         self.prompt_info = prompt_info
         self.messages = messages
@@ -61,7 +63,7 @@ class BoundPrompt:
         return FormattedPrompt(
             self.prompt_info,
             self.messages,
-            cast(Union[str, List[dict[str, str]]], llm_format)
+            cast(Union[str, List[Dict[str, str]]], llm_format)
         )
 
 
@@ -69,7 +71,7 @@ class TemplatePrompt:
     def __init__(
             self,
             prompt_info: PromptInfo,
-            messages: List[dict[str, str]]
+            messages: List[Dict[str, str]]
     ):
         self.prompt_info = prompt_info
         self.messages = messages
@@ -82,19 +84,85 @@ class TemplatePrompt:
         return BoundPrompt(self.prompt_info, bound_messages)
 
 
+class TemplateResolver(ABC):
+    @abstractmethod
+    def get_prompts(self, project_id: str, environment: str) -> PromptTemplates:
+        pass
+
+
+class FilesystemTemplateResolver(TemplateResolver):
+
+    def __init__(self, freeplay_directory: Path):
+        FilesystemTemplateResolver.__validate_freeplay_directory(freeplay_directory)
+        self.prompts_directory = freeplay_directory / "freeplay" / "prompts"
+
+    def get_prompts(self, project_id: str, environment: str) -> PromptTemplates:
+        self.__validate_prompt_directory(project_id, environment)
+
+        directory = self.prompts_directory / project_id / environment
+        prompt_file_paths = directory.glob("*.json")
+
+        prompt_list = []
+        for prompt_file_path in prompt_file_paths:
+            json_dom = json.loads(prompt_file_path.read_text())
+
+            prompt_list.append(PromptTemplateWithMetadata(
+                prompt_template_id=json_dom.get('prompt_template_id'),
+                prompt_template_version_id=json_dom.get('prompt_template_version_id'),
+                name=json_dom.get('name'),
+                content=json_dom.get('content'),
+                flavor_name=json_dom.get('metadata').get('flavor_name'),
+                params=json_dom.get('metadata').get('params')
+            ))
+
+        return PromptTemplates(prompt_list)
+
+    @staticmethod
+    def __validate_freeplay_directory(freeplay_directory: Path) -> None:
+        if not freeplay_directory.is_dir():
+            raise FreeplayConfigurationError(
+                "Path for prompt templates is not a valid directory (%s)" % freeplay_directory
+            )
+
+        prompts_directory = freeplay_directory / "freeplay" / "prompts"
+        if not prompts_directory.is_dir():
+            raise FreeplayConfigurationError(
+                "Invalid path for prompt templates (%s). "
+                "Did not find a freeplay/prompts directory underneath." % freeplay_directory
+            )
+
+    def __validate_prompt_directory(self, project_id: str, environment: str) -> None:
+        maybe_prompt_dir = self.prompts_directory / project_id / environment
+        if not maybe_prompt_dir.is_dir():
+            raise FreeplayConfigurationError(
+                "Could not find prompt template directory for project ID %s and environment %s." %
+                (project_id, environment)
+            )
+
+
+class APITemplateResolver(TemplateResolver):
+
+    def __init__(self, call_support: CallSupport):
+        self.call_support = call_support
+
+    def get_prompts(self, project_id: str, environment: str) -> PromptTemplates:
+        return self.call_support.get_prompts(
+            project_id=project_id,
+            tag=environment
+        )
+
+
 class Prompts:
-    def __init__(self, call_support: CallSupport) -> None:
+    def __init__(self, call_support: CallSupport, template_resolver: TemplateResolver) -> None:
         self.call_support = call_support
+        self.template_resolver = template_resolver
 
     def get_all(self, project_id: str, environment: str) -> PromptTemplates:
         return self.call_support.get_prompts(project_id=project_id, tag=environment)
 
     def get(self, project_id: str, template_name: str, environment: str) -> TemplatePrompt:
-        prompt_template = self.call_support.get_prompt(
-            project_id=project_id,
-            template_name=template_name,
-            environment=environment
-        )
+        prompt_templates = self.template_resolver.get_prompts(project_id, environment)
+        prompt_template = self.call_support.find_template_by_name(prompt_templates, template_name)
 
         messages = json.loads(prompt_template.content)
 

{freeplay-0.2.35 → freeplay-0.2.37}/src/freeplay/thin/resources/recordings.py

@@ -1,7 +1,7 @@
 import json
 import logging
 from dataclasses import dataclass
-from typing import Optional, List
+from typing import Dict, Optional, List
 
 from freeplay import api_support
 from freeplay.completions import PromptTemplateWithMetadata, OpenAIFunctionCall
@@ -50,7 +50,7 @@ class TestRunInfo:
 
 @dataclass
 class RecordPayload:
-    all_messages: List[dict[str, str]]
+    all_messages: List[Dict[str, str]]
     inputs: InputVariables
 
     session_info: SessionInfo