freeplay 0.1.0__tar.gz → 0.1.1__tar.gz

{freeplay-0.1.0 → freeplay-0.1.1}/PKG-INFO

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

{freeplay-0.1.0 → freeplay-0.1.1}/freeplay/freeplay.py

@@ -7,10 +7,23 @@ from . import api_support
 T = t.TypeVar("T")
 
 
+@dataclass
+class ChatMessage:
+    role: str
+    content: str
+
+    def format(self, **kwargs: str) -> dict[str, str]:
+        return {
+            'role': self.role,
+            'content': self.content.format(**kwargs)
+        }
+
+
 @dataclass
 class FreePlayPromptTemplate:
     name: str
     content: str
+    messages: list[ChatMessage]
 
 
 @dataclass
@@ -21,13 +34,22 @@ class FreePlayProjectSession:
     def __all_names(self) -> list[str]:
         return [t.name for t in self.prompt_templates]
 
-    def get_template(self, template_name: str) -> str:
-        for t in self.prompt_templates:
-            if t.name == template_name:
-                return t.content
+    def __find_template(self, template_name: str) -> FreePlayPromptTemplate:
+        for template in self.prompt_templates:
+            if template.name == template_name:
+                return template
 
         raise Exception(f'Template with name {template_name} not found. Available names are: {self.__all_names()}')
 
+    def get_template(self, template_name: str) -> str:
+        return self.__find_template(template_name).content
+
+    def get_formatted_chat_messages(self, template_name: str, **kwargs: str) -> list[dict[str, str]]:
+        return [
+            message.format(**kwargs)
+            for message in self.__find_template(template_name).messages
+        ]
+
 
 @dataclass
 class FreePlayTestRunRecord:

{freeplay-0.1.0 → freeplay-0.1.1}/pyproject.toml

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