From 30e60362dee687094d1a4b539232500062a58a86 Mon Sep 17 00:00:00 2001 From: Alberto Cetoli Date: Sun, 4 Aug 2024 15:32:19 +0100 Subject: [PATCH 1/2] added entailer model to the templates --- wafl/templates/config.json | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/wafl/templates/config.json b/wafl/templates/config.json index 7af9893e..91e004e0 100644 --- a/wafl/templates/config.json +++ b/wafl/templates/config.json @@ -28,5 +28,9 @@ "text_embedding_model": { "model_host": "localhost", "model_port": 8080 + }, + "entailer_model": { + "model_host": "localhost", + "model_port": 8080 } } From 148c8f3503cac2589e41f5aaca39b3606248e7a2 Mon Sep 17 00:00:00 2001 From: Alberto Cetoli Date: Sun, 4 Aug 2024 15:33:36 +0100 Subject: [PATCH 2/2] fixed uncalled await --- wafl/answerer/answerer_implementation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wafl/answerer/answerer_implementation.py b/wafl/answerer/answerer_implementation.py index 9a2c25df..baca3f93 100644 --- a/wafl/answerer/answerer_implementation.py +++ b/wafl/answerer/answerer_implementation.py @@ -165,5 +165,5 @@ def add_memories_to_facts(facts: str, memories: List[str]) -> str: def select_best_rules_using_entailer(conversation: Conversation, rules_as_strings: List[str], entailer: Entailer, num_rules: int) -> str: query_text = conversation.get_last_speaker_utterance("user") - rules_as_strings = sorted(rules_as_strings, key=lambda x: entailer.get_score(query_text, x), reverse=True) + rules_as_strings = sorted(rules_as_strings, key=lambda x: await entailer.get_score(query_text, x), reverse=True) return rules_as_strings[:num_rules]