igbot-base 0.0.8__tar.gz → 0.0.9__tar.gz

{igbot_base-0.0.8 → igbot_base-0.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.8
+Version: 0.0.9
 Summary: Base classes for igbot
 Author-email: Igor Kopeć <igor.kopec95@gmail.com>
 License: LGPL-3.0-or-later

{igbot_base-0.0.8 → igbot_base-0.0.9}/igbot_base/llm.py

@@ -30,7 +30,7 @@ class Llm(ABC):
         history.revert_to_snapshot()
 
     def call(self, user_query: str, history: LlmMemory, params: dict) -> str:
-        #history.set_snapshot()
+        history.set_snapshot()
         try:
             return self._call(user_query, history, params)
         except Exception as e:

igbot_base-0.0.9/igbot_base/retriever.py

@@ -0,0 +1,24 @@
+from abc import ABC, abstractmethod
+
+
+class RetrieverResponse:
+
+    def __init__(self, documents):
+        self.documents = documents
+
+    def get_content_list(self):
+        return [doc.page_content for doc in self.documents]
+
+    def get_content_string(self, delimiter: str):
+        contents = self.get_content_list()
+        return delimiter.join(contents)
+
+
+class Retriever(ABC):
+
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    def get_relevant_data(self, query: str) -> RetrieverResponse:
+        pass

{igbot_base-0.0.8 → igbot_base-0.0.9}/igbot_base.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.8
+Version: 0.0.9
 Summary: Base classes for igbot
 Author-email: Igor Kopeć <igor.kopec95@gmail.com>
 License: LGPL-3.0-or-later

{igbot_base-0.0.8 → igbot_base-0.0.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "igbot_base"
-version = "0.0.8"
+version = "0.0.9"
 description = "Base classes for igbot"
 authors = [{name = "Igor Kopeć", email = "igor.kopec95@gmail.com"}]
 license={text="LGPL-3.0-or-later"}

igbot_base-0.0.8/igbot_base/retriever.py

@@ -1,11 +0,0 @@
-from abc import ABC, abstractmethod
-
-
-class Retriever(ABC):
-
-    def __init__(self):
-        pass
-
-    @abstractmethod
-    def get_relevant_data(self, query: str):
-        pass