Skip to content

Commit

Permalink
update prompts for tools
Browse files Browse the repository at this point in the history
  • Loading branch information
omar-sol committed Jul 29, 2024
1 parent 34b6f5e commit a1b20d4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def update_query_engine_tools(selected_sources):
"Transformers Docs": (
custom_retriever_transformers,
"Transformers_information",
"""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.""",
"""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.""",
),
"PEFT Docs": (
custom_retriever_peft,
Expand All @@ -40,7 +40,7 @@ def update_query_engine_tools(selected_sources):
"LlamaIndex Docs": (
custom_retriever_llama_index,
"LlamaIndex_information",
"""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.""",
"""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.""",
),
"OpenAI Cookbooks": (
custom_retriever_openai_cookbooks,
Expand Down

0 comments on commit a1b20d4

Please sign in to comment.