removed extractors

fractalego · Dec 15, 2024 · cfd4844 · cfd4844
1 parent 1de7c42
commit cfd4844
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 31 deletions.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,6 @@
         "wafl.connectors.remote",
         "wafl.data_objects",
         "wafl.events",
-        "wafl.extractors",
         "wafl.handlers",
         "wafl.inference",
         "wafl.interface",

diff --git a/tests/test_connection.py b/tests/test_connection.py
@@ -12,7 +12,7 @@
 class TestConnection(TestCase):
     def test__connection_to_generative_model_can_generate_text(self):
         config = Configuration.load_local_config()
-        connector = RemoteLLMConnector(config.get_value("llm_model"))
+        connector = RemoteLLMConnector(config)
         prediction = asyncio.run(
             connector.predict(
                 PromptCreator.create_from_one_instruction(
@@ -25,7 +25,7 @@ def test__connection_to_generative_model_can_generate_text(self):
 
     def test__connection_to_generative_model_can_generate_text_within_tags(self):
         config = Configuration.load_local_config()
-        connector = RemoteLLMConnector(config.get_value("llm_model"))
+        connector = RemoteLLMConnector(config)
         connector._num_prediction_tokens = 200
         text = 'Generate a full paragraph based on this chapter title " The First Contact". The theme of the paragraph is space opera. Include the characters "Alberto" and "Maria". Write at least three sentences.'
         prompt = f"""
@@ -43,7 +43,7 @@ def test__connection_to_generative_model_can_generate_text_within_tags(self):
 
     def test__connection_to_generative_model_can_generate_a_python_list(self):
         config = Configuration.load_local_config()
-        connector = RemoteLLMConnector(config.get_value("llm_model"))
+        connector = RemoteLLMConnector(config)
         connector._num_prediction_tokens = 200
         prompt = "Generate a Python list of 4 chapters names for a space opera book. The output needs to be a python list of strings: "
         prediction = asyncio.run(

diff --git a/tests/test_entailer.py b/tests/test_entailer.py
@@ -12,7 +12,7 @@
 class TestConnection(TestCase):
     def test__entailer_connector(self):
         config = Configuration.load_local_config()
-        connector = RemoteEntailerConnector(config.get_value("entailer_model"))
+        connector = RemoteEntailerConnector(config)
         prediction = asyncio.run(
             connector.predict(
                 "The first contact is a romance novel set in the middle ages.",

diff --git a/wafl/extractors/__init__.py b/wafl/extractors/__init__.py
diff --git a/wafl/extractors/utils.py b/wafl/extractors/utils.py
diff --git a/wafl/variables.py b/wafl/variables.py
@@ -1,9 +1,9 @@
 def get_variables():
     return {
-        "version": "0.1.2",
+        "version": "0.1.3",
     }
 
 
 def is_supported(wafl_llm_version):
-    supported_versions = ["0.1.0"]
+    supported_versions = ["0.1.1"]
     return wafl_llm_version in supported_versions