freeplay 0.3.24__tar.gz → 0.3.25__tar.gz

{freeplay-0.3.24 → freeplay-0.3.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: freeplay
-Version: 0.3.24
+Version: 0.3.25
 Summary: 
 License: MIT
 Author: FreePlay Engineering

{freeplay-0.3.24 → freeplay-0.3.25}/pyproject.toml

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

{freeplay-0.3.24 → freeplay-0.3.25}/src/freeplay/freeplay_cli.py

@@ -50,7 +50,53 @@ def download(project_id: str, environment: str, output_dir: str) -> None:
         click.echo(f"Found {len(prompts.prompt_templates)} prompt templates")
 
         for prompt in prompts.prompt_templates:
-            __write_single_file(environment, output_dir, project_id, prompt)
+            __write_single_file(environment, output_dir, prompt)
+    except FreeplayClientError as e:
+        print(f"Error downloading templates: {e}.\nIs your project ID correct?", file=sys.stderr)
+        exit(1)
+    except FreeplayServerError as e:
+        print(f"Error on Freeplay's servers downloading templates: {e}.\nTry again after a short wait.",
+              file=sys.stderr)
+        exit(2)
+    except Exception as e:
+        print(f"Error downloading templates: {e}", file=sys.stderr)
+        exit(3)
+
+
+@cli.command()
+@click.option("--environment", required=True, help="The environment from which the prompts will be pulled.", default="latest")
+@click.option("--output-dir", required=True, help="The directory where the prompts will be saved.")
+def download_all(environment: str, output_dir: str) -> None:
+    if "FREEPLAY_API_KEY" not in os.environ:
+        print("FREEPLAY_API_KEY is not set. It is required to run the freeplay command.", file=sys.stderr)
+        exit(4)
+
+    if "FREEPLAY_SUBDOMAIN" not in os.environ:
+        print("FREEPLAY_SUBDOMAIN is not set. It is required to run the freeplay command.", file=sys.stderr)
+        exit(4)
+
+    FREEPLAY_API_KEY = os.environ["FREEPLAY_API_KEY"]
+    freeplay_api_url = f'https://{os.environ["FREEPLAY_SUBDOMAIN"]}.freeplay.ai/api'
+
+    if "FREEPLAY_API_URL" in os.environ:
+        freeplay_api_url = f'{os.environ["FREEPLAY_API_URL"]}/api'
+        click.echo("Using URL override for Freeplay specified in the FREEPLAY_API_URL environment variable")
+
+    click.echo(
+        f"Downloading prompts for environment {environment}, "
+        f"to directory {output_dir} from {freeplay_api_url}")
+
+    fp_client = Freeplay(
+        freeplay_api_key=FREEPLAY_API_KEY,
+        api_base=freeplay_api_url
+    )
+
+    try:
+        prompts: PromptTemplates = fp_client.prompts.get_all_for_environment(environment=environment)
+        click.echo(f"Found {len(prompts.prompt_templates)} prompt templates")
+
+        for prompt in prompts.prompt_templates:
+            __write_single_file(environment, output_dir, prompt)
     except FreeplayClientError as e:
         print(f"Error downloading templates: {e}.\nIs your project ID correct?", file=sys.stderr)
         exit(1)
@@ -66,10 +112,9 @@ def download(project_id: str, environment: str, output_dir: str) -> None:
 def __write_single_file(
         environment: str,
         output_dir: str,
-        project_id: str,
         prompt: PromptTemplate
 ) -> None:
-    directory = __root_dir(environment, output_dir, project_id)
+    directory = __root_dir(environment, output_dir, prompt.project_id)
     basename = f'{prompt.prompt_template_name}'
     prompt_path = directory / f'{basename}.json'
     click.echo("Writing prompt file: %s" % prompt_path)
@@ -87,6 +132,6 @@ def __write_single_file(
 
 
 def __root_dir(environment: str, output_dir: str, project_id: str) -> Path:
-    directory = Path(output_dir) / "freeplay" / "prompts" / project_id / environment
+    directory = Path(output_dir).resolve() / "freeplay" / "prompts" / project_id / environment
     os.makedirs(directory, exist_ok=True)
     return directory

{freeplay-0.3.24 → freeplay-0.3.25}/src/freeplay/resources/prompts.py

@@ -557,6 +557,9 @@ class Prompts:
 
         return TemplatePrompt(prompt_info, prompt.content, prompt.tool_schema)
 
+    def get_all_for_environment(self, environment: str) -> PromptTemplates:
+        return self.call_support.get_prompts_for_environment(environment=environment)
+
     def get_by_version_id(self, project_id: str, template_id: str, version_id: str) -> TemplatePrompt:
         prompt = self.template_resolver.get_prompt_version_id(project_id, template_id, version_id)
 

{freeplay-0.3.24 → freeplay-0.3.25}/src/freeplay/support.py

@@ -76,6 +76,17 @@ class SummaryStatistics:
     human_evaluation: Dict[str, Any]
 
 
+@dataclass
+class ProjectInfo:
+    id: str
+    name: str
+
+
+@dataclass
+class ProjectInfos:
+    projects: List[ProjectInfo]
+
+
 class PromptTemplateEncoder(JSONEncoder):
     def default(self, prompt_template: PromptTemplate) -> Dict[str, Any]:
         return prompt_template.__dict__
@@ -89,6 +100,7 @@ class TestCaseTestRunResponse:
         self.history: Optional[List[Dict[str, Any]]] = test_case.get('history')
         self.custom_metadata: Optional[Dict[str, str]] = test_case.get('custom_metadata')
 
+
 class TraceTestCaseTestRunResponse:
     def __init__(self, test_case: Dict[str, Any]):
         self.id: str = test_case['test_case_id']
@@ -96,6 +108,7 @@ class TraceTestCaseTestRunResponse:
         self.output: Optional[str] = test_case.get('output')
         self.custom_metadata: Optional[Dict[str, str]] = test_case.get('custom_metadata')
 
+
 class TestRunResponse:
     def __init__(
             self,
@@ -185,6 +198,26 @@ class CallSupport:
 
         return maybe_prompts
 
+    def get_prompts_for_environment(self, environment: str) -> PromptTemplates:
+        projects_response = api_support.get_raw(
+            api_key=self.freeplay_api_key,
+            url=f'{self.api_base}/v2/projects/all'
+        )
+        if projects_response.status_code != 200:
+            raise freeplay_response_error("Error getting prompt templates", projects_response)
+
+        maybe_projects: Optional[ProjectInfos] = try_decode(ProjectInfos, projects_response.content)
+        if maybe_projects is None:
+            raise FreeplayServerError('Failed to parse list of projects from server')
+
+        prompt_templates = PromptTemplates([])
+        for project in maybe_projects.projects:
+            prompt_templates.prompt_templates.extend(
+                self.get_prompts(project.id, environment).prompt_templates
+            )
+
+        return prompt_templates
+
     def get_prompt(self, project_id: str, template_name: str, environment: str) -> PromptTemplate:
         response = api_support.get_raw(
             api_key=self.freeplay_api_key,