From 0a0f30c53b4af9d4de72541a00f0bd1e5ef273e3 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Sun, 19 Jan 2025 12:14:28 -0800 Subject: [PATCH] Update relevant agent tool descriptions Remove text (as by default, must output text), and improve the Notes description for clarity --- src/khoj/utils/helpers.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/khoj/utils/helpers.py b/src/khoj/utils/helpers.py index e3cab3c7..3dfbc876 100644 --- a/src/khoj/utils/helpers.py +++ b/src/khoj/utils/helpers.py @@ -354,7 +354,7 @@ command_descriptions = { } command_descriptions_for_agent = { - ConversationCommand.General: "Agent can use the agents knowledge base and general knowledge.", + ConversationCommand.General: "Agent can use its own knowledge base and general knowledge.", ConversationCommand.Notes: "Agent can search the personal knowledge base for information, as well as its own.", ConversationCommand.Online: "Agent can search the internet for information.", ConversationCommand.Webpage: "Agent can read suggested web pages for information.", @@ -388,7 +388,6 @@ mode_descriptions_for_llm = { mode_descriptions_for_agent = { ConversationCommand.Image: "Agent can generate images in response. It cannot not use this to generate charts and graphs.", ConversationCommand.Automation: "Agent can schedule a task to run at a scheduled date, time and frequency in response.", - ConversationCommand.Text: "Agent can generate text in response.", ConversationCommand.Diagram: "Agent can generate a visual representation that requires primitives like lines, rectangles, and text.", } @@ -433,6 +432,18 @@ def generate_random_name(): return name +def generate_random_internal_agent_name(): + random_name = generate_random_name() + + random_name.replace(" ", "_") + + random_number = random.randint(1000, 9999) + + name = f"{random_name}{random_number}" + + return name + + def batcher(iterable, max_n): "Split an iterable into chunks of size max_n" it = iter(iterable)