From 81eb2d24a6add775e46eab9c89873394d15a7b64 Mon Sep 17 00:00:00 2001 From: dilyararimovna Date: Fri, 28 Apr 2023 19:04:57 +0300 Subject: [PATCH 1/2] feat: smaller context for prompt selector --- annotators/prompt_selector/server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/annotators/prompt_selector/server.py b/annotators/prompt_selector/server.py index 4debf0fbd5..13accd7cd1 100644 --- a/annotators/prompt_selector/server.py +++ b/annotators/prompt_selector/server.py @@ -40,7 +40,10 @@ def get_result(request): context_ids = [] for context_id, context in enumerate(contexts): - str_context = "\n".join(context) + if len(context[-1].split()) < 5: + str_context = "\n".join(context[-3]) + else: + str_context = context[-1] for prompt in PROMPTS: pairs += [[str_context, prompt]] context_ids += [context_id] From e2148603b0dfde0194428220088e7bf7dee53cf5 Mon Sep 17 00:00:00 2001 From: dilyararimovna Date: Fri, 28 Apr 2023 19:09:27 +0300 Subject: [PATCH 2/2] fix: index error --- annotators/prompt_selector/server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/annotators/prompt_selector/server.py b/annotators/prompt_selector/server.py index 13accd7cd1..507dac16a0 100644 --- a/annotators/prompt_selector/server.py +++ b/annotators/prompt_selector/server.py @@ -41,7 +41,7 @@ def get_result(request): for context_id, context in enumerate(contexts): if len(context[-1].split()) < 5: - str_context = "\n".join(context[-3]) + str_context = "\n".join(context[-3:]) else: str_context = context[-1] for prompt in PROMPTS: