chore: add more models

webisopen · Nov 5, 2024 · b4265fa · b4265fa
1 parent 30b2006
commit b4265fa
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/openagent/conf/llm_provider.py b/openagent/conf/llm_provider.py
@@ -15,6 +15,10 @@
     "llama3.2": {"name": "llama3.2", "supports_tools": True},
     "mistral-nemo": {"name": "mistral-nemo", "supports_tools": True},
     "darkmoon/olmo:7B-instruct-q6-k": {"name": "olmo", "supports_tools": False},
+    'llama3.1': {'name': 'llama3.1', 'supports_tools': True},
+    "qwen2.5": {"name": "qwen2.5", "supports_tools": True},
+    "mistral": {"name": "mistral", "supports_tools": True},
+    "qwen2": {"name": "qwen2", "supports_tools": True},
 }
 
 MODELS_ICONS = {
@@ -24,6 +28,8 @@
     "mistral-nemo": "/public/mistral.png",
     "mistral-large": "/public/mistral.png",
     "olmo": "/public/olmo.png",
+    "qwen2": "/public/qwen.png",
+    "qwen2.5": "/public/qwen.png",
 }
 
 

diff --git a/public/qwen.png b/public/qwen.png
diff --git a/tests/run_test.py b/tests/run_test.py
@@ -13,6 +13,10 @@
 ]
 
 OPENSOURCE_MODELS = [
+    {"name": "qwen2", "function_call_support": True},
+    {"name": "mistral", "function_call_support": True},
+    {"name": "qwen2.5", "function_call_support": True},
+    {"name": "llama3.1", "function_call_support": True},
     {"name": "llama3.2", "function_call_support": True},
     {"name": "mistral-nemo", "function_call_support": True},
     # {"name": "darkmoon/olmo:7B-instruct-q6-k", "function_call_support": False}
@@ -95,7 +99,7 @@ def run_model_tests(model_name):
     print(f"\nTesting model: {model_name}")
     stats = TestStats()
     test_files = ["supervisor_chain.py"] \
-                 + glob.glob("agent_trajectory/*.py")
+                 # + glob.glob("agent_trajectory/*.py")
     pytest.main(["--count=1", "-n", "11"] + test_files + [f"--model={model_name}"] + sys.argv[1:], plugins=[stats])
     return stats.calculate_model_score()