channel id config working

discord_bot/__main__.py

@@ -16,7 +16,7 @@ qa_engine = QAEngine(
 )
 client = DiscordClient(
     qa_engine=qa_engine,
-    channel_ids=config.discotd_channel_ids,
+    channel_ids=config.discord_channel_ids,
     num_last_messages=config.num_last_messages,
     use_names_in_context=config.use_names_in_context,
     enable_commands=config.enable_commands,

discord_bot/client/client.py

@@ -46,7 +46,7 @@ class DiscordClient(discord.Client):
             'The number of last messages in context should be at least 1'
 
         self.qa_engine: QAEngine = qa_engine
-        self.channel_ids: list[int] = channel_ids
+        self.channel_ids: list[int] = eval(channel_ids)
         self.num_last_messages: int = num_last_messages
         self.use_names_in_context: bool = use_names_in_context
         self.enable_commands: bool = enable_commands

qa_engine/config.py

@@ -36,7 +36,7 @@ class Config:
 
     # Discord bot config - optional
     discord_token: str = get_env('DISCORD_TOKEN', '-', warn=False)
-    discotd_channel_ids: list[int] = get_env('DISCORD_CHANNEL_IDS', field(default_factory=list), warn=False)
+    discord_channel_ids: list[int] = get_env('DISCORD_CHANNEL_IDS', field(default_factory=list), warn=False)
     num_last_messages: int = int(get_env('NUM_LAST_MESSAGES', 2, warn=False))
     use_names_in_context: bool = eval(get_env('USE_NAMES_IN_CONTEXT', 'False', warn=False))
     enable_commands: bool = eval(get_env('ENABLE_COMMANDS', 'True', warn=False))

qa_engine/qa_engine.py

@@ -86,9 +86,6 @@ class TransformersPipelineModel(LLM):
         )
 
     def _call(self, prompt: str, stop: Optional[list[str]] = None) -> str:
-        from llmlingua import PromptCompressor
-        llm_lingua = PromptCompressor()
-        prompt = llm_lingua.compress_prompt(prompt, instruction="", question="", target_token=400)
         output_text = self.pipeline(prompt)[0]['generated_text']
         output_text = output_text.replace(prompt+'\n', '')
         return output_text