fix: Resolve Open Source Model Initialization in model_factory.py (#646)

camel-ai · Jun 16, 2024 · 99a11fa · 99a11fa
1 parent 53b9308
commit 99a11fa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/camel/models/model_factory.py b/camel/models/model_factory.py
@@ -64,5 +64,9 @@ def create(
         else:
             raise ValueError(f"Unknown model type `{model_type}` is input")
 
-        inst = model_class(model_type, model_config_dict, api_key)
+        if model_type.is_open_source:
+            inst = model_class(model_type, model_config_dict)
+        else:
+            inst = model_class(model_type, model_config_dict, api_key)
+
         return inst