sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-30 19:03:01 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Construct available data sources, output modes as a bullet list in prompts

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-03-11 13:08:04 +05:30
parent f28fb89af8
commit 9a516bed47
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/khoj/routers/helpers.py
View file

@ -153,15 +153,17 @@ async def aget_relevant_information_sources(query: str, conversation_history: di
""" """
tool_options = dict() tool_options = dict()
tool_options_str = ""
for tool, description in tool_descriptions_for_llm.items(): for tool, description in tool_descriptions_for_llm.items():
tool_options[tool.value] = description tool_options[tool.value] = description
tool_options_str += f'- "{tool.value}": "{description}"\n'
chat_history = construct_chat_history(conversation_history) chat_history = construct_chat_history(conversation_history)
relevant_tools_prompt = prompts.pick_relevant_information_collection_tools.format( relevant_tools_prompt = prompts.pick_relevant_information_collection_tools.format(
query=query, query=query,
tools=str(tool_options), tools=tool_options_str,
chat_history=chat_history, chat_history=chat_history,
) )
@ -195,15 +197,17 @@ async def aget_relevant_output_modes(query: str, conversation_history: dict):
""" """
mode_options = dict() mode_options = dict()
mode_options_str = ""
for mode, description in mode_descriptions_for_llm.items(): for mode, description in mode_descriptions_for_llm.items():
mode_options[mode.value] = description mode_options[mode.value] = description
mode_options_str += f'- "{mode.value}": "{description}"\n'
chat_history = construct_chat_history(conversation_history) chat_history = construct_chat_history(conversation_history)
relevant_mode_prompt = prompts.pick_relevant_output_mode.format( relevant_mode_prompt = prompts.pick_relevant_output_mode.format(
query=query, query=query,
modes=str(mode_options), modes=mode_options_str,
chat_history=chat_history, chat_history=chat_history,
) )