diff --git a/evaluation_script.py b/evaluation_script.py index 6495e50..2bbc72f 100644 --- a/evaluation_script.py +++ b/evaluation_script.py @@ -459,6 +459,12 @@ def make_profiles(agent_names, models, apis, template_profile="profiles/collab_p "model": models[index], "url": url } + elif apis[index] == "ollama": + profile["model"] = { + "api": "ollama", + "model": models[index], + "embedding": "ollama" + } else: profile["model"] = models[index]