shaytan/ai-agents-masterclass
1
0
Fork 0
mirror of https://github.com/coleam00/ai-agents-masterclass.git synced 2025-11-29 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updating LLM eval for Llama 3.2 testing

Browse Source
This commit is contained in:
Cole Medin
2024-09-28 21:15:54 -05:00
parent 0408d5b9f7
commit 07cd054a1b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
llm-agent-evaluation-framework/runnable.py
View File

@@ -26,9 +26,10 @@ model_mapping = {
"gpt": ChatOpenAI,
"claude": ChatAnthropic,
"groq": ChatGroq,
"llama": ChatHuggingFace # ChatHuggingFace doesn't work for tool calling yet with HuggingFaceEndpoint but will in the future
"llama": ChatGroq
}
# Support for HuggingFace with local models coming soon! This function isn't used yet.
@st.cache_resource
def get_local_model():
return HuggingFaceEndpoint(
@@ -54,7 +55,7 @@ tools = [tool for _, tool in available_functions.items()]
for key, chatbot_class in model_mapping.items():
if key in model.lower():
chatbot = chatbot_class(model=model) if key != "llama" else chatbot_class(llm=get_local_model())
chatbot = chatbot_class(model=model) if key != "huggingface" else chatbot_class(llm=get_local_model())
break
chatbot_with_tools = chatbot.bind_tools(tools)