beekeeper-core 1.0.2__tar.gz → 1.0.3__tar.gz

{beekeeper_core-1.0.2 → beekeeper_core-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beekeeper-core
-Version: 1.0.2
+Version: 1.0.3
 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.2 → beekeeper_core-1.0.3}/beekeeper/core/embeddings/base.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import List
+from typing import List, Union
 
 import numpy as np
 from beekeeper.core.document import Document
@@ -43,17 +43,15 @@ class BaseEmbedding(TransformerComponent, ABC):
         return "BaseEmbedding"
 
     @abstractmethod
-    def embed_texts(self, texts: List[str]) -> List[Embedding]:
-        """Embed a list of text strings."""
-
-    def embed_text(self, query: str) -> Embedding:
-        """Embed a single text string."""
-        return self.embed_texts([query])[0]
+    def embed_text(
+        self, input: Union[str, List[str]]
+    ) -> Union[Embedding, List[Embedding]]:
+        """Embed one or more text strings."""
 
     def embed_documents(self, documents: List[Document]) -> List[Document]:
         """Embed a list of documents and set them in the 'embedding' attribute."""
         texts = [document.get_content() for document in documents]
-        embeddings = self.embed_texts(texts)
+        embeddings = self.embed_text(texts)
 
         for document, embedding in zip(documents, embeddings):
             document.embedding = embedding
@@ -61,7 +59,15 @@ class BaseEmbedding(TransformerComponent, ABC):
         return documents
 
     @deprecated(
-        reason="'get_text_embedding()' is deprecated and will be removed in a future version. Use 'embed_text' instead.",
+        reason="'embed_texts()' is deprecated and will be removed in a future version. Use 'embed_text()' instead.",
+        version="1.0.3",
+        action="always",
+    )
+    def embed_texts(self, texts: List[str]) -> List[Embedding]:
+        return self.embed_text(texts)
+
+    @deprecated(
+        reason="'get_text_embedding()' is deprecated and will be removed in a future version. Use 'embed_text()' instead.",
         version="1.0.2",
         action="always",
     )
@@ -69,7 +75,7 @@ class BaseEmbedding(TransformerComponent, ABC):
         return self.embed_text(query)
 
     @deprecated(
-        reason="'get_texts_embedding()' is deprecated and will be removed in a future version. Use 'embed_texts' instead.",
+        reason="'get_texts_embedding()' is deprecated and will be removed in a future version. Use 'embed_texts()' instead.",
         version="1.0.2",
         action="always",
     )
@@ -77,7 +83,7 @@ class BaseEmbedding(TransformerComponent, ABC):
         return self.embed_texts(texts)
 
     @deprecated(
-        reason="'get_documents_embedding()' is deprecated and will be removed in a future version. Use 'embed_documents' instead.",
+        reason="'get_documents_embedding()' is deprecated and will be removed in a future version. Use 'embed_documents()' instead.",
         version="1.0.2",
         action="always",
     )

{beekeeper_core-1.0.2 → beekeeper_core-1.0.3}/beekeeper/core/evaluation/context_similarity.py

@@ -53,10 +53,10 @@ class ContextSimilarityEvaluator(BaseModel):
             )
 
         evaluation_result = {"contexts_score": [], "score": 0}
-        candidate_embedding = self.embed_model.get_text_embedding(generated_text)
+        candidate_embedding = self.embed_model.embed_text(generated_text)
 
         for context in contexts:
-            context_embedding = self.embed_model.get_text_embedding(context)
+            context_embedding = self.embed_model.embed_text(context)
             evaluation_result["contexts_score"].append(
                 self.embed_model.similarity(
                     candidate_embedding,

beekeeper_core-1.0.3/beekeeper/core/observers/__init__.py

@@ -0,0 +1,3 @@
+from beekeeper.core.observers.base import BaseObserver, ModelObserver, PromptObserver
+
+__all__ = (["BaseObserver", "ModelObserver", "PromptObserver"],)

{beekeeper_core-1.0.2 → beekeeper_core-1.0.3}/beekeeper/core/observers/base.py

@@ -3,6 +3,7 @@ from typing import Optional
 
 from beekeeper.core.observers.types import PayloadRecord
 from beekeeper.core.prompts import PromptTemplate
+from deprecated import deprecated
 
 
 class BaseObserver(ABC):
@@ -13,7 +14,7 @@ class BaseObserver(ABC):
         return "BaseObserver"
 
 
-class ModelObserver(BaseObserver):
+class PromptObserver(BaseObserver):
     """Abstract base class defining the interface for prompt observability."""
 
     def __init__(self, prompt_template: Optional[PromptTemplate] = None) -> None:
@@ -21,11 +22,20 @@ class ModelObserver(BaseObserver):
 
     @classmethod
     def class_name(cls) -> str:
-        return "ModelObserver"
+        return "PromptObserver"
 
     @abstractmethod
     def __call__(self, payload: PayloadRecord) -> None:
-        """ModelObserver."""
+        """PromptObserver."""
+
+
+@deprecated(
+    reason="'ModelObserver()' is deprecated and will be removed in a future version. Use 'PromptObserver()' instead.",
+    version="1.0.3",
+    action="always",
+)
+class ModelObserver(PromptObserver):
+    """DEPRECATED: This class is deprecated and kept only for backward compatibility."""
 
 
 class TelemetryObserver(BaseObserver):

{beekeeper_core-1.0.2 → beekeeper_core-1.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "beekeeper-core"
-version = "1.0.2"
+version = "1.0.3"
 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" }

beekeeper_core-1.0.2/beekeeper/core/observers/__init__.py

@@ -1,3 +0,0 @@
-from beekeeper.core.observers.base import BaseObserver, ModelObserver
-
-__all__ = (["BaseObserver", "ModelObserver"],)