beekeeper-core 1.0.4__tar.gz → 1.0.6__tar.gz

{beekeeper_core-1.0.4 → beekeeper_core-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beekeeper-core
-Version: 1.0.4
+Version: 1.0.6
 Summary: Load any data in one line of code and connect with AI applications
 Project-URL: Repository, https://github.com/beekeeper-ai/beekeeper
 Author-email: Leonardo Furnielis <leonardofurnielis@outlook.com>

beekeeper_core-1.0.6/beekeeper/core/observers/types.py

@@ -0,0 +1,11 @@
+from pydantic import BaseModel
+
+
+class PayloadRecord(BaseModel):
+    """Payload record."""
+
+    input_text: str
+    generated_text: str
+    generated_token_count: int
+    input_token_count: int
+    response_time: int

{beekeeper_core-1.0.4 → beekeeper_core-1.0.6}/beekeeper/core/prompts/base.py

@@ -22,6 +22,18 @@ class PromptTemplate(BaseModel):
     def __init__(self, template: str):
         super().__init__(template=template)
 
+    @classmethod
+    def from_value(cls, value: str) -> "PromptTemplate":
+        if isinstance(value, cls):
+            return value
+
+        if isinstance(value, str):
+            return cls(value)
+
+        raise TypeError(
+            f"Invalid type for parameter 'prompt_template'. Expected str or PromptTemplate, but received {type(value).__name__}."
+        )
+
     def format(self, **kwargs):
         """
         Formats the template using the provided dynamic variables.

{beekeeper_core-1.0.4 → beekeeper_core-1.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "beekeeper-core"
-version = "1.0.4"
+version = "1.0.6"
 description = "Load any data in one line of code and connect with AI applications"
 authors = [{ name = "Leonardo Furnielis", email = "leonardofurnielis@outlook.com" }]
 license = { text = "Apache-2.0" }