Omar Solano commited on
Commit
a1b20d4
Β·
1 Parent(s): 34b6f5e

update prompts for tools

Browse files
Files changed (1) hide show
  1. scripts/main.py +2 -2
scripts/main.py CHANGED
@@ -25,7 +25,7 @@ def update_query_engine_tools(selected_sources):
25
  "Transformers Docs": (
26
  custom_retriever_transformers,
27
  "Transformers_information",
28
- """Useful for general questions asking about the artificial intelligence (AI) field. Employ this tool to fetch general information on topics such as language models theory (transformer architectures), tips on prompting, models, quantization, etc.""",
29
  ),
30
  "PEFT Docs": (
31
  custom_retriever_peft,
@@ -40,7 +40,7 @@ def update_query_engine_tools(selected_sources):
40
  "LlamaIndex Docs": (
41
  custom_retriever_llama_index,
42
  "LlamaIndex_information",
43
- """Useful for questions asking about retrieval augmented generation (RAG) with LLMs and embedding models. It is the documentation of the LlamaIndex framework, includes info about fine-tuning embedding models, building chatbots, and agents with llms, using vector databases, embeddings, information retrieval with cosine similarity or bm25, etc.""",
44
  ),
45
  "OpenAI Cookbooks": (
46
  custom_retriever_openai_cookbooks,
 
25
  "Transformers Docs": (
26
  custom_retriever_transformers,
27
  "Transformers_information",
28
+ """Useful for general questions asking about the artificial intelligence (AI) field. Employ this tool to fetch information on topics such as language models (LLMs) models such as Llama3 and theory (transformer architectures), tips on prompting, quantization, etc.""",
29
  ),
30
  "PEFT Docs": (
31
  custom_retriever_peft,
 
40
  "LlamaIndex Docs": (
41
  custom_retriever_llama_index,
42
  "LlamaIndex_information",
43
+ """Useful for questions asking about retrieval augmented generation (RAG) with LLMs and embedding models. It is the documentation of a framework, includes info about fine-tuning embedding models, building chatbots, and agents with llms, using vector databases, embeddings, information retrieval with cosine similarity or bm25, etc.""",
44
  ),
45
  "OpenAI Cookbooks": (
46
  custom_retriever_openai_cookbooks,